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

worker-svr/internal/repository/monthlyusage/repository.go

@@ -0,0 +1,51 @@
+package monthlyusage
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/os/gtime"
+
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/dao"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model/do"
+)
+
+// Repository applies the current usage result to the card information table.
+// It deliberately does not write TDengine history for this full-sync flow.
+type Repository struct{}
+
+type UsageResult struct {
+	CardID  int64
+	UsageMB float64
+}
+
+func NewRepository() *Repository {
+	return &Repository{}
+}
+
+func (r *Repository) ApplyBatch(ctx context.Context, results []UsageResult) error {
+	if len(results) == 0 {
+		return nil
+	}
+	cardIDs := make([]int64, 0, len(results))
+	var statement strings.Builder
+	statement.WriteString("CASE " + dao.TSimCardInfo.Columns().Id + " ")
+	for _, result := range results {
+		if result.CardID <= 0 {
+			return fmt.Errorf("monthly usage result has invalid cardId %d", result.CardID)
+		}
+		fmt.Fprintf(&statement, "WHEN %d THEN %f ", result.CardID, result.UsageMB)
+		cardIDs = append(cardIDs, result.CardID)
+	}
+	statement.WriteString("END")
+	_, err := dao.TSimCardInfo.Ctx(ctx).
+		WhereIn(dao.TSimCardInfo.Columns().Id, cardIDs).
+		Data(do.TSimCardInfo{
+			UsedFlowMb:   gdb.Raw(statement.String()),
+			FlowSyncTime: gtime.Now(),
+		}).
+		Update()
+	return err
+}

worker-svr/internal/repository/monthlyusage/repository_test.go

@@ -0,0 +1,45 @@
+package monthlyusage
+
+import (
+	"context"
+	"testing"
+
+	_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+func TestApplyBatchUpdatesDifferentCardsWithOneCaseStatement(t *testing.T) {
+	if err := gdb.SetConfig(gdb.Config{
+		"default": gdb.ConfigGroup{{Type: "sqlite", Link: "sqlite::@file(:memory:)"}},
+	}); err != nil {
+		t.Fatal(err)
+	}
+	ctx := context.Background()
+	if _, err := g.DB().Exec(ctx, `CREATE TABLE t_sim_card_info (
+id INTEGER PRIMARY KEY,
+used_flow_mb REAL,
+flow_sync_time DATETIME
+)`); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := g.DB().Exec(ctx, `INSERT INTO t_sim_card_info(id, used_flow_mb) VALUES (1, 0), (2, 0), (3, 9)`); err != nil {
+		t.Fatal(err)
+	}
+	if err := NewRepository().ApplyBatch(ctx, []UsageResult{
+		{CardID: 1, UsageMB: 1.5},
+		{CardID: 2, UsageMB: 2.75},
+	}); err != nil {
+		t.Fatal(err)
+	}
+	var rows []struct {
+		ID         int64   `orm:"id"`
+		UsedFlowMB float64 `orm:"used_flow_mb"`
+	}
+	if err := g.DB().Model("t_sim_card_info").Fields("id,used_flow_mb").OrderAsc("id").Scan(&rows); err != nil {
+		t.Fatal(err)
+	}
+	if len(rows) != 3 || rows[0].UsedFlowMB != 1.5 || rows[1].UsedFlowMB != 2.75 || rows[2].UsedFlowMB != 9 {
+		t.Fatalf("rows = %+v", rows)
+	}
+}