offline

type CSVSource struct {
	Reader    *bufio.Reader
	Delimiter string
}

type CreateTableOpt struct {
	TableName string
	Entity    *types.Entity
	Features  types.FeatureList
	IsCDC     bool
}

type ExportOpt struct {
	SnapshotTable string
	CdcTable      *string
	UnixMilli     *int64
	EntityName    string
	Features      types.FeatureList
	Limit         *uint64
}

type ImportOpt struct {
	Entity            *types.Entity
	Features          types.FeatureList
	Header            []string
	Revision          *int64
	SnapshotTableName string
	Source            *CSVSource
	NoPK              bool // TODO: to import cdc data temporarily for testing
}

type JoinOneGroupOpt struct {
	GroupName           string
	Category            types.Category
	Features            types.FeatureList
	RevisionRanges      []*RevisionRange
	Entity              types.Entity
	EntityRowsTableName string
	ValueNames          []string
}

type JoinOpt struct {
	Entity           types.Entity
	EntityRows       <-chan types.EntityRow
	FeatureMap       map[string]types.FeatureList
	RevisionRangeMap map[string][]*RevisionRange
	ValueNames       []string
}

type PushOpt struct {
	GroupID      int
	Revision     int64
	EntityName   string
	FeatureNames []string
	Records      []types.StreamRecord
}

type RevisionRange struct {
	MinRevision   int64
	MaxRevision   int64
	SnapshotTable string
	CdcTable      string
}

type SnapshotOpt struct {
	Group        *types.Group
	Features     types.FeatureList
	Revision     int64
	PrevRevision int64
}

type Store interface {
	Join(ctx context.Context, opt JoinOpt) (*types.JoinResult, error)
	Export(ctx context.Context, opt ExportOpt) (<-chan types.ExportRecord, <-chan error)
	Import(ctx context.Context, opt ImportOpt) (int64, error)
	Push(ctx context.Context, opt PushOpt) error

	CreateTable(ctx context.Context, opt CreateTableOpt) error
	TableSchema(ctx context.Context, tableName string) (*types.DataTableSchema, error)
	Snapshot(ctx context.Context, opt SnapshotOpt) error

	Ping(ctx context.Context) error
	io.Closer
}