make agent host configurable

Co-authored-by: Pavel Busko <[email protected]>

Author: pbusko

src/cmd/forwarder-agent/app/config.go

@@ -11,6 +11,7 @@ import (
 // GRPC stores the configuration for the router as a server using a PORT
 // with mTLS certs and as a client.
 type GRPC struct {
+	Host         string   `env:"AGENT_HOST, report"`
 	Port         uint16   `env:"AGENT_PORT, report"`
 	CAFile       string   `env:"AGENT_CA_FILE_PATH, required, report"`
 	CertFile     string   `env:"AGENT_CERT_FILE_PATH, required, report"`
@@ -40,6 +41,7 @@ type Config struct {
 func LoadConfig() Config {
 	cfg := Config{
 		GRPC: GRPC{
+			Host: "127.0.0.1",
 			Port: 3458,
 		},
 	}

src/cmd/forwarder-agent/app/forwarder_agent.go

@@ -139,7 +139,7 @@ func (s *ForwarderAgent) Run() {
 	rx := v2.NewReceiver(diode, im, omm)
 
 	s.v2srv = v2.NewServer(
-		fmt.Sprintf("127.0.0.1:%d", s.grpc.Port),
+		fmt.Sprintf("%s:%d", s.grpc.Host, s.grpc.Port),
 		rx,
 		grpc.Creds(serverCreds),
 		grpc.MaxRecvMsgSize(10*1024*1024),

src/cmd/loggregator-agent/app/app_v2.go

@@ -130,7 +130,7 @@ func (a *AppV2) Start() {
 	)
 	go tx.Start()
 
-	agentAddress := fmt.Sprintf("127.0.0.1:%d", a.config.GRPC.Port)
+	agentAddress := fmt.Sprintf("%s:%d", a.config.GRPC.Host, a.config.GRPC.Port)
 	log.Printf("agent v2 API started on addr %s", agentAddress)
 
 	var es envelopeSetter

src/cmd/loggregator-agent/app/config.go

@@ -13,6 +13,7 @@ import (
 // GRPC stores the configuration for the router as a server using a PORT
 // with mTLS certs and as a client.
 type GRPC struct {
+	Host         string   `env:"AGENT_HOST"`
 	Port         uint16   `env:"AGENT_PORT"`
 	CAFile       string   `env:"AGENT_CA_FILE"`
 	CertFile     string   `env:"AGENT_CERT_FILE"`
@@ -50,6 +51,7 @@ func LoadConfig() (*Config, error) {
 			Port: 14824,
 		},
 		GRPC: GRPC{
+			Host: "127.0.0.1",
 			Port: 3458,
 		},
 	}

src/cmd/syslog-agent/app/config.go

@@ -14,6 +14,7 @@ import (
 // GRPC stores the configuration for the router as a server using a PORT
 // with mTLS certs and as a client.
 type GRPC struct {
+	Host         string   `env:"AGENT_HOST,                     report"`
 	Port         int      `env:"AGENT_PORT,                     report"`
 	CAFile       string   `env:"AGENT_CA_FILE_PATH,   required, report"`
 	CertFile     string   `env:"AGENT_CERT_FILE_PATH, required, report"`
@@ -63,6 +64,7 @@ func LoadConfig() Config {
 		},
 		GRPC: GRPC{
 			Port: 3458,
+			Host: "127.0.0.1",
 		},
 		AggregateConnectionRefreshInterval: 1 * time.Minute,
 		DefaultDrainMetadata:               true,

src/cmd/syslog-agent/app/syslog_agent.go

@@ -263,7 +263,7 @@ func (s *SyslogAgent) Run() {
 
 	rx := v2.NewReceiver(diode, im, omm)
 	s.v2Srv = v2.NewServer(
-		fmt.Sprintf("127.0.0.1:%d", s.grpc.Port),
+		fmt.Sprintf("%s:%d", s.grpc.Host, s.grpc.Port),
 		rx,
 		grpc.Creds(serverCreds),
 		grpc.MaxRecvMsgSize(10*1024*1024),

src/cmd/udp-forwarder/app/config.go

@@ -19,8 +19,9 @@ type GRPC struct {
 
 // Config holds the configuration for the UDP agent
 type Config struct {
-	UseRFC3339           bool `env:"USE_RFC3339"`
-	UDPPort              int  `env:"UDP_PORT, report"`
+	UseRFC3339           bool   `env:"USE_RFC3339"`
+	Host                 string `env:"AGENT_HOST, report"`
+	UDPPort              int    `env:"UDP_PORT, report"`
 	LoggregatorAgentGRPC GRPC
 	Deployment           string `env:"DEPLOYMENT, report"`
 	Job                  string `env:"JOB, report"`
@@ -33,6 +34,7 @@ type Config struct {
 // LoadConfig reads from the environment to create a Config.
 func LoadConfig(log *log.Logger) Config {
 	cfg := Config{
+		Host:    "127.0.0.1",
 		UDPPort: 3457,
 		LoggregatorAgentGRPC: GRPC{
 			Addr: "127.0.0.1:3458",

src/cmd/udp-forwarder/app/udp_forwarder.go

@@ -25,6 +25,7 @@ type Metrics interface {
 
 type UDPForwarder struct {
 	grpc         GRPC
+	host         string
 	udpPort      int
 	pprofServer  *http.Server
 	pprofPort    uint16
@@ -44,6 +45,7 @@ type UDPForwarder struct {
 func NewUDPForwarder(cfg Config, l *log.Logger, m Metrics) *UDPForwarder {
 	return &UDPForwarder{
 		grpc:         cfg.LoggregatorAgentGRPC,
+		host:         cfg.Host,
 		udpPort:      cfg.UDPPort,
 		pprofPort:    cfg.MetricsServer.PprofPort,
 		debugMetrics: cfg.MetricsServer.DebugMetrics,
@@ -97,13 +99,13 @@ func (u *UDPForwarder) Run() {
 	dropsondeUnmarshaller := ingress.NewUnMarshaller(w)
 	u.mu.Lock()
 	u.nr, err = ingress.NewNetworkReader(
-		fmt.Sprintf("127.0.0.1:%d", u.udpPort),
+		fmt.Sprintf("%s:%d", u.host, u.udpPort),
 		dropsondeUnmarshaller,
 		u.metrics,
 	)
 	u.mu.Unlock()
 	if err != nil {
-		u.log.Fatalf("Failed to listen on 127.0.0.1:%d: %s", u.udpPort, err)
+		u.log.Fatalf("Failed to listen on %s:%d: %s", u.host, u.udpPort, err)
 	}
 
 	go u.nr.StartReading()