db

type EsDI interface {
	NewSearch() *elastic.Client
	GetIndexByTimeRange(startTime, endTime time.Time, typ string) string
	GetIndexByTime(d time.Time, _type string) string
}

type EsV7Conf struct {
	Hosts []string `yaml:"host"`
	Index string
}

type MongoConf struct {
	Uri       string `yaml:"uri"`
	User      string `yaml:"user"`
	Pass      string `yaml:"pass"`
	DefaultDB string `yaml:"defaul"`
	// contains filtered or unexported fields
}

type MongoDBClient interface {
	GetCoreDB() *mongo.Database
	GetUserDB() *mongo.Database
	WithSession(f func(sc mongo.SessionContext) error) error
	AbortTransaction(sc mongo.SessionContext) error
	CommitTransaction(sc mongo.SessionContext) error
	Close()
}

type MongoDI interface {
	NewMongoDBClient(ctx context.Context, userDB string) (MongoDBClient, error)
	SetAuth(user, pwd string)
	GetUri() string
}

type RedisClient interface {
	Close() error
	Ping() string
	CountKeys() (int, error)
	Get(k string) ([]byte, error)
	Set(k string, v interface{}, exp time.Duration) (string, error)
	Del(k string) (int64, error)
	LPush(k string, v interface{}) (int64, error)
	RPop(k string) ([]byte, error)
	HMGet(key string, field ...string) []interface{}
	HMSet(key string, values ...interface{}) error
	Exists(key string) bool
	Expired(key string, d time.Duration) (bool, error)
}

type RedisConf struct {
	Host string `yaml:"host"`
	Pwd  string `yaml:"pass"`
	DB   int    `yaml:"db"`
}

type RedisDI interface {
	NewRedisClient(ctx context.Context) (RedisClient, error)
}