package rabbitmq

import (
	"context"

	"github.com/gogf/gf/v2/frame/g"

	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
	mq "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/library/rabbitmq"
	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
)

func (c *Consumers) initTrafficUsageMonthlyRetry(ctx context.Context) error {
	if !mq.Enabled(ctx) {
		g.Log().Info(ctx, "traffic usage monthly retry consumer skipped, RabbitMQ disabled")
		return nil
	}
	if err := mq.ConsumeJSON[model.TrafficUsageMonthlyRetryMessage](
		ctx,
		consts.QueueTrafficUsageMonthlyRetry,
		"iotlcm-traffic-usage-monthly",
		c.handleTrafficUsageMonthlyRetry,
	); err != nil {
		return err
	}
	g.Log().Infof(ctx, "traffic usage monthly retry consumer started, queue:%s", consts.QueueTrafficUsageMonthlyRetry)
	return nil
}

func (c *Consumers) handleTrafficUsageMonthlyRetry(ctx context.Context, message model.TrafficUsageMonthlyRetryMessage) error {
	err := validateTrafficUsageIdentity(message.SimKey, message.Card)
	if err == nil {
		err = c.trafficUsage.RetryMonthly(ctx, message)
	}
	if err == nil {
		return nil
	}
	message.Attempt++
	message.LastError = err.Error()
	permanent := trafficUsageFailureSnapshot(err, &message.UsageMb)
	return publishTrafficUsageFailure(ctx, consts.QueueTrafficUsageMonthlyRetry, message, message.Attempt, permanent)
}