shishengliang/lingniu-vehicle-ingest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(go): add capacity health check tool

Browse Source
This commit is contained in:
lingniu
2026-07-03 19:54:29 +08:00
parent 68b729fa7b
commit 3edd7462c7
6 changed files with 349 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
go/vehicle-gateway/internal/capacity/check_test.go Normal file
View File

@@ -0,0 +1,66 @@
package capacity
import (
"strings"
"testing"
)
func TestEvaluateReportsOKWhenCriticalBacklogsAreZero(t *testing.T) {
report := Evaluate(map[string]string{
"gateway": `vehicle_gateway_active_connections{protocol="JT808"} 50000
vehicle_gateway_connection_rejections_total{protocol="JT808",reason="max_connections"} 0
vehicle_async_sink_queue_depth{sink="nats"} 0`,
"fast-writer": `vehicle_fast_writer_nats_consumer_ack_pending{consumer="vehicle-fast-writer",stream="VEHICLE_INGEST"} 0
vehicle_fast_writer_nats_consumer_pending{consumer="vehicle-fast-writer",stream="VEHICLE_INGEST"} 0
vehicle_fast_writer_batch_pending_messages 0`,
"history": `vehicle_history_batch_pending_messages 0
vehicle_history_batch_pending_rows 0
vehicle_history_kafka_lag{topic="vehicle.raw.go.jt808.v1",partition="0"} 0`,
})
if report.Status != StatusOK {
t.Fatalf("status = %s, want ok: %#v", report.Status, report)
}
if report.Totals.ActiveConnections != 50000 {
t.Fatalf("active connections = %d, want 50000", report.Totals.ActiveConnections)
}
if len(report.Findings) != 0 {
t.Fatalf("findings = %#v, want none", report.Findings)
}
}
func TestEvaluateReportsDegradedForPendingAndRejects(t *testing.T) {
report := Evaluate(map[string]string{
"gateway": `vehicle_gateway_active_connections{protocol="JT808"} 120000
vehicle_gateway_connection_rejections_total{protocol="JT808",reason="max_connections"} 3
vehicle_async_sink_queue_depth{sink="nats"} 25000`,
"bridge": `vehicle_bridge_nats_consumer_ack_pending{consumer="vehicle-kafka-bridge",stream="VEHICLE_INGEST"} 101
vehicle_bridge_nats_consumer_pending{consumer="vehicle-kafka-bridge",stream="VEHICLE_INGEST"} 12001
vehicle_bridge_batch_pending_messages 1001`,
"fast-writer": `vehicle_fast_writer_nats_consumer_ack_pending{consumer="vehicle-fast-writer",stream="VEHICLE_INGEST"} 11
vehicle_fast_writer_nats_consumer_pending{consumer="vehicle-fast-writer",stream="VEHICLE_INGEST"} 12001
vehicle_fast_writer_batch_pending_messages 9`,
"history": `vehicle_history_batch_pending_messages 6
vehicle_history_kafka_lag{topic="vehicle.raw.go.jt808.v1",partition="0"} 42`,
})
if report.Status != StatusDegraded {
t.Fatalf("status = %s, want degraded: %#v", report.Status, report)
}
joined := strings.Join(report.Findings, "\n")
for _, want := range []string{
"gateway connection rejections",
"async sink queue depth",
"bridge ack pending",
"bridge consumer pending",
"bridge batch pending",
"fast writer ack pending",
"fast writer consumer pending",
"history batch pending",
"kafka lag",
} {
if !strings.Contains(joined, want) {
t.Fatalf("finding missing %q in:\n%s", want, joined)
}
}
}