cmd

type DB interface {

	// Read reads a record from the database and returns a map of each field/value pair.
	// table: The name of the table.
	// key: The record key of the record to read.
	// readPreference:  route read operations to the primaries (0) or to the secondaries/replicas (1)
	// fields: The list of fields to read, nil|empty for reading all.
	Read(ctx context.Context, table string, key string, readPreference int64, fields []string) (map[string][]byte, error)

	// Insert inserts a record in the database. Any field/value pairs will be written into the
	// database.
	// table: The name of the table.
	// key: The record key of the record to insert.
	// numReplicas: number of replicas that acknowledged the insert.
	// numReplicasTimeout: timeout in milliseconds to ack.
	// values: A map of field/value pairs to insert in the record.
	Insert(ctx context.Context, table string, key string, numReplicas int64, numReplicasTimeout int64, values map[string][]byte) error
}

type DBCreator interface {
	Create(p *properties.Properties) (DB, error)
}

type Datapoint struct {
	Error   bool
	Latency uint64
	Type    int64
}

type TestResult struct {
	Metadata   string `json:"Metadata"`
	Clients    uint64 `json:"Clients"`
	MaxRps     uint64 `json:"MaxRps"`
	RandomSeed int64  `json:"RandomSeed"`

	StartTime      int64 `json:"StartTime"`
	EndTime        int64 `json:"EndTime"`
	DurationMillis int64 `json:"DurationMillis"`

	// Populated after benchmark
	// Benchmark Totals
	Totals map[string]float64 `json:"Totals"`

	// Overall Rates
	OverallCommandRate []float64 `json:"OverallCommandRate"`

	EncodedWriteHistogram []byte `json:"EncodedWriteHistogram"`
	EncodedReadHistogram  []byte `json:"EncodedReadHistogram"`

	// Overall Client Quantiles
	OverallClientLatencies map[string]map[string]float64 `json:"OverallClientLatencies"`
	// contains filtered or unexported fields
}

type TestSpec struct {
	Rps                      uint64
	Clients                  uint64
	Duration                 time.Duration
	StartAtUnix              int64
	ReadPreference           int64
	NumReplicas              int64
	NumReplicasTimeoutMillis int64
	Addr                     string
	Password                 string
	Username                 string
	ReadProportion           float64
	InsertProportion         float64
	RecordCount              uint64
	DataSize                 uint64
}