From 3fb49693d9e225bb74e27a12dcc691b78ec7ea8f Mon Sep 17 00:00:00 2001 From: Moraxyc Date: Thu, 5 Dec 2024 17:01:21 +0800 Subject: [PATCH] feat: add listen_host (#550) --- cmd/dashboard/main.go | 4 ++-- model/config.go | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cmd/dashboard/main.go b/cmd/dashboard/main.go index 8bcf9b2..49680fb 100644 --- a/cmd/dashboard/main.go +++ b/cmd/dashboard/main.go @@ -112,7 +112,7 @@ func main() { singleton.InitDBFromPath(dashboardCliParam.DatebaseLocation) initSystem() - l, err := net.Listen("tcp", fmt.Sprintf(":%d", singleton.Conf.ListenPort)) + l, err := net.Listen("tcp", fmt.Sprintf("%s:%d", singleton.Conf.ListenHost, singleton.Conf.ListenPort)) if err != nil { log.Fatal(err) } @@ -135,7 +135,7 @@ func main() { go dispatchReportInfoTask() if err := graceful.Graceful(func() error { - log.Println("NEZHA>> Dashboard::START", singleton.Conf.ListenPort) + log.Printf("NEZHA>> Dashboard::START ON %s:%d", singleton.Conf.ListenHost, singleton.Conf.ListenPort) return muxServer.Serve(l) }, func(c context.Context) error { log.Println("NEZHA>> Graceful::START") diff --git a/model/config.go b/model/config.go index 1efe3f3..fd4c784 100644 --- a/model/config.go +++ b/model/config.go @@ -30,6 +30,7 @@ type Config struct { JWTSecretKey string `mapstructure:"jwt_secret_key" json:"jwt_secret_key,omitempty"` AgentSecretKey string `mapstructure:"agent_secret_key" json:"agent_secret_key,omitempty"` ListenPort uint `mapstructure:"listen_port" json:"listen_port,omitempty"` + ListenHost string `mapstructure:"listen_host" json:"listen_host,omitempty"` InstallHost string `mapstructure:"install_host" json:"install_host,omitempty"` TLS bool `mapstructure:"tls" json:"tls,omitempty"` Location string `mapstructure:"location" json:"location,omitempty"` // 时区，默认为 Asia/Shanghai