feat(platform-api): enforce request timeout
This commit is contained in:
@@ -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,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user