types

type BackendType string

type BigQueryOpt struct {
	ProjectID   string `yaml:"project_id"`
	DatasetID   string `yaml:"dataset_id"`
	Credentials string `yaml:"credentials"`
}

type CassandraOpt struct {
	Hosts    []string      `yaml:"hosts"`
	User     string        `yaml:"user"`
	Password string        `yaml:"password"`
	KeySpace string        `yaml:"keyspace"`
	Timeout  time.Duration `yaml:"timeout"`
}

type Category = string

type ChannelExportOpt struct {
	RevisionID   int
	FeatureNames []string
	Limit        *uint64
}

type ChannelJoinOpt struct {
	FeatureFullNames []string
	EntityRows       <-chan EntityRow
	ValueNames       []string
}

type CreateEntityOpt struct {
	EntityName  string
	Description string
}

type CreateFeatureOpt struct {
	FeatureName string
	GroupName   string
	ValueType   ValueType
	Description string
}

type CreateGroupOpt struct {
	GroupName   string
	EntityName  string
	Category    Category
	Description string
}

type CsvFileDataSource struct {
	InputFilePath string
	Delimiter     string
}

type CsvReaderDataSource struct {
	Reader    io.Reader
	Delimiter string
}

type DataSourceType int

type DataTableFieldSchema struct {
	Name      string
	ValueType ValueType
}

type DataTableSchema struct {
	Fields []DataTableFieldSchema
}

type DynamoDBOpt struct {
	Region          string `yaml:"region"`
	EndpointURL     string `yaml:"endpoint-url"`
	AccessKeyID     string `yaml:"access-key-id"`
	SecretAccessKey string `yaml:"secret-access-key"`
	SessionToken    string `yaml:"session-token"`
	Source          string `yaml:"source"`
}

type Entity struct {
	ID   int    `db:"id"`
	Name string `db:"name"`

	Description string    `db:"description"`
	CreateTime  time.Time `db:"create_time"`
	ModifyTime  time.Time `db:"modify_time"`
}

type EntityList []*Entity

type EntityRow struct {
	EntityKey string
	UnixMilli int64
	Values    []string
}

type ExportOpt struct {
	RevisionID     int
	FeatureNames   []string
	Limit          *uint64
	OutputFilePath string
}

type ExportRecord []interface{}

type ExportResult struct {
	Header []string
	Data   <-chan ExportRecord
	// contains filtered or unexported fields
}

type Feature struct {
	ID        int       `db:"id"`
	Name      string    `db:"name"`
	FullName  string    `db:"full_name"`
	ValueType ValueType `db:"value_type"`

	Description string    `db:"description"`
	CreateTime  time.Time `db:"create_time"`
	ModifyTime  time.Time `db:"modify_time"`

	GroupID int `db:"group_id"`
	Group   *Group
}

type FeatureList []*Feature

type FeatureValues struct {
	EntityName       string
	EntityKey        string
	FeatureFullNames []string
	FeatureValueMap  map[string]interface{}
}

type Group struct {
	ID       int      `db:"id"`
	Name     string   `db:"name"`
	Category Category `db:"category"`

	Description string    `db:"description"`
	CreateTime  time.Time `db:"create_time"`
	ModifyTime  time.Time `db:"modify_time"`

	EntityID         int  `db:"entity_id"`
	OnlineRevisionID *int `db:"online_revision_id"`

	Entity *Entity
}

type GroupList []*Group

type ImportOpt struct {
	GroupName   string
	Description string
	Revision    *int64

	DataSourceType      DataSourceType
	CsvFileDataSource   *CsvFileDataSource
	CsvReaderDataSource *CsvReaderDataSource
	TableLinkDataSource *TableLinkDataSource
}

type JoinOpt struct {
	FeatureFullNames []string
	InputFilePath    string
	OutputFilePath   string
}

type JoinResult struct {
	Header []string
	Data   <-chan []interface{}
}

type ListFeatureOpt struct {
	EntityName       *string
	GroupName        *string
	FeatureFullNames *[]string
}

type MetadataStoreConfig struct {
	Backend  BackendType  `yaml:"-"`
	Postgres *PostgresOpt `yaml:"postgres"`
	MySQL    *MySQLOpt    `yaml:"mysql"`
	SQLite   *SQLiteOpt   `yaml:"sqlite"`
	TiDB     *MySQLOpt    `yaml:"tidb"`
}

type MySQLOpt struct {
	Host     string `yaml:"host"`
	Port     string `yaml:"port"`
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Database string `yaml:"database"`
}

type OfflineStoreConfig struct {
	Backend   BackendType   `yaml:"-"`
	Postgres  *PostgresOpt  `yaml:"postgres"`
	MySQL     *MySQLOpt     `yaml:"mysql"`
	Snowflake *SnowflakeOpt `yaml:"snowflake"`
	BigQuery  *BigQueryOpt  `yaml:"bigquery"`
	Redshift  *RedshiftOpt  `yaml:"redshift"`
	SQLite    *SQLiteOpt    `yaml:"sqlite"`
	TiDB      *MySQLOpt     `yaml:"tidb"`
}

type OnlineGetOpt struct {
	FeatureFullNames []string
	EntityKey        string
}

type OnlineMultiGetOpt struct {
	FeatureFullNames []string
	EntityKeys       []string
}

type OnlineStoreConfig struct {
	Backend   BackendType   `yaml:"-"`
	Postgres  *PostgresOpt  `yaml:"postgres"`
	Redis     *RedisOpt     `yaml:"redis"`
	MySQL     *MySQLOpt     `yaml:"mysql"`
	DynamoDB  *DynamoDBOpt  `yaml:"dynamodb"`
	Cassandra *CassandraOpt `yaml:"cassandra"`
	SQLite    *SQLiteOpt    `yaml:"sqlite"`
	TiDB      *MySQLOpt     `yaml:"tidb"`
	TiKV      *TiKVOpt      `yaml:"tikv"`
}

type OomStoreConfig struct {
	MetadataStore MetadataStoreConfig `yaml:"metadata-store"`
	OfflineStore  OfflineStoreConfig  `yaml:"offline-store"`
	OnlineStore   OnlineStoreConfig   `yaml:"online-store"`
}

type PostgresOpt struct {
	Host     string `yaml:"host"`
	Port     string `yaml:"port"`
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Database string `yaml:"database"`
}

type PushOpt struct {
	EntityKey     string
	GroupName     string
	FeatureNames  []string
	FeatureValues []interface{}
}

type RedisOpt struct {
	Host     string `yaml:"host"`
	Port     string `yaml:"port"`
	Password string `yaml:"password"`
	Database int    `yaml:"database"`
}

type RedshiftOpt = PostgresOpt

type Revision struct {
	ID            int    `db:"id"`
	Revision      int64  `db:"revision"`
	SnapshotTable string `db:"snapshot_table"`
	CdcTable      string `db:"cdc_table"`
	Anchored      bool   `db:"anchored"`

	Description string    `db:"description"`
	CreateTime  time.Time `db:"create_time"`
	ModifyTime  time.Time `db:"modify_time"`

	GroupID int `db:"group_id"`
	Group   *Group
}

type RevisionList []*Revision

type SQLiteOpt struct {
	DBFile string `yaml:"db-file"`
}

type SnowflakeOpt struct {
	Account  string `yaml:"account"`
	User     string `yaml:"user"`
	Password string `yaml:"password"`
	Database string `yaml:"database"`
}

type StreamRecord struct {
	GroupID   int
	EntityKey string
	UnixMilli int64
	Values    []interface{}
}

type SyncOpt struct {
	RevisionID int
	PurgeDelay int
}

type TableLinkDataSource struct {
	TableName string
}

type TiKVOpt struct {
	PdAddrs []string `yaml:"pd-addrs"`
}

type UpdateEntityOpt struct {
	EntityName     string
	NewDescription *string
}

type UpdateFeatureOpt struct {
	FeatureFullName string
	NewDescription  *string
}

type UpdateGroupOpt struct {
	GroupName           string
	NewDescription      *string
	NewOnlineRevisionID *int
}

type ValueType int