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,78 @@
+package zset
+
+import (
+	"context"
+	"fmt"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/fullsync/supply"
+	"testing"
+	"time"
+)
+
+func TestSupplyBatchAndLeaseView(t *testing.T) {
+	s := newLockStore(t)
+	ctx := context.Background()
+	run := "supply-test"
+	if err := s.CreateRun(ctx, RunMeta{RunID: run, Status: "producing", CurrentRound: 1}); err != nil {
+		t.Fatal(err)
+	}
+	keys := make([]string, 130)
+	now := time.Now()
+	for i := range keys {
+		keys[i] = fmt.Sprintf("channel-%03d", i)
+	}
+	a, _ := s.RoundActiveKey(run, 1, keys[0])
+	l, _ := s.RoundLeaseKey(run, 1, keys[1])
+	if _, err := s.redis.Do(ctx, "ZADD", a, now.UnixMilli(), "ready", now.Add(time.Hour).UnixMilli(), "future"); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := s.redis.Do(ctx, "ZADD", l, now.Add(-time.Hour).UnixMilli(), "expired"); err != nil {
+		t.Fatal(err)
+	}
+	view, err := s.ReadSupply(ctx, run, keys, now)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(view) != 130 || view[keys[0]].Ready != 1 || view[keys[1]].Leases != 1 || !view[keys[129]].Eligible {
+		t.Fatalf("bad view %+v", view)
+	}
+	meta, _ := s.RunMetaKey(run)
+	s.redis.Do(ctx, "HSET", meta, "blocked:"+keys[2], 1, "currentRound", 2)
+	view, err = s.ReadSupply(ctx, run, keys, now)
+	if err != nil {
+		t.Fatal(err)
+	}
+	for _, v := range view {
+		if v.Eligible {
+			t.Fatal("retry round eligible")
+		}
+	}
+}
+func TestSupplyOwnerAndTerminalGuard(t *testing.T) {
+	s := newLockStore(t)
+	ctx := context.Background()
+	run := "supply-guard"
+	s.CreateRun(ctx, RunMeta{RunID: run, Status: "producing", CurrentRound: 1})
+	lock := "flow:producer:active"
+	s.AcquireLockOrTakeover(ctx, lock, "old", run, time.Minute)
+	encoded, _ := supply.New("old", time.Now(), []string{"a"}).Encode()
+	ok, err := s.SaveSupply(ctx, run, lock, "old", encoded)
+	if err != nil || !ok {
+		t.Fatalf("save %v %v", ok, err)
+	}
+	s.AcquireLockOrTakeover(ctx, lock, "new", run, time.Minute)
+	ok, err = s.SaveSupply(ctx, run, lock, "old", encoded)
+	if err != nil || ok {
+		t.Fatalf("old owner save %v %v", ok, err)
+	}
+	meta, _ := s.RunMetaKey(run)
+	s.redis.Do(ctx, "HSET", meta, "status", "completed")
+	ok, err = s.SaveSupply(ctx, run, lock, "new", encoded)
+	if err != nil || ok {
+		t.Fatalf("terminal save %v %v", ok, err)
+	}
+	s.redis.Do(ctx, "HSET", meta, "supplyObservation", "{broken")
+	m, err := s.GetRunMeta(ctx, run)
+	if err != nil || m.SupplyObservation == nil || !m.SupplyObservation.Partial {
+		t.Fatalf("invalid telemetry poisoned business %+v %v", m, err)
+	}
+}