fix: verify raw freshness by received time

This commit is contained in:
lingniu
2026-07-02 09:16:45 +08:00
parent 0a024f9a76
commit 318486b1e9
4 changed files with 130 additions and 20 deletions

View File

@@ -23,6 +23,7 @@ type RawFrameQuery struct {
Phone string Phone string
DeviceID string DeviceID string
MessageID string MessageID string
OrderBy string
DateFrom string DateFrom string
DateTo string DateTo string
Limit int Limit int
@@ -366,6 +367,7 @@ func normalizeRawFrameQuery(query RawFrameQuery) RawFrameQuery {
query.Phone = strings.TrimSpace(query.Phone) query.Phone = strings.TrimSpace(query.Phone)
query.DeviceID = strings.TrimSpace(query.DeviceID) query.DeviceID = strings.TrimSpace(query.DeviceID)
query.MessageID = strings.TrimSpace(query.MessageID) query.MessageID = strings.TrimSpace(query.MessageID)
query.OrderBy = normalizeRawFrameOrderBy(query.OrderBy)
query.DateFrom = normalizeDateTimeLiteral(query.DateFrom) query.DateFrom = normalizeDateTimeLiteral(query.DateFrom)
query.DateTo = normalizeDateTimeLiteral(query.DateTo) query.DateTo = normalizeDateTimeLiteral(query.DateTo)
if query.Limit <= 0 { if query.Limit <= 0 {
@@ -408,7 +410,7 @@ func buildRawFrameSQL(table string, query RawFrameQuery) (string, []any) {
if len(where) > 0 { if len(where) > 0 {
sqlText += " WHERE " + strings.Join(where, " AND ") sqlText += " WHERE " + strings.Join(where, " AND ")
} }
sqlText += " ORDER BY ts DESC LIMIT " + strconv.Itoa(query.Limit) + " OFFSET " + strconv.Itoa(query.Offset) sqlText += " ORDER BY " + rawFrameOrderColumn(query.OrderBy) + " DESC LIMIT " + strconv.Itoa(query.Limit) + " OFFSET " + strconv.Itoa(query.Offset)
return sqlText, nil return sqlText, nil
} }
@@ -490,6 +492,23 @@ func rawFrameWhere(query RawFrameQuery) []string {
return where return where
} }
func normalizeRawFrameOrderBy(value string) string {
value = strings.ToLower(strings.TrimSpace(value))
switch value {
case "receivedat", "received_at":
return "receivedAt"
default:
return "ts"
}
}
func rawFrameOrderColumn(value string) string {
if normalizeRawFrameOrderBy(value) == "receivedAt" {
return "received_at"
}
return "ts"
}
func locationWhere(query LocationQuery) []string { func locationWhere(query LocationQuery) []string {
var where []string var where []string
add := func(clause string) { add := func(clause string) {
@@ -697,6 +716,7 @@ func parseRawFrameQuery(r *http.Request) (RawFrameQuery, error) {
Phone: values.Get("phone"), Phone: values.Get("phone"),
DeviceID: values.Get("deviceId"), DeviceID: values.Get("deviceId"),
MessageID: values.Get("messageId"), MessageID: values.Get("messageId"),
OrderBy: values.Get("orderBy"),
DateFrom: values.Get("dateFrom"), DateFrom: values.Get("dateFrom"),
DateTo: values.Get("dateTo"), DateTo: values.Get("dateTo"),
Limit: limit, Limit: limit,

View File

@@ -370,3 +370,17 @@ func TestBuildRawFrameSQLUsesLiteralsForTDengine(t *testing.T) {
} }
} }
} }
func TestBuildRawFrameSQLCanOrderByReceivedAt(t *testing.T) {
sqlText, args := buildRawFrameSQL("lingniu_vehicle_ts.raw_frames", RawFrameQuery{
Protocol: "JT808",
OrderBy: "receivedAt",
Limit: 1,
})
if len(args) != 0 {
t.Fatalf("expected no query args for TDengine, got %#v", args)
}
if !strings.Contains(sqlText, "ORDER BY received_at DESC LIMIT 1 OFFSET 0") {
t.Fatalf("sql should order by received_at: %s", sqlText)
}
}

View File

@@ -40,6 +40,14 @@ class Check:
message: str message: str
@dataclass(frozen=True)
class CheckWindow:
date_from: str
date_to: str
stat_date: str
max_raw_age_minutes: float | None
def shanghai_day_window(now: dt.datetime | None = None) -> tuple[str, str]: def shanghai_day_window(now: dt.datetime | None = None) -> tuple[str, str]:
now = now or dt.datetime.now(tz=SHANGHAI) now = now or dt.datetime.now(tz=SHANGHAI)
local = now.astimezone(SHANGHAI) local = now.astimezone(SHANGHAI)
@@ -48,6 +56,23 @@ def shanghai_day_window(now: dt.datetime | None = None) -> tuple[str, str]:
return start.isoformat(timespec="seconds"), end.isoformat(timespec="seconds") return start.isoformat(timespec="seconds"), end.isoformat(timespec="seconds")
def resolve_check_window(
raw_date: str | None,
max_raw_age_minutes: float,
now: dt.datetime | None = None,
) -> CheckWindow:
current = (now or dt.datetime.now(tz=SHANGHAI)).astimezone(SHANGHAI)
today = current.date()
if raw_date:
day = dt.date.fromisoformat(raw_date)
start = dt.datetime(day.year, day.month, day.day, tzinfo=SHANGHAI)
date_from, date_to = shanghai_day_window(start)
freshness = max_raw_age_minutes if day == today else None
return CheckWindow(date_from, date_to, raw_date, freshness)
date_from, date_to = shanghai_day_window(current)
return CheckWindow(date_from, date_to, date_from[:10], max_raw_age_minutes)
def api_url(base_url: str, path: str, params: dict[str, Any]) -> str: def api_url(base_url: str, path: str, params: dict[str, Any]) -> str:
url = base_url.rstrip("/") + path url = base_url.rstrip("/") + path
if not params: if not params:
@@ -129,12 +154,15 @@ def parse_tdengine_utc_timestamp(value: str) -> dt.datetime:
def latest_age_minutes(items: list[Any], now: dt.datetime | None = None) -> float | None: def latest_age_minutes(items: list[Any], now: dt.datetime | None = None) -> float | None:
if not items or not isinstance(items[0], dict) or not items[0].get("ts"): if not items or not isinstance(items[0], dict):
return None
timestamp = items[0].get("received_at") or items[0].get("ts")
if not timestamp:
return None return None
current = now or dt.datetime.now(tz=dt.timezone.utc) current = now or dt.datetime.now(tz=dt.timezone.utc)
if current.tzinfo is None: if current.tzinfo is None:
current = current.replace(tzinfo=dt.timezone.utc) current = current.replace(tzinfo=dt.timezone.utc)
latest = parse_tdengine_utc_timestamp(str(items[0]["ts"])) latest = parse_tdengine_utc_timestamp(str(timestamp))
return max(0.0, (current.astimezone(dt.timezone.utc) - latest).total_seconds() / 60) return max(0.0, (current.astimezone(dt.timezone.utc) - latest).total_seconds() / 60)
@@ -291,7 +319,7 @@ def build_check_specs(
min_stat: int, min_stat: int,
max_raw_age_minutes: float | None = None, max_raw_age_minutes: float | None = None,
) -> list[CheckSpec]: ) -> list[CheckSpec]:
raw_params = {"dateFrom": date_from, "dateTo": date_to, "limit": 1} raw_params = {"dateFrom": date_from, "dateTo": date_to, "limit": 1, "orderBy": "receivedAt"}
gb32960_history_params = {"protocol": "GB32960", "dateFrom": date_from, "dateTo": date_to, "limit": 1} gb32960_history_params = {"protocol": "GB32960", "dateFrom": date_from, "dateTo": date_to, "limit": 1}
jt808_history_params = {"protocol": "JT808", "dateFrom": date_from, "dateTo": date_to, "limit": 1} jt808_history_params = {"protocol": "JT808", "dateFrom": date_from, "dateTo": date_to, "limit": 1}
yutong_mqtt_history_params = {"protocol": "YUTONG_MQTT", "dateFrom": date_from, "dateTo": date_to, "limit": 1} yutong_mqtt_history_params = {"protocol": "YUTONG_MQTT", "dateFrom": date_from, "dateTo": date_to, "limit": 1}
@@ -489,24 +517,15 @@ def parse_args(argv: list[str]) -> argparse.Namespace:
def main(argv: list[str]) -> int: def main(argv: list[str]) -> int:
args = parse_args(argv) args = parse_args(argv)
if args.date: window = resolve_check_window(args.date, args.max_raw_age_minutes)
day = dt.date.fromisoformat(args.date)
start = dt.datetime(day.year, day.month, day.day, tzinfo=SHANGHAI)
date_from, date_to = shanghai_day_window(start)
stat_date = args.date
max_raw_age_minutes = None
else:
date_from, date_to = shanghai_day_window()
stat_date = date_from[:10]
max_raw_age_minutes = args.max_raw_age_minutes
specs = build_check_specs( specs = build_check_specs(
date_from=date_from, date_from=window.date_from,
date_to=date_to, date_to=window.date_to,
stat_date=stat_date, stat_date=window.stat_date,
min_raw=args.min_raw, min_raw=args.min_raw,
min_history=args.min_history, min_history=args.min_history,
min_stat=args.min_stat, min_stat=args.min_stat,
max_raw_age_minutes=max_raw_age_minutes, max_raw_age_minutes=window.max_raw_age_minutes,
) )
checks, payloads = query_payloads(args.base_url, specs, args.timeout) checks, payloads = query_payloads(args.base_url, specs, args.timeout)
checks.extend(run_checks(args.base_url, build_realtime_specs(payloads), args.timeout)) checks.extend(run_checks(args.base_url, build_realtime_specs(payloads), args.timeout))
@@ -514,8 +533,8 @@ def main(argv: list[str]) -> int:
print(json.dumps({ print(json.dumps({
"status": status, "status": status,
"baseUrl": args.base_url, "baseUrl": args.base_url,
"dateFrom": date_from, "dateFrom": window.date_from,
"dateTo": date_to, "dateTo": window.date_to,
"checks": [asdict(check) for check in checks], "checks": [asdict(check) for check in checks],
}, ensure_ascii=False, indent=2)) }, ensure_ascii=False, indent=2))
return 0 if status == "pass" else 1 return 0 if status == "pass" else 1

View File

@@ -32,6 +32,26 @@ class GoNativeProdSmokeTest(unittest.TestCase):
"&dateTo=2026-07-03T00%3A00%3A00%2B08%3A00&limit=1", "&dateTo=2026-07-03T00%3A00%3A00%2B08%3A00&limit=1",
) )
def test_explicit_today_date_still_enforces_raw_freshness(self):
now = dt.datetime(2026, 7, 2, 9, 30, tzinfo=smoke.SHANGHAI)
window = smoke.resolve_check_window("2026-07-02", 15.0, now)
self.assertEqual(window.date_from, "2026-07-02T00:00:00+08:00")
self.assertEqual(window.date_to, "2026-07-03T00:00:00+08:00")
self.assertEqual(window.stat_date, "2026-07-02")
self.assertEqual(window.max_raw_age_minutes, 15.0)
def test_explicit_historical_date_disables_raw_freshness(self):
now = dt.datetime(2026, 7, 2, 9, 30, tzinfo=smoke.SHANGHAI)
window = smoke.resolve_check_window("2026-07-01", 15.0, now)
self.assertEqual(window.date_from, "2026-07-01T00:00:00+08:00")
self.assertEqual(window.date_to, "2026-07-02T00:00:00+08:00")
self.assertEqual(window.stat_date, "2026-07-01")
self.assertIsNone(window.max_raw_age_minutes)
def test_total_check_passes_when_total_reaches_minimum(self): def test_total_check_passes_when_total_reaches_minimum(self):
check = smoke.check_total( check = smoke.check_total(
"jt808.raw", "jt808.raw",
@@ -82,6 +102,23 @@ class GoNativeProdSmokeTest(unittest.TestCase):
self.assertIn("gb32960.daily_mileage", names) self.assertIn("gb32960.daily_mileage", names)
self.assertIn("jt808.daily_total_mileage", names) self.assertIn("jt808.daily_total_mileage", names)
def test_raw_checks_order_by_received_at_for_ingest_freshness(self):
specs = smoke.build_check_specs(
date_from="2026-07-02T00:00:00+08:00",
date_to="2026-07-03T00:00:00+08:00",
stat_date="2026-07-02",
min_raw=1,
min_history=1,
min_stat=1,
max_raw_age_minutes=15,
)
raw_specs = [spec for spec in specs if spec.name.endswith(".raw")]
self.assertTrue(raw_specs)
for spec in raw_specs:
self.assertEqual(spec.params.get("orderBy"), "receivedAt")
def test_vehicle_identifier_prefers_vin_over_vehicle_key(self): def test_vehicle_identifier_prefers_vin_over_vehicle_key(self):
identifier = smoke.vehicle_identifier({ identifier = smoke.vehicle_identifier({
"vin": "LB9A32A20R0LS1343", "vin": "LB9A32A20R0LS1343",
@@ -185,6 +222,26 @@ class GoNativeProdSmokeTest(unittest.TestCase):
self.assertEqual(check.status, "fail") self.assertEqual(check.status, "fail")
self.assertIn("latest_age_minutes=60.0", check.message) self.assertIn("latest_age_minutes=60.0", check.message)
def test_total_check_uses_received_at_for_raw_freshness_when_present(self):
now = dt.datetime(2026, 7, 1, 18, 0, 0, tzinfo=dt.timezone.utc)
check = smoke.check_total(
"gb32960.raw",
{
"total": 1,
"items": [{
"ts": "2026-07-01 17:00:00",
"received_at": "2026-07-01 17:58:00",
}],
},
minimum=1,
max_age_minutes=15,
now=now,
)
self.assertEqual(check.status, "pass")
self.assertIn("latest_age_minutes=2.0", check.message)
def test_total_check_passes_when_latest_sample_is_fresh(self): def test_total_check_passes_when_latest_sample_is_fresh(self):
now = dt.datetime(2026, 7, 1, 18, 0, 0, tzinfo=dt.timezone.utc) now = dt.datetime(2026, 7, 1, 18, 0, 0, tzinfo=dt.timezone.utc)