gdb

type DB struct {
	ObjectDB // support redis db or mongo db(TODO)
	RedisClient
}

type Generic interface {
	IsErrNil(err error) bool
	Exists(ctx context.Context, key string) (bool, error)
	TTL(ctx context.Context, key string) (time.Duration, error)
	Del(ctx context.Context, key string) (uint32, error)
	Expire(ctx context.Context, key string, expiration time.Duration) (bool, error)
	ExpireAt(ctx context.Context, key string, t time.Time) (bool, error)
	ExpireAtTS(ctx context.Context, key string, ts int64) (bool, error)
}

type Hash interface {
	HGet(ctx context.Context, key, field string) (string, error)
	HSet(ctx context.Context, key string, values ...any) error
	HMSet(ctx context.Context, key string, values ...any) (bool, error)
	HMGet(ctx context.Context, key string, fields ...string) ([]any, error)
	HGetAll(ctx context.Context, key string) (map[string]string, error)
	HDel(ctx context.Context, key string, fields ...string) (int64, error)
	HLen(ctx context.Context, key string) (int64, error)
}

type JsonMarshaller struct{}

type KOMapping interface {
	Mapping() map[string]string
}

type Marshaller interface {
	Marshal(interface{}) ([]byte, error)
	Unmarshal([]byte, interface{}) error
}

type Mode uint8

type ObjectDB interface {
	SetKOMapping(map[string]string)
	CheckKeyObjMatch(key string, obj any)
	// GetObject get data from db of the key, and unmarshal into obj.
	// obj should be a struct point, and should be memory allocated
	GetObject(ctx context.Context, key string, obj any) error
	// SetObject set data into db by the key, the data is unmarshalled from obj.
	// obj should be a struct point, and not nil.
	SetObject(ctx context.Context, key string, obj any) error
	// SetObjectEX set data into db by the key with expiration, the data is unmarshalled from obj.
	// obj should be a struct point, and not nil.
	SetObjectEX(ctx context.Context, key string, obj any, expiration time.Duration) error
	// GetObjects get datas from db of all keys, and unmarshal into objs.
	// objs should be a slice of struct points, and slice should be memory allocated.
	GetObjects(ctx context.Context, keys []string, objs any) error
	// SetObjects set datas into db by keys, the datas is marshalled from objs.
	// objs should be a slice.
	SetObjects(ctx context.Context, keys []string, objs any) error
	// SetObjectsEX set datas into db by keys with expiration, the datas is unmarshalled from objs.
	// objs should be a slice.
	SetObjectsEX(ctx context.Context, keys []string, objs any, expiration time.Duration) error

	// HSetObjects set filed value pairs into db by the key, value will be marshalled before set.
	HSetObjects(ctx context.Context, key string, values ...any) error
	// HGetObject get data from db with the key and the field, and unmarshal into obj.
	// obj should be a struct point, and should be memory allocated
	HGetObject(ctx context.Context, key string, field string, obj any) error
	// HMGetObjects get datas from db with the key and the fields, and unmarshall into objs.
	// objs should be a slice of struct points, and the slice should be memory allocated.
	HMGetObjects(ctx context.Context, key string, fields []string, objs any) error
	// HGetAllObjects get datas from db with the key and the fields, and unmarshall into objs,
	// field_name will be set into fields, objs should be a point of a slice of same struct or struct points.
	HGetAllObjects(ctx context.Context, key string, fields *[]string, objs any) error

	// ZAddObjects add score and member pairs into db by the key. score should be a number,
	// member will be marshalled before set.
	ZAddObjects(ctx context.Context, key string, values ...any) (int64, error)
	// ZRangeObjects ZRange members from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct points.
	ZRangeObjects(ctx context.Context, key string, start, stop int64, objs any) error
	// ZRangeObjectsByScore ZRange members from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct points.
	ZRangeObjectsByScore(ctx context.Context, key string, min, max string, objs any) error
	// ZRangeObjectsWithScores ZRange members with scores from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct or struct points.
	ZRangeObjectsWithScores(ctx context.Context, key string, start, stop int64, objs any) (scores []float64, err error)
	// ZRangeObjectsByScoreWithScores ZRange members with scores from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct or struct points.
	ZRangeObjectsByScoreWithScores(ctx context.Context, key string, min, max string, objs any) (scores []float64, err error)
	// ZRevRangeObjects ZRevRange members from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct points.
	ZRevRangeObjects(ctx context.Context, key string, start, stop int64, objs any) error
	// ZRevRangeObjectsByScore ZRevRangeByScore members from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct points.
	ZRevRangeObjectsByScore(ctx context.Context, key string, min, max string, objs any) error
	// ZRevRangeObjectsWithScores ZRevRange members with scores from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct or struct points.
	ZRevRangeObjectsWithScores(ctx context.Context, key string, start, stop int64, objs any) (scores []float64, err error)
	// ZRevRangeObjectsByScoreWithScores ZRevRangeByScore members with scores from zset of the key, and unmarshall members into objs.
	// objs should be a point of a slice of struct or struct points.
	ZRevRangeObjectsByScoreWithScores(ctx context.Context, key string, min, max string, objs any) (scores []float64, err error)
	ZRankObject(ctx context.Context, key string, member any) (int64, error)
	ZRevRankObject(ctx context.Context, key string, member any) (int64, error)
	ZScoreObject(ctx context.Context, key string, member any) (float64, error)
	ZRemObjects(ctx context.Context, key string, members ...any) (int64, error)
}

type Pipeliner interface {
	PipelinerObject
	Exec(ctx context.Context) ([]redis.Cmder, error)
	// pipeline zset
	ZRemRangeByScore(ctx context.Context, key string, min, max float64) error
}

type PipelinerObject interface {
	GetObject(ctx context.Context, key string, obj any) error
	HSetObjects(ctx context.Context, key string, values ...any) error
	ZAddObjects(ctx context.Context, key string, values ...any) error
	ZRemObjects(ctx context.Context, key string, values ...any) error
}

type ProtoCompressMarshaller struct{}

type ProtoMarshaller struct{}

type RedisClient interface {
	io.Closer
	Generic
	String
	Hash
	SortedSet
	ObjectDB
	Pipeline() Pipeliner
	TxPipeline() Pipeliner
}

type RedisClientOption struct {
	Mode                Mode
	Addr                string
	Password            string
	Db                  int
	PoolSize            int
	Addrs               map[string]string
	ClusterAddrs        []string
	ClusterMaxRedirects int
	ClusterReadOnly     bool
	DialTimeout         time.Duration
	ReadTimeout         time.Duration
	WriteTimeout        time.Duration
	Marshaller          gmarshaller.Marshaller
	Hooks               []redis.Hook
}

type SelfMarshaller interface {
	Marshal() ([]byte, error)
	Unmarshal([]byte) error
}

type SortedSet interface {
	ZAdd(ctx context.Context, key string, values ...any) (int64, error)
	ZCard(ctx context.Context, key string) (int64, error)
	ZCount(ctx context.Context, key string, min, max string) (int64, error)
	ZIncrBy(ctx context.Context, key string, increment float64, member any) (float64, error)
	ZRange(ctx context.Context, key string, start, stop int64) ([]string, error)
	ZRangeByScore(ctx context.Context, key string, min, max string) ([]string, error)
	ZRangeWithScores(ctx context.Context, key string, start, stop int64) ([]redis.Z, error)
	ZRangeByScoreWithScores(ctx context.Context, key string, min, max string) ([]redis.Z, error)
	ZRevRange(ctx context.Context, key string, start, stop int64) ([]string, error)
	ZRevRangeByScore(ctx context.Context, key string, min, max string) ([]string, error)
	ZRevRangeWithScores(ctx context.Context, key string, start, stop int64) ([]redis.Z, error)
	ZRevRangeByScoreWithScores(ctx context.Context, key string, min, max string) ([]redis.Z, error)
	ZRank(ctx context.Context, key string, member any) (int64, error)
	ZRevRank(ctx context.Context, key string, member any) (int64, error)
	ZRem(ctx context.Context, key string, members ...any) (int64, error)
	ZRemRangeByRank(ctx context.Context, key string, start, stop int64) (int64, error)
	ZRemRangeByScore(ctx context.Context, key string, min, max string) (int64, error)
	ZScore(ctx context.Context, key string, member any) (float64, error)
}

type String interface {
	SetNX(ctx context.Context, key string, value any) (bool, error)
	SetEXNX(ctx context.Context, key string, value any, expiration time.Duration) (bool, error)
	Set(ctx context.Context, key string, value any) error
	SetEX(ctx context.Context, key string, value any, expiration time.Duration) error
	Get(ctx context.Context, key string) (string, error)
	Incr(ctx context.Context, key string) (int64, error)
	IncrBy(ctx context.Context, key string, value int64) (int64, error)
	BatchGet(ctx context.Context, keys []string) ([]*redis.StringCmd, error)
	BatchSet(ctx context.Context, keys []string, values []any, expiration time.Duration) error
	BatchDel(ctx context.Context, keys []string) error
}

type TraceHook struct {
	Tracer     opentracing.Tracer
	Instance   string
	RedisMode  string
	Marshaller string
}