workload

type Constructor func(WorkloadConfig) (Workload, error)

type WorkerPool struct {
	Context context.Context
	Config  *config.Config
	Schema  schema.Schema

	Pool *pool.Pool
	Jobs []pool.Job

	MetricsRegistry metrics.Registry
	// contains filtered or unexported fields
}

type WorkerPoolLoadJob struct {
	Context         context.Context
	Client          *spanner.Client
	TableName       string
	RowCount        int
	Statement       string
	GeneratorMap    data.GeneratorMap
	Batch           bool
	BatchSize       int
	WaitGroup       *sync.WaitGroup
	MetricsRegistry metrics.Registry
}

type WorkerPoolRunJob struct {
	Context           context.Context
	Client            *spanner.Client
	TableName         string
	ReadMap           data.GeneratorMap // Generate data for point reads
	WriteMap          data.GeneratorMap // Generate data for writes
	OperationSelector selector.Selector // Weghted choice selector (read or write)
	WaitGroup         *sync.WaitGroup
	StaleReads        bool // Should we perform stale reads? If not, strong reads
	Staleness         time.Duration
	Operations        int // How many operations to perform

	Table         schema.Table
	ReadGenerator *sample.SampleGenerator

	MetricsRegistry metrics.Registry
	// contains filtered or unexported fields
}

type Workload interface {
	// Initialize is called once before any operations can proceed
	Initialize() error
	Load(string) error
	Run(string) error
	Stop() error
}

type WorkloadConfig struct {
	Context        context.Context
	Config         *config.Config
	Schema         schema.Schema
	MetricRegistry metrics.Registry
}