types

type BackendType string

type CreateEntityOpt struct {
	Name        string
	Length      int
	Description string
}

type CreateFeatureGroupOpt struct {
	Name        string
	EntityName  string
	Description string
}

type CreateFeatureOpt struct {
	FeatureName string
	GroupID     int
	DBValueType string
	Description string
}

type CsvDataSource struct {
	Reader    io.Reader
	Delimiter string
}

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

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

type EntityList []*Entity

type EntityRow struct {
	EntityKey string `db:"entity_key"`
	UnixTime  int64  `db:"unix_time"`
}

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

type Feature struct {
	ID          int    `db:"id"`
	Name        string `db:"name"`
	ValueType   string `db:"value_type"`
	DBValueType string `db:"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   *FeatureGroup
}

type FeatureDataSet map[string][]FeatureKV

type FeatureGroup struct {
	ID       int    `db:"id"`
	Name     string `db:"name"`
	Category string `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 FeatureGroupList []*FeatureGroup

type FeatureKV struct {
	FeatureName  string
	FeatureValue interface{}
}

type FeatureList []*Feature

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

type GetHistoricalFeatureValuesOpt struct {
	FeatureIDs []int
	EntityRows <-chan EntityRow
}

type GetOnlineFeatureValuesOpt struct {
	FeatureNames []string
	EntityKey    string
}

type GetRevisionOpt struct {
	GroupName  *string
	Revision   *int64
	RevisionId *int
}

type ImportBatchFeaturesOpt struct {
	GroupID     int
	Description string
	DataSource  CsvDataSource
	Revision    *int64
}

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

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

type MetadataStoreConfig struct {
	Backend  BackendType  `yaml:"backend"`
	Postgres *PostgresOpt `yaml:"postgres"`
}

type MultiGetOnlineFeatureValuesOpt struct {
	FeatureIDs []int
	EntityKeys []string
}

type OfflineStoreConfig struct {
	Backend  BackendType  `yaml:"backend"`
	Postgres *PostgresOpt `yaml:"postgres"`
}

type OnlineStoreConfig struct {
	Backend  BackendType  `yaml:"backend"`
	Postgres *PostgresOpt `yaml:"postgres"`
	Redis    *RedisOpt    `yaml:"redis"`
}

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 RawFeatureValueRecord struct {
	Record []interface{}
	Error  error
}

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

type Revision struct {
	ID        int    `db:"id"`
	Revision  int64  `db:"revision"`
	DataTable string `db:"data_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   *FeatureGroup
}

type RevisionList []*Revision

type SyncOpt struct {
	RevisionId int
}

type UpdateEntityOpt struct {
	EntityName     string
	NewDescription string
}

type UpdateFeatureGroupOpt struct {
	GroupName        string
	Description      *string
	OnlineRevisionId *int
}

type UpdateFeatureOpt struct {
	FeatureName    string
	NewDescription string
}