package service

import (
	"context"
	"errors"
	"github.com/go-kratos/kratos/v2/log"
	"voucher/internal/pkg/mq"
)

func (j *VoucherService) GetOrderCreateConfig() *mq.ConsumerConfig {
	elm, ok := j.bc.RocketMQ.EventMap["order"]
	if !ok {
		return nil
	}

	if !elm.IsOpenConsumer {
		log.Warnf("order MQ is not open")
		return nil
	}

	return &mq.ConsumerConfig{
		TopicName:       elm.Topic,
		GroupName:       elm.Group,
		PerCoroutineCnt: int(elm.PerCoroutineCnt),
	}
}

func (j *VoucherService) OrderCreateConsumer(ctx context.Context, msg *mq.ConsumerMessage) error {

	orderNo := msg.GetShardingKey()
	if orderNo == "" {
		log.Error("orderCreate 消费异常,获取 orderNo 失败")
		return errors.New("orderCreate 消费异常,获取 orderNo 失败")
	}

	if err := j.VoucherBiz.OrderConsume(ctx, orderNo); err != nil {
		log.Errorf("order 消费异常,orderNo:%s,error: %s", orderNo, err.Error())
	}

	return nil
}

func (j *VoucherService) GetOrderQueryConfig() *mq.ConsumerConfig {
	elm, ok := j.bc.RocketMQ.EventMap["query"]
	if !ok {
		return nil
	}

	if !elm.IsOpenConsumer {
		log.Warnf("query MQ is not open")
		return nil
	}

	return &mq.ConsumerConfig{
		TopicName:       elm.Topic,
		GroupName:       elm.Group,
		PerCoroutineCnt: int(elm.PerCoroutineCnt),
	}
}

func (j *VoucherService) OrderQueryConsumer(ctx context.Context, msg *mq.ConsumerMessage) error {

	orderNo := msg.GetShardingKey()
	if orderNo == "" {
		log.Error("orderQuery 消费异常,获取 orderNo 失败")
		return errors.New("orderQuery 消费异常,获取 orderNo 失败")
	}

	if err := j.VoucherBiz.QueryConsume(ctx, orderNo); err != nil {
		log.Errorf("query 消费异常,orderNo:%s,error: %s", orderNo, err.Error())
	}

	return nil
}

func (j *VoucherService) GetOrderNotifyConfig() *mq.ConsumerConfig {
	elm, ok := j.bc.RocketMQ.EventMap["notify"]
	if !ok {
		return nil
	}

	if !elm.IsOpenConsumer {
		log.Warnf("notify MQ is not open")
		return nil
	}

	return &mq.ConsumerConfig{
		TopicName:       elm.Topic,
		GroupName:       elm.Group,
		PerCoroutineCnt: int(elm.PerCoroutineCnt),
	}
}

func (j *VoucherService) OrderNotifyConsumer(ctx context.Context, msg *mq.ConsumerMessage) error {

	orderNo := msg.GetShardingKey()
	if orderNo == "" {
		log.Error("orderNotify 消费异常,获取 orderNo 失败")
		return errors.New("orderNotify 消费异常,获取 orderNo 失败")
	}

	if err := j.VoucherBiz.NotifyConsume(ctx, orderNo); err != nil {
		log.Errorf("notify 消费异常,orderNo:%s,error: %s", orderNo, err.Error())
	}

	return nil
}