Overview

[dependencies]
env_logger = "0.11"

// Enable logging with default WARN level
env_logger::init();
// Or with a specific level: RUST_LOG=gpudb_api=debug cargo run

use gpudb_api::{
    Kinetica, KineticaOptions, BulkInserter, BulkInserterOptions,
    KineticaRecord, ShardKeyExtractor, ShardKeyValue, ShardKeyValues,
    avro_schema, AvroSchemaBuilder, TypePropertiesBuilder,
};
use smallvec::smallvec;
use std::sync::Arc;
use std::collections::HashMap;
use serde::Serialize;

// Define your record type
#[derive(Serialize, Clone, Default)]
struct MyRecord {
    id: i32,
    name: String,
}

impl ShardKeyExtractor for MyRecord {
    fn shard_key_values(&self) -> ShardKeyValues<'_> {
        smallvec![("id", ShardKeyValue::Int(self.id))]
    }
}

impl KineticaRecord for MyRecord {
    fn avro_schema(&self) -> &'static str {
        avro_schema!(
            AvroSchemaBuilder::new("MyRecord")
                .int_field("id")
                .string_field("name")
        )
    }

    fn type_properties() -> HashMap<String, Vec<String>> {
        TypePropertiesBuilder::new().build()
    }
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Create client with authentication
    let mut options = KineticaOptions::new();
    options.set_username("admin")
           .set_password("password")
           .set_use_snappy(false);

    let kinetica = Arc::new(
        Kinetica::new(vec!["http://localhost:9191".to_string()], options).await?
    );

    // Simple usage: Create bulk inserter with default options
    let inserter = BulkInserter::<MyRecord>::new(
        kinetica.clone(),
        "my_table".into(),
        50000,  // batch size
        8       // num workers
    ).await?;

    // Or with options:
    let mut bulk_options = BulkInserterOptions::new();
    bulk_options.set_update_on_existing_pk(true);  // Enable upsert mode

    let inserter_with_options = BulkInserter::<MyRecord>::with_options(
        kinetica,
        "my_table".into(),
        50000,
        8,
        Some(bulk_options),
    ).await?;

    // Insert records...

    Ok(())
}