chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

worker-svr/internal/consumer/rabbitmq/traffic_usage_daily_start.go

@@ -0,0 +1,49 @@
+package rabbitmq
+
+import (
+	"context"
+	"errors"
+
+	"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"
+	dailyusageproducer "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/producer/dailyusage"
+)
+
+func (c *Consumers) initTrafficUsageDailyStart(ctx context.Context) error {
+	if !mq.Enabled(ctx) {
+		g.Log().Info(ctx, "traffic usage daily start consumer skipped, RabbitMQ disabled")
+		return nil
+	}
+	if err := mq.ConsumeJSON[model.TrafficUsageDailyStartMessage](
+		ctx,
+		consts.QueueTrafficUsageDailyStart,
+		"iotlcm-traffic-usage-daily-start",
+		c.handleTrafficUsageDailyStart,
+	); err != nil {
+		return err
+	}
+	g.Log().Infof(ctx, "traffic usage daily start consumer started, queue:%s", consts.QueueTrafficUsageDailyStart)
+	return nil
+}
+
+func (c *Consumers) handleTrafficUsageDailyStart(ctx context.Context, message model.TrafficUsageDailyStartMessage) error {
+	if message.RunID == "" || message.QueryDate == "" {
+		g.Log().Errorf(ctx, "traffic usage daily start message dropped, invalid message:%+v", message)
+		return nil
+	}
+	err := c.trafficUsageDailyStart.Produce(ctx, message)
+	if errors.Is(err, dailyusageproducer.ErrAlreadyRunning) {
+		g.Log().Warningf(ctx, "traffic usage daily start skipped because previous run is active, runId:%s", message.RunID)
+		return nil
+	}
+	// A run that already finished must not be resurrected by a redelivered
+	// message. Ack it instead of retrying forever.
+	if errors.Is(err, dailyusageproducer.ErrRunFinished) {
+		g.Log().Warningf(ctx, "traffic usage daily start skipped because run already finished, runId:%s", message.RunID)
+		return nil
+	}
+	return err
+}