62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
|
package consumer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/infraboard/mcube/v2/ioc"
|
||
|
|
"github.com/infraboard/mcube/v2/ioc/config/log"
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
|
||
|
|
ioc_kafka "github.com/infraboard/mcube/v2/ioc/config/kafka"
|
||
|
|
ioc_mongo "github.com/infraboard/mcube/v2/ioc/config/mongo"
|
||
|
|
kafka "github.com/segmentio/kafka-go"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
ioc.Controller().Registry(&consumer{
|
||
|
|
GroupId: "maudit",
|
||
|
|
Topics: []string{"maudit_new"},
|
||
|
|
ctx: context.Background(),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// 业务具体实现
|
||
|
|
type consumer struct {
|
||
|
|
// 继承模版
|
||
|
|
ioc.ObjectImpl
|
||
|
|
|
||
|
|
// 模块子Logger
|
||
|
|
log *zerolog.Logger
|
||
|
|
|
||
|
|
//
|
||
|
|
reader *kafka.Reader
|
||
|
|
// 允许时上下文
|
||
|
|
ctx context.Context
|
||
|
|
|
||
|
|
// 消费组Id
|
||
|
|
GroupId string `toml:"group_id" json:"group_id" yaml:"group_id" env:"GROUP_ID"`
|
||
|
|
// 当前这个消费者 配置的topic
|
||
|
|
Topics []string `toml:"topic" json:"topic" yaml:"topic" env:"TOPIC"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// 对象名称
|
||
|
|
func (i *consumer) Name() string {
|
||
|
|
return "maudit_consumer"
|
||
|
|
}
|
||
|
|
|
||
|
|
// 初始化
|
||
|
|
func (i *consumer) Init() error {
|
||
|
|
// 对象
|
||
|
|
i.log = log.Sub(i.Name())
|
||
|
|
i.log.Debug().Msgf("database: %s", ioc_mongo.Get().Database)
|
||
|
|
i.reader = ioc_kafka.ConsumerGroup(i.GroupId, i.Topics)
|
||
|
|
|
||
|
|
go i.Run(i.ctx)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *consumer) Close(ctx context.Context) error {
|
||
|
|
i.ctx.Done()
|
||
|
|
return nil
|
||
|
|
}
|