chore: 并入 iotlcm-admin-svr 历史到 admin-svr/（原提交 49818c6）

2026-11-02 13:27:59 +08:00
parent 976980a3f5 49818c6dbe
commit 193f7393c0
1250 changed files with 154111 additions and 0 deletions
@@ -0,0 +1,126 @@
+package trafficusage
+
+import (
+	"context"
+	"database/sql"
+	"database/sql/driver"
+	"errors"
+	"io"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcfg"
+
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/tdengine"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
+)
+
+type routingDriver struct{ statements []string }
+type routingConnection struct{ owner *routingDriver }
+type routingRows struct {
+	count   bool
+	emitted bool
+}
+
+func (recorder *routingDriver) Open(string) (driver.Conn, error) {
+	return &routingConnection{owner: recorder}, nil
+}
+func (connection *routingConnection) Prepare(string) (driver.Stmt, error) {
+	return nil, errors.New("unexpected prepare")
+}
+func (connection *routingConnection) Close() error { return nil }
+func (connection *routingConnection) Begin() (driver.Tx, error) {
+	return nil, errors.New("unexpected transaction")
+}
+func (connection *routingConnection) Ping(context.Context) error { return nil }
+func (connection *routingConnection) ExecContext(_ context.Context, statement string, _ []driver.NamedValue) (driver.Result, error) {
+	connection.owner.statements = append(connection.owner.statements, statement)
+	return driver.RowsAffected(1), nil
+}
+func (connection *routingConnection) QueryContext(_ context.Context, statement string, _ []driver.NamedValue) (driver.Rows, error) {
+	connection.owner.statements = append(connection.owner.statements, statement)
+	return &routingRows{count: strings.Contains(statement, "COUNT(*)")}, nil
+}
+func (rows *routingRows) Columns() []string {
+	if rows.count {
+		return []string{"count"}
+	}
+	return []string{"ts", "usage_mb", "package_code", "iccid"}
+}
+func (rows *routingRows) Close() error { return nil }
+func (rows *routingRows) Next(values []driver.Value) error {
+	if !rows.count || rows.emitted {
+		return io.EOF
+	}
+	rows.emitted = true
+	values[0] = int64(0)
+	return nil
+}
+
+var routingRecorder = &routingDriver{}
+
+func init() {
+	sql.Register("admin_traffic_split_routing", routingRecorder)
+}
+
+func TestTrafficUsageReadWriteDeleteSplitRouting(t *testing.T) {
+	recorder := routingRecorder
+	recorder.statements = nil
+	previous := g.Cfg().GetAdapter()
+	t.Cleanup(func() { _ = tdengine.Close(); g.Cfg().SetAdapter(previous) })
+	adapter, err := gcfg.NewAdapterContent("tsdb:\n  database: TdEngine\n  tdengine:\n    type: admin_traffic_split_routing\n    dsn: test\n    dbName: legacy\n    dailyDbName: iotlcm_daily\n    monthlyDbName: iotlcm_monthly\n")
+	if err != nil {
+		t.Fatal(err)
+	}
+	g.Cfg().SetAdapter(adapter)
+	_ = tdengine.Close()
+	usage := NewTrafficUsage()
+	ctx := context.Background()
+	period := time.Date(2026, 10, 21, 0, 0, 0, 0, time.FixedZone("CST", 8*60*60))
+	base := model.TrafficUsageBase{Iccid: "8986112520609988658", UsageMb: 12.5, PackageCode: "test"}
+	for _, test := range []struct {
+		name      string
+		database  string
+		stable    string
+		operation func() error
+	}{
+		{"write_daily", "iotlcm_daily", "sim_card_usage_d_", func() error {
+			return usage.WriteDaily(ctx, []model.TrafficUsageDailyItem{{TrafficUsageBase: base, Date: period}})
+		}},
+		{"write_monthly", "iotlcm_monthly", "sim_card_usage_m_", func() error {
+			return usage.WriteMonthly(ctx, []model.TrafficUsageMonthlyItem{{TrafficUsageBase: base, Month: period}})
+		}},
+		{"read_daily", "iotlcm_daily", "sim_card_usage_d_", func() error {
+			_, _, err := usage.QueryDaily(ctx, model.TrafficUsageQueryInput{Iccid: base.Iccid})
+			return err
+		}},
+		{"read_monthly", "iotlcm_monthly", "sim_card_usage_m_", func() error {
+			_, _, err := usage.QueryMonthly(ctx, model.TrafficUsageQueryInput{Iccid: base.Iccid})
+			return err
+		}},
+		{"delete_daily", "iotlcm_daily", "sim_card_usage_d_", func() error {
+			return usage.DeleteDaily(ctx, model.TrafficUsageDeleteInput{Iccid: base.Iccid, StartTime: period, EndTime: period.AddDate(0, 0, 1)})
+		}},
+		{"delete_monthly", "iotlcm_monthly", "sim_card_usage_m_", func() error {
+			return usage.DeleteMonthly(ctx, model.TrafficUsageDeleteInput{Iccid: base.Iccid, StartTime: period, EndTime: period.AddDate(0, 1, 0)})
+		}},
+	} {
+		t.Run(test.name, func(t *testing.T) {
+			recorder.statements = nil
+			if err := test.operation(); err != nil {
+				t.Fatal(err)
+			}
+			if len(recorder.statements) == 0 {
+				t.Fatal("no SQL issued")
+			}
+			for _, statement := range recorder.statements {
+				normalized := strings.ReplaceAll(statement, "\x60", "")
+				if !strings.Contains(normalized, test.database+"."+test.stable+base.Iccid) || strings.Contains(normalized, "legacy.") {
+					t.Errorf("wrong target database: %s", statement)
+				}
+			}
+		})
+	}
+}