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

fix(go): keep fast writer tdengine pool safe by default

Browse Source
This commit is contained in:
lingniu
2026-07-03 19:33:45 +08:00
parent 06866b0fa7
commit fb34969602
4 changed files with 51 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
go/vehicle-gateway/cmd/nats-fast-writer/main_test.go
View File

@@ -197,6 +197,35 @@ func TestRecordFastNATSConsumerInfoMetrics(t *testing.T) {
}
}
func TestLoadConfigDefaultsTDenginePoolToSingleConnection(t *testing.T) {
t.Setenv("FAST_WRITER_WORKERS", "12")
t.Setenv("FAST_WRITER_TDENGINE_MAX_OPEN_CONNS", "")
cfg := loadConfig()
if cfg.TDengineMaxOpenConns != 1 {
t.Fatalf("TDengineMaxOpenConns = %d, want 1", cfg.TDengineMaxOpenConns)
}
if cfg.TDengineMaxIdleConns != 1 {
t.Fatalf("TDengineMaxIdleConns = %d, want 1", cfg.TDengineMaxIdleConns)
}
}
func TestLoadConfigReadsTDenginePoolOverride(t *testing.T) {
t.Setenv("FAST_WRITER_WORKERS", "12")
t.Setenv("FAST_WRITER_TDENGINE_MAX_OPEN_CONNS", "4")
t.Setenv("FAST_WRITER_TDENGINE_MAX_IDLE_CONNS", "2")
cfg := loadConfig()
if cfg.TDengineMaxOpenConns != 4 {
t.Fatalf("TDengineMaxOpenConns = %d, want 4", cfg.TDengineMaxOpenConns)
}
if cfg.TDengineMaxIdleConns != 2 {
t.Fatalf("TDengineMaxIdleConns = %d, want 2", cfg.TDengineMaxIdleConns)
}
}
type recordingFastAppender struct {
count int
batchCount int