targets

type Batch interface {
	Len() uint
	Append(data.LoadedPoint)
}

type BatchFactory interface {
	// New returns a new Batch to add Points to
	New() Batch
}

type Benchmark interface {
	// GetDataSource returns the DataSource to use for this Benchmark
	GetDataSource() DataSource

	// GetBatchFactory returns the BatchFactory to use for this Benchmark
	GetBatchFactory() BatchFactory

	// GetPointIndexer returns the PointIndexer to use for this Benchmark
	GetPointIndexer(maxPartitions uint) PointIndexer

	// GetProcessor returns the Processor to use for this Benchmark
	GetProcessor() Processor

	// GetDBCreator returns the DBCreator to use for this Benchmark
	GetDBCreator() DBCreator
}

type ConfigurableBenchmark interface {
	Benchmark
	SetConfig(batchSize uint, workers uint)
}

type ConstantIndexer struct{}

type DBCreator interface {
	// Init should set up any connection or other setup for talking to the DB, but should NOT create any databases
	Init()

	// DBExists checks if a database with the given name currently exists.
	DBExists(dbName string) bool

	// CreateDB creates a database with the given name.
	CreateDB(dbName string) error

	// RemoveOldDB removes an existing database with the given name.
	RemoveOldDB(dbName string) error
}

type DBCreatorCloser interface {
	DBCreator

	// Close cleans up any database connections
	Close()
}

type DBCreatorPost interface {
	DBCreator

	// PostCreateDB does further initialization after the database is created
	PostCreateDB(dbName string) error
}

type DataSource interface {
	NextItem() data.LoadedPoint
	Headers() *common.GeneratedDataHeaders
}

type ImplementedTarget interface {
	Benchmark(targetDB string, dataSourceConfig *source.DataSourceConfig, v *viper.Viper) (Benchmark, error)
	Serializer() serialize.PointSerializer
	// TargetSpecificFlags adds to the supplied flagSet a number of target-specific
	// flags that will be enabled only when executing a command for this specific
	// target database.
	// flagPrefix is a string that should be concatenated with the names of all flags defined here
	// it is needed to prevent namespace collisions and the ability to override properties
	// defined in the yaml config
	TargetSpecificFlags(flagPrefix string, flagSet *pflag.FlagSet)
	TargetName() string
}

type PointIndexer interface {
	// GetIndex returns a partition for the given Point
	GetIndex(data.LoadedPoint) uint
}

type Processor interface {
	// Init does per-worker setup needed before receiving data
	Init(workerNum int, doLoad, hashWorkers bool)
	// ProcessBatch handles a single batch of data
	ProcessBatch(b Batch, doLoad bool) (metricCount, rowCount uint64)
}

type ProcessorCloser interface {
	Processor
	// Close cleans up after a Processor
	Close(doLoad bool)
}