servermanager

pkg/server/server_manager.go

@@ -18,6 +18,7 @@ const (
 	EtcdServersPrefixCnt = 2
 	EnvTCPProxy          = "TCP_PROXY_ADDR"
 	EnvHTTPProxy         = "HTTP_PROXY_ADDR"
+	EnvUDPProxy          = "UDP_PROXY_ADDR"
 )
 
 // ServerManager server manager
@@ -72,6 +73,13 @@ func (mgr *ServerManager) RegisterServer() error {
 		addr, _ := fixHostIp(*confRPCHost)
 		response, err = kapi.Set(context.Background(), prefix+FlagRPCHost+"/"+addr, addr, opt)
 	}
+	if serverInstance.udpsvr != nil {
+		addr := os.Getenv(EnvUDPProxy)
+		if addr == "" {
+			addr, _ = fixHostIp(*confUDPHost)
+		}
+		response, err = kapi.Set(context.Background(), prefix+FlagUDPHost+"/"+addr, addr, opt)
+	}
 	if serverInstance.httpsvr != nil {
 		addr := os.Getenv(EnvHTTPProxy)
 		if addr == "" {