feat(go): persist realtime snapshots to mysql

go/vehicle-gateway/internal/realtime/snapshot_writer_test.go

@@ -0,0 +1,90 @@
+package realtime
+
+import (
+	"context"
+	"database/sql"
+	"strings"
+	"testing"
+
+	"lingniu-vehicle-ingest/go/vehicle-gateway/internal/envelope"
+)
+
+func TestSnapshotWriterEnsuresSchemaAndUpsertsParsedJSON(t *testing.T) {
+	exec := &recordingSnapshotExec{}
+	writer := NewSnapshotWriter(exec)
+
+	if err := writer.EnsureSchema(context.Background()); err != nil {
+		t.Fatalf("EnsureSchema() error = %v", err)
+	}
+	if err := writer.Update(context.Background(), envelope.FrameEnvelope{
+		Protocol:       envelope.ProtocolGB32960,
+		MessageID:      "0x02",
+		VIN:            "VIN001",
+		Phone:          "13307795425",
+		DeviceID:       "iccid-1",
+		SourceEndpoint: "1.2.3.4:32960",
+		EventTimeMS:    1782918600000,
+		ReceivedAtMS:   1782918601000,
+		Parsed: map[string]any{
+			"data_units": []any{
+				map[string]any{"name": "vehicle", "type": "0x01", "value": map[string]any{"soc_percent": 90}},
+				map[string]any{"name": "gd_fc_vendor_tlv", "type": "vendor", "value": map[string]any{"foo": "bar"}},
+			},
+		},
+		Fields: map[string]any{envelope.FieldSOCPercent: 90},
+	}); err != nil {
+		t.Fatalf("Update() error = %v", err)
+	}
+
+	if len(exec.calls) != 2 {
+		t.Fatalf("exec calls = %d, want 2", len(exec.calls))
+	}
+	if !strings.Contains(exec.calls[0].query, "CREATE TABLE IF NOT EXISTS vehicle_realtime_snapshot") {
+		t.Fatalf("schema query = %s", exec.calls[0].query)
+	}
+	upsert := exec.calls[1]
+	if !strings.Contains(upsert.query, "ON DUPLICATE KEY UPDATE") {
+		t.Fatalf("upsert query = %s", upsert.query)
+	}
+	if got, want := upsert.args[0], "GB32960"; got != want {
+		t.Fatalf("protocol arg = %#v, want %q", got, want)
+	}
+	if got, want := upsert.args[1], "VIN001"; got != want {
+		t.Fatalf("vehicle key arg = %#v, want %q", got, want)
+	}
+	parsedJSON, ok := upsert.args[10].(string)
+	if !ok {
+		t.Fatalf("parsed_json arg type = %T", upsert.args[10])
+	}
+	if !strings.Contains(parsedJSON, "gd_fc_vendor_tlv") || !strings.Contains(parsedJSON, "vehicle") {
+		t.Fatalf("parsed_json should contain original and vendor parsed data: %s", parsedJSON)
+	}
+}
+
+func TestSnapshotWriterSkipsUnknownVehicleKey(t *testing.T) {
+	exec := &recordingSnapshotExec{}
+	writer := NewSnapshotWriter(exec)
+
+	if err := writer.Update(context.Background(), envelope.FrameEnvelope{
+		Protocol: envelope.ProtocolJT808,
+	}); err != nil {
+		t.Fatalf("Update() error = %v", err)
+	}
+	if len(exec.calls) != 0 {
+		t.Fatalf("exec calls = %d, want 0", len(exec.calls))
+	}
+}
+
+type recordingSnapshotExec struct {
+	calls []snapshotExecCall
+}
+
+type snapshotExecCall struct {
+	query string
+	args  []any
+}
+
+func (e *recordingSnapshotExec) ExecContext(_ context.Context, query string, args ...any) (sql.Result, error) {
+	e.calls = append(e.calls, snapshotExecCall{query: query, args: args})
+	return nil, nil
+}