From cc81b80397e02eb7a005811f721f502e32defed4 Mon Sep 17 00:00:00 2001 From: marvzhang Date: Sun, 8 Mar 2020 17:10:13 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=8A=82=E7=82=B9=E5=90=8D?= =?UTF-8?q?=E7=A7=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/services/node.go | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/backend/services/node.go b/backend/services/node.go index f8db160a..89187a81 100644 --- a/backend/services/node.go +++ b/backend/services/node.go @@ -15,6 +15,7 @@ import ( "github.com/globalsign/mgo" "github.com/globalsign/mgo/bson" "github.com/gomodule/redigo/redis" + "github.com/spf13/viper" "runtime/debug" "time" ) @@ -104,6 +105,19 @@ func UpdateNodeStatus() { model.ResetNodeStatusToOffline(list) } +func getNodeName(data *Data) string { + registerType := viper.GetString("server.register.type") + if registerType == constants.RegisterTypeMac { + return data.Ip + } else if registerType == constants.RegisterTypeIp { + return data.Ip + } else if registerType == constants.RegisterTypeHostname { + return data.Hostname + } else { + return data.Ip + } +} + // 处理节点信息 func handleNodeInfo(key string, data *Data) { // 添加同步锁 @@ -122,7 +136,7 @@ func handleNodeInfo(key string, data *Data) { // 数据库不存在该节点 node = model.Node{ Key: key, - Name: data.Ip, + Name: getNodeName(data), Ip: data.Ip, Port: "8000", Mac: data.Mac,