infra

type Coco struct{}

type Elastic struct {
	*elasticsearch.Client
	Coco
}

type ElasticConfig struct {
	Addresses []string
	Username  string
	Password  string
}

type Infra interface {
	CocoInfra() // 用于标记特定的接口实现方式
}

type KafkaMessage = kafka.Message

type KafkaReader struct {
	*kafka.Reader
	Coco
}

type KafkaReaderConfig struct {
	Brokers     []string
	Topic       string
	GroupID     string
	GroupTopics []string
}

type KafkaWriter struct {
	*kafka.Writer
	Coco
}

type KafkaWriterConfig struct {
	Brokers      []string
	Topic        string
	ClientID     string
	DialTimeout  time.Duration
	BatchTimeout time.Duration
	Async        bool
}

type Mongo struct {
	*mongo.Client
	Coco
}

type MongoConfig struct {
	URL             string
	ConnectTimeout  time.Duration // 初始化连接超时时间
	SocketTimeout   time.Duration
	MaxConnIdleTime time.Duration
	MaxPoolSize     uint64
}

type Postgres struct {
	*pg.DB
	Coco
}

type PostgresConfig struct {
	Addr     string
	User     string
	Password string
	Database string
}

type Redis struct {
	*redis.Client
	Coco
}

type RedisConfig struct {
	URL            string
	ConnectTimeout time.Duration // 初始化连接超时时间
}

type Sentry struct {
	Coco
}

type TransactionFn = func(sessionContext mongo.SessionContext) (interface{}, error)