feat kafka

This commit is contained in:
2026-01-27 17:14:05 +08:00
parent 7271181501
commit a93b5fd1aa
3 changed files with 10 additions and 7 deletions

View File

@@ -13,12 +13,12 @@ type Client struct {
consumer sarama.ConsumerGroup
}
func Init(cfg *config.KafkaConfig) error {
func Init(cfg *config.KafkaConfig, appName string) error {
producer, err := getAsyncProducer(cfg)
if err != nil {
return err
}
consumer, err := getConsumer(cfg)
consumer, err := getConsumer(cfg, appName)
if err != nil {
return err
}
@@ -40,13 +40,13 @@ func getAsyncProducer(cfg *config.KafkaConfig) (sarama.AsyncProducer, error) {
return sarama.NewAsyncProducer(cfg.Brokers, conf)
}
func getConsumer(cfg *config.KafkaConfig) (sarama.ConsumerGroup, error) {
func getConsumer(cfg *config.KafkaConfig, appName string) (sarama.ConsumerGroup, error) {
conf := sarama.NewConfig()
conf.Consumer.Return.Errors = true
conf.Consumer.Group.Session.Timeout = 10 * time.Second
conf.Consumer.Offsets.AutoCommit.Enable = false
conf.Consumer.Offsets.Initial = sarama.OffsetOldest
return sarama.NewConsumerGroup(cfg.Brokers, cfg.GroupID, conf)
return sarama.NewConsumerGroup(cfg.Brokers, appName, conf)
}
func Close() error {