增加当前节点本地定时缓存，修改部分潜在BUG，启动时Mongo或者redis无法正常连接时，进入启动等待

backend/model/node.go

@@ -26,7 +26,7 @@ type Node struct {
 	Key string `json:"key" bson:"key"`
 
 	// 前端展示
-	IsMaster bool `json:"is_master"`
+	IsMaster bool `json:"is_master" bson:"is_master"`
 
 	UpdateTs     time.Time `json:"update_ts" bson:"update_ts"`
 	CreateTs     time.Time `json:"create_ts" bson:"create_ts"`
@@ -43,64 +43,13 @@ func IsMaster() bool {
 }
 
 // 获取本机节点
-// TODO: 这里职责不单一，需要重构
 func GetCurrentNode() (Node, error) {
 	// 获得注册的key值
 	key, err := register.GetRegister().GetKey()
 	if err != nil {
 		return Node{}, err
 	}
-
-	// 从数据库中获取当前节点
-	var node Node
-	errNum := 0
-	for {
-		// 如果错误次数超过10次
-		if errNum >= 10 {
-			return node, errors.New("cannot get current node")
-		}
-
-		// 尝试获取节点
-		node, err = GetNodeByKey(key)
-		// 如果获取失败
-		if err != nil {
-			// 如果为主节点，表示为第一次注册，插入节点信息
-			// update: 增加具体错误过滤。防止加入多个master节点，后续需要职责拆分，
-			//只在master节点运行的时候才检测master节点的信息是否存在
-			if IsMaster() && err == mgo.ErrNotFound {
-				// 获取本机信息
-				ip, mac, hostname, key, err := GetNodeBaseInfo()
-				if err != nil {
-					debug.PrintStack()
-					return node, err
-				}
-
-				// 生成节点
-				node = Node{
-					Key:      key,
-					Id:       bson.NewObjectId(),
-					Ip:       ip,
-					Name:     key,
-					Mac:      mac,
-					Hostname: hostname,
-					IsMaster: true,
-				}
-				if err := node.Add(); err != nil {
-					return node, err
-				}
-				return node, nil
-			}
-			// 增加错误次数
-			errNum++
-
-			// 5秒后重试
-			time.Sleep(5 * time.Second)
-			continue
-		}
-		// 跳出循环
-		break
-	}
-	return node, nil
+	return GetNodeByKey(key)
 }
 
 func (n *Node) Save() error {
@@ -290,3 +239,33 @@ func ResetNodeStatusToOffline(list []string) {
 		}
 	}
 }
+
+func UpdateMasterNodeInfo(key string, ip string, mac string, hostname string) error {
+	s, c := database.GetCol("nodes")
+	defer s.Close()
+	c.UpdateAll(bson.M{
+		"is_master": true,
+	}, bson.M{
+		"is_master": false,
+	})
+	_, err := c.Upsert(bson.M{
+		"key": key,
+	}, bson.M{
+		"$set": bson.M{
+			"ip":             ip,
+			"port":           "8000",
+			"mac":            mac,
+			"hostname":       hostname,
+			"is_master":      true,
+			"update_ts":      time.Now(),
+			"update_ts_unix": time.Now().Unix(),
+		},
+		"$setOnInsert": bson.M{
+			"key":       key,
+			"name":      key,
+			"create_ts": time.Now(),
+			"_id":       bson.NewObjectId(),
+		},
+	})
+	return err
+}