MiFaZhan
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
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)
|
|
}
|