client

type Client struct {
	plugin.UnimplementedSource
	batchwriter.UnimplementedDeleteRecord
	// contains filtered or unexported fields
}

type Logger struct {
	Base zerolog.Logger
}

type Spec struct {
	// Endpoint for the database. Supported schemes are `wss://` and `ws://`, the default port is `8182`.
	Endpoint string `json:"endpoint" jsonschema:"required,pattern=^wss?://[^\n]+$"`

	// Whether to skip TLS verification. Defaults to `false`. This should be set on a macOS environment when connecting to an AWS Neptune endpoint.
	Insecure bool `json:"insecure" jsonschema:"default=false"`

	// Authentication mode to use. `basic` uses static credentials, `aws` uses AWS IAM authentication.
	AuthMode authMode `json:"auth_mode" jsonschema:"default=none"`

	// Username to connect to the database. Required when `auth_mode` is `basic`.
	Username string `json:"username"`

	// Password to connect to the database. Required when `auth_mode` is `basic`.
	Password string `json:"password"`

	// Number of retries on `ConcurrentModificationException` before giving up for each batch.
	// Retries are exponentially backed off.
	MaxRetries int `json:"max_retries" jsonschema:"minimum=1,default=5"`

	// AWS region to use for AWS IAM authentication. Required when `auth_mode` is `aws`.
	AWSRegion string `json:"aws_region" jsonschema:"example=us-east-1"`

	// AWS Neptune host header to use with AWS IAM authentication. Required when `auth_mode` is `aws`.
	AWSNeptuneHost string `json:"aws_neptune_host" jsonschema:"pattern=^[\\w\\.-]+$,example=my-neptune.cluster.us-east-1.neptune.amazonaws.com"`

	// Maximum number of concurrent connections to the database. Defaults to the number of CPUs.
	MaxConcurrentConnections int `json:"max_concurrent_connections" jsonschema:"minimum=1"`

	// Whether to use all Gremlin types or just a basic subset.
	// Should remain `false` for Amazon Neptune compatibility.
	CompleteTypes bool `json:"complete_types" jsonschema:"default=false"`

	// Number of records to batch together before sending to the database.
	BatchSize int `json:"batch_size" jsonschema:"minimum=1,default=200"`

	// Number of bytes (as Arrow buffer size) to batch together before sending to the database.
	BatchSizeBytes int `json:"batch_size_bytes" jsonschema:"minimum=1,default=4194304"`
}