@@ -3,8 +3,6 @@ package main
33import (
44 "context"
55 "fmt"
6- "os"
7- "strconv"
86 "time"
97
108 "github.com/inverse-inc/go-utils/log"
@@ -43,12 +41,11 @@ type Backend struct {
4341 LastCheck time.Time
4442}
4543
46- // getHealthCheckPort returns the health check port from environment variable or default
47- func getHealthCheckPort () int {
48- if portStr := os .Getenv ("PFUDPPROXY_HEALTH_CHECK_PORT" ); portStr != "" {
49- if port , err := strconv .Atoi (portStr ); err == nil && port > 0 && port < 65536 {
50- return port
51- }
44+ // getHealthCheckPort returns the health check port from FingerbankSettingsCollector or default
45+ func getHealthCheckPort (ctx context.Context ) int {
46+ collector := pfconfigdriver.GetType [pfconfigdriver.FingerbankSettingsCollector ](ctx )
47+ if port , err := collector .Port .Int64 (); err == nil && port > 0 && port < 65536 {
48+ return int (port )
5249 }
5350 return DefaultHealthCheckPort
5451}
@@ -57,7 +54,7 @@ func getHealthCheckPort() int {
5754func LoadConfig (ctx context.Context ) (* ProxyConfig , error ) {
5855 config := & ProxyConfig {
5956 Ports : []int {PortNetFlow , PortSFlow },
60- HealthCheckPort : getHealthCheckPort (),
57+ HealthCheckPort : getHealthCheckPort (ctx ),
6158 HealthCheckPath : DefaultHealthCheckPath ,
6259 HealthCheckInterval : DefaultHealthCheckInterval ,
6360 HealthCheckTimeout : DefaultHealthCheckTimeout ,
0 commit comments