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

2026-11-02 13:28:03 +08:00
parent 6de2e503cc 96f0f4eacd
commit 2923c78755
422 changed files with 80993 additions and 0 deletions
@@ -0,0 +1,122 @@
+package monthlyusage
+
+import (
+	"context"
+	"reflect"
+	"sort"
+	"strings"
+	"testing"
+
+	_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
+	"github.com/gogf/gf/v2/database/gdb"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/dao"
+)
+
+// SQLite runs the SAME predicate and ordering with only the two MySQL hints
+// removed. This checks full eligibility/TaskID parity, not MySQL access plans.
+// Production EXPLAIN must still validate the unmodified channelPageSQL.
+type sqliteChannelScanner struct{ DatabaseScanner }
+
+func (s *sqliteChannelScanner) ScanChannel(ctx context.Context, key string, last int64, size int) ([]Card, error) {
+	query := strings.ReplaceAll(channelPageSQL, "FORCE INDEX (idx_card_chan_key)", "")
+	query = strings.ReplaceAll(query, "STRAIGHT_JOIN", "INNER JOIN")
+	var rows []struct {
+		ID      int64  `orm:"id"`
+		ICCID   string `orm:"iccid"`
+		MSISDN  string `orm:"msisdn"`
+		IMSI    string `orm:"imsi"`
+		ChanKey string `orm:"chan_key"`
+	}
+	if err := dao.TSimCardInfo.DB().Raw(query, key, last, size).Ctx(ctx).Scan(&rows); err != nil {
+		return nil, err
+	}
+	cards := make([]Card, 0, len(rows))
+	for _, row := range rows {
+		cards = append(cards, Card{ID: row.ID, ICCID: row.ICCID, MSISDN: row.MSISDN, IMSI: row.IMSI, ChanKey: row.ChanKey})
+	}
+	return cards, nil
+}
+
+func TestDatabaseQualificationAndCompleteTaskIDParity(t *testing.T) {
+	// No application DB/config: isolated SQLite file and loopback Redis only.
+	if err := gdb.SetConfig(gdb.Config{"default": gdb.ConfigGroup{{Type: "sqlite", Link: "sqlite::@file(" + t.TempDir() + "/cards.db)"}}}); err != nil {
+		t.Fatal(err)
+	}
+	ctx := contextForTest(t)
+	db := dao.TSimCardInfo.DB()
+	t.Cleanup(func() { _ = db.Close(context.Background()) })
+	for _, sql := range []string{
+		`CREATE TABLE sys_dept (dept_id INTEGER PRIMARY KEY, tenant_id TEXT, deleted_at DATETIME)`,
+		`CREATE TABLE t_sim_card_info (id INTEGER PRIMARY KEY, iccid TEXT, msisdn TEXT, imsi TEXT, chan_key TEXT, current_package_code TEXT, dept_id INTEGER, tenant_id TEXT)`,
+		`CREATE INDEX idx_card_chan_key ON t_sim_card_info(chan_key)`,
+		`CREATE TABLE t_sim_channels (chan_key TEXT, temp_key TEXT, sync_enabled INTEGER, operator INTEGER)`,
+		`INSERT INTO sys_dept VALUES (1,'t1',NULL),(2,'t2',NULL),(3,'t1','2026-01-01')`,
+		`INSERT INTO t_sim_channels VALUES ('a','sim_yd_v5',0,1),('b','sim_dx_v1',0,2),('disabled','sim_yd_v5',1,1),('no-template','',0,1)`,
+		`INSERT INTO t_sim_card_info VALUES
+		(1,'i1','m1','s1','a','pkg',1,'t1'),
+		(2,'i2','m2','s2','b','pkg',2,'t2'),
+		(3,'i3','','','a','pkg',1,'t1'),
+		(4,'','','','a','pkg',1,'t1'),
+		(5,'i5','','','a','',1,'t1'),
+		(6,'i6','','','a','pkg',1,'wrong-tenant'),
+		(7,'i7','','','a','pkg',3,'t1'),
+		(8,'i8','','','b','pkg',99,'t2'),
+		(9,'i9','','','disabled','pkg',1,'t1'),
+		(10,'i10','','','no-template','pkg',1,'t1'),
+		(11,'i11','','','unknown','pkg',1,'t1'),
+		(12,'i12','','','','pkg',1,'t1'),
+		(13,'i13','','','b','pkg',2,'t2')`,
+	} {
+		if _, err := db.Exec(ctx, sql); err != nil {
+			t.Fatal(err)
+		}
+	}
+	store := testStore(t)
+	for _, scope := range [][]string{nil, {"a"}, {"b", "disabled", "no-template", "unknown"}, {"unknown"}} {
+		var sets [][]string
+		for _, mode := range []string{"serial", "channel"} {
+			scanner := &sqliteChannelScanner{}
+			p := testProducer(t, scanner, store, Config{PageSize: 2})
+			run := "db-" + mode
+			if err := p.Produce(ctx, scopedMessage(run, scope...)); err != nil {
+				t.Fatal(err)
+			}
+			ids := taskIDs(t, store, run)
+			for i, id := range ids {
+				ids[i] = strings.TrimPrefix(id, run+":")
+			}
+			sort.Strings(ids)
+			sets = append(sets, ids)
+			stateKey, _ := store.TaskStateKey(run)
+			metaKey, _ := store.RunMetaKey(run)
+			chanKey, _ := store.RunChannelsKey(run)
+			_, _ = store.RedisClient().Do(ctx, "DEL", stateKey, metaKey, chanKey)
+			for _, key := range []string{"a", "b"} {
+				activeKey, _ := store.RoundActiveKey(run, 1, key)
+				_, _ = store.RedisClient().Do(ctx, "DEL", activeKey)
+			}
+			_, _ = store.RedisClient().Do(ctx, "SREM", store.ActiveRunsKey(), run)
+		}
+		if !reflect.DeepEqual(sets[0], sets[1]) {
+			t.Fatalf("scope=%v serial=%v channel=%v", scope, sets[0], sets[1])
+		}
+		if scope == nil {
+			want := []string{operation + ":1", operation + ":13", operation + ":2", operation + ":3"}
+			sort.Strings(want)
+			if !reflect.DeepEqual(sets[0], want) {
+				t.Fatalf("qualification set=%v want=%v", sets[0], want)
+			}
+		}
+	}
+}
+
+func TestChannelSQLShapeAndParameters(t *testing.T) {
+	for _, term := range []string{"FORCE INDEX (idx_card_chan_key)", "STRAIGHT_JOIN sys_dept", "c.chan_key = ?", "c.id > ?", "c.iccid <> ''", "c.current_package_code <> ''", "d.deleted_at IS NULL", "c.tenant_id = d.tenant_id", "ORDER BY c.id ASC", "LIMIT ?"} {
+		if !strings.Contains(channelPageSQL, term) {
+			t.Fatalf("missing %q", term)
+		}
+	}
+	if strings.Contains(channelPageSQL, "<=") || strings.Count(channelPageSQL, "?") != 3 {
+		t.Fatal("SQL contains upper bound or changed bind parameter count")
+	}
+}