feat(platform-api): enforce request timeout

This commit is contained in:
lingniu
2026-07-04 00:55:43 +08:00
parent 081f8b5099
commit 4bfaf8c4c8
5 changed files with 57 additions and 1 deletions

View File

@@ -3,6 +3,7 @@ package config
import (
"os"
"strconv"
"time"
)
type Config struct {
@@ -18,6 +19,7 @@ type Config struct {
TDengineDatabase string
CapacityCheckBin string
AuthToken string
RequestTimeout time.Duration
}
func Load() Config {
@@ -34,6 +36,7 @@ func Load() Config {
TDengineDatabase: env("TDENGINE_DATABASE", "lingniu_vehicle_ts"),
CapacityCheckBin: os.Getenv("CAPACITY_CHECK_BIN"),
AuthToken: os.Getenv("AUTH_TOKEN"),
RequestTimeout: time.Duration(envInt("REQUEST_TIMEOUT_MS", 5000)) * time.Millisecond,
}
}

View File

@@ -0,0 +1,16 @@
package config
import (
"testing"
"time"
)
func TestLoadReadsRequestTimeout(t *testing.T) {
t.Setenv("REQUEST_TIMEOUT_MS", "1200")
cfg := Load()
if cfg.RequestTimeout != 1200*time.Millisecond {
t.Fatalf("RequestTimeout = %s, want 1.2s", cfg.RequestTimeout)
}
}