package middleware

import (
	"gitee.com/penghengben/devcloud-mini/maudit/apps/log"
	kafka "github.com/segmentio/kafka-go"
)

func NewAuditSenderConfig() *AuditSenderConfig {
	return &AuditSenderConfig{
		Borkers: []string{"192.168.204.11:9092"},
		Topic:   log.DEFAULT_OPERATE_LOG_TOPIC,
	}
}

// consumer
// kafka的配置 由maudit负责定义
type AuditSenderConfig struct {
	Borkers []string `toml:"brokers" env:"MAUDIT_KAFKA_BROKERS" envSeparator:","`
	Topic   string   `toml:"topic" env:"MAUDIT_KAFKA_TOPIC"`
}

func (k *AuditSenderConfig) GetWriter() *kafka.Writer {
	w := &kafka.Writer{
		Addr:                   kafka.TCP(k.Borkers...),
		Topic:                  k.Topic,
		Balancer:               &kafka.LeastBytes{},
		AllowAutoTopicCreation: true,
	}
	return w
}