mysqlbatch

type BenchmarkConfig struct {
	DB       DBConfig
	Prepare  bool
	PrepareC PrepareConfig
	Pipeline PipelineConfig
	Validate ValidateConfig
}

type BenchmarkResult struct {
	PrepareDuration  time.Duration
	PipelineDuration time.Duration
	ValidateDuration time.Duration
	TotalDuration    time.Duration
}

type DBConfig struct {
	Host     string
	Port     int
	User     string
	Password string
	Database string
	Params   map[string]string
}

type FlowConfig struct {
	Source    FlowSourceConfig    `json:"source"`
	Transform FlowTransformConfig `json:"transform"`
	Sink      FlowSinkConfig      `json:"sink"`
}

type FlowSinkConfig struct {
	Type   string     `json:"type"`
	DB     DBConfig   `json:"db"`
	Config SinkConfig `json:"config"`
}

type FlowSourceConfig struct {
	Type   string       `json:"type"`
	DB     DBConfig     `json:"db"`
	Config SourceConfig `json:"config"`
}

type FlowTransformConfig struct {
	Type       string            `json:"type"`
	PluginPath string            `json:"plugin_path"`
	Reducers   int               `json:"reducers"`
	Workers    int               `json:"workers"`
	InRAM      bool              `json:"in_ram"`
	Port       int               `json:"port"`
	Params     map[string]string `json:"params"`
}

type PipelineConfig struct {
	DB         DBConfig // backward compatibility fallback when SourceDB/SinkDB are not set
	SourceDB   DBConfig
	SinkDB     DBConfig
	Source     SourceConfig
	Sink       SinkConfig
	PluginPath string
	Reducers   int
	Workers    int
	InRAM      bool
	Port       int
}

type PrepareConfig struct {
	SourceTable string
	Rows        int64
	KeyMod      int64
}

type SinkConfig struct {
	TargetTable string
	KeyColumn   string
	ValColumn   string
	InputGlob   string
	Replace     bool
	BatchSize   int
}

type SourceConfig struct {
	Table      string
	PKColumn   string
	KeyColumn  string
	ValColumn  string
	Where      string
	Shards     int
	Parallel   int
	OutputDir  string
	FilePrefix string
}

type ValidateConfig struct {
	SourceTable string
	SourceKey   string
	SourceVal   string
	TargetTable string
	TargetKey   string
	TargetVal   string
}