orm

type BaseShardEntity struct {
	// contains filtered or unexported fields
}

type DBConfig struct {
	User          string            `yaml:"user"`
	Pass          string            `yaml:"pass"`
	URL           string            `yaml:"url"`
	Schema        string            `yaml:"schema"`
	MaxConn       int               `yaml:"maxConn"`
	MaxIdle       int               `yaml:"maxIdle"`
	MaxTimeSecond int               `yaml:"maxTimeSecond"`
	Charset       string            `yaml:"charset"`
	Ext           map[string]string `yaml:"ext"`
}

type DBConfigurer interface {
	c.Configurer
	DBConfig() *DBConfig
}

type DBError struct {
	Msg string
	Err error
}

type DBService interface {
	c.Initable
	OpCreator
}

type DBShardConfig struct {
	Shards  map[string]*DBConfig `yaml:"shards"`
	Default string               `yaml:"default"`
}

type DBShardConfigurer interface {
	c.Configurer
	DBShardConfig() *DBShardConfig
}

type Entity interface {
	TableName() string
}

type EntityShardConfig struct {
	// pkgPath -> entity name -> rules
	Entities map[string]map[string][]*EntityShardRuleConfig `yaml:"entities"`
	// contains filtered or unexported fields
}

type EntityShardConfigurer interface {
	c.Configurer
	EntityShardConfig() *EntityShardConfig
}

type EntityShardRuleConfig struct {
	Name       string   `yaml:"name"`        //名称
	DBShard    *OneRule `yaml:"db_shard"`    //数据库实例的配置
	TableShard *OneRule `yaml:"table_shard"` //数据库表的配置
	Default    bool     `yaml:"default"`     //是否是默认规则
	// contains filtered or unexported fields
}

type EntitySlice []Entity

type HashRule struct {
	Count      int64  `yaml:"count"`       //hash的个数
	NamePrefix string `yaml:"name_prefix"` //名称的前缀
	FieldName  string `yaml:"field_name"`  //hash取值的字段名
}

type Meta interface {
	Name() string
	Type() reflect.Type
	FieldValue(entity Entity, name string) (val interface{}, err error)
}

type NamedRule struct {
	Name string `yaml:"name"`
}

type NullTime = sql.NullTime

type NumRangeRule struct {
	FieldName   string `yaml:"field_name"`   //分片取值的字段名
	DefaultName string `yaml:"default_name"` //默认名称
	Ranges      []*struct {
		Begin int64  `yaml:"begin"`
		End   int64  `yaml:"end"`
		Name  string `yaml:"name"`
	} `yaml:"ranges"`
}

type OneRule struct {
	Hash     *HashRule     `yaml:"hash"`
	Named    *NamedRule    `yaml:"named"`
	NumRange *NumRangeRule `yaml:"num_range"`
	// contains filtered or unexported fields
}

type Op struct {
	// contains filtered or unexported fields
}

type OpCreator interface {
	//NewOp create a new Op
	NewOp() (*Op, error)
}

type OpTxFunc func(tx *sql.Tx) (interface{}, error)

type Pool struct {
	// contains filtered or unexported fields
}

type PoolFunc func(config *DBConfig) (pool *Pool, err error)

type ReplColumn struct {
	Repl     string
	ParamVal interface{}
}

type ShardDBService interface {
	DBService
	// NewOpByShardName create op by shard name
	NewOpByShardName(name string) (op *Op, err error)
	// NewOpByEntity create Op for entity with rule name,if rule name is empty use default rule
	NewOpByEntity(entity Entity, ruleName string) (op *Op, err error)
	// contains filtered or unexported methods
}

type ShardEntity interface {
	Entity
	//TableShardFunc table分片函数
	TableShardFunc() ShardHandler
	//SetTableShardFunc table设置分片函数
	SetTableShardFunc(ShardHandler)
}

type ShardHandler func() (shardName string, err error)

type ShardPolicy string

type ShardRule interface {
	c.Configurer
	// Policy 返回策略名称
	Policy() ShardPolicy
	// Shard 计算分片的名称
	Shard(val interface{}) (shardName string, err error)
	// ShardFieldName 用于分片的字段名称
	ShardFieldName() string
}

type SimpleDBService struct {
	Config DBConfigurer `inject:"_"`
	// contains filtered or unexported fields
}

type SimpleShardDBService struct {
	DBShardConfig     DBShardConfigurer     `inject:"_"`
	EntityShardConfig EntityShardConfigurer `inject:"_,optional"`
	// contains filtered or unexported fields
}