package kafka

import (
	"github.com/Shopify/sarama"
	"live-gateway/pkg/logger"
)

type Consumer struct {
	client     sarama.Client
	topic      string
	consumer   sarama.Consumer
	partitions []int32
}

func NewKafkaConsumer(addr []string, topic string) (*Consumer, error) {
	p := Consumer{}
	p.topic = topic

	config := sarama.NewConfig()
	config.Version = sarama.V3_1_0_0
	config.Consumer.Offsets.Initial = sarama.OffsetNewest

	var err error
	p.client, err = sarama.NewClient(addr, config)
	if err != nil {
		logger.SLog.Error("new kafka client err:", err)
		return nil, err
	}

	p.consumer, err = sarama.NewConsumerFromClient(p.client)
	if err != nil {
		logger.SLog.Error("new kafka consumer err:", err)
		return nil, err
	}
	p.partitions, err = p.consumer.Partitions(topic)
	if err != nil {
		logger.SLog.Errorf("get partitions for topic %s err", topic)
		return nil, err
	}
	return &p, nil
}