From 2fda41d9ee2dd9eb34527cb4bcf6e052b355ba71 Mon Sep 17 00:00:00 2001
From: Marvin Zhang <tikazyq@gmail.com>
Date: Wed, 24 Jul 2019 21:03:30 +0800
Subject: [PATCH] added network diagram

---
 backend/model/node.go  | 3 +++
 backend/routes/node.go | 9 +++++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/backend/model/node.go b/backend/model/node.go
index 63461ced..4c51e41d 100644
--- a/backend/model/node.go
+++ b/backend/model/node.go
@@ -18,6 +18,9 @@ type Node struct {
 	Mac         string        `json:"mac" bson:"mac"`
 	Description string        `json:"description" bson:"description"`
 
+	// 前端展示
+	IsMaster bool `json:"is_master"`
+
 	UpdateTs time.Time `json:"update_ts" bson:"update_ts"`
 	CreateTs time.Time `json:"create_ts" bson:"create_ts"`
 }
diff --git a/backend/routes/node.go b/backend/routes/node.go
index e9233b76..756af958 100644
--- a/backend/routes/node.go
+++ b/backend/routes/node.go
@@ -9,15 +9,20 @@ import (
 )
 
 func GetNodeList(c *gin.Context) {
-	results, err := model.GetNodeList(nil)
+	nodes, err := model.GetNodeList(nil)
 	if err != nil {
 		HandleError(http.StatusInternalServerError, c, err)
 		return
 	}
+
+	for i, node := range nodes {
+		nodes[i].IsMaster = services.IsMasterNode(node.Id.Hex())
+	}
+
 	c.JSON(http.StatusOK, Response{
 		Status:  "ok",
 		Message: "success",
-		Data:    results,
+		Data:    nodes,
 	})
 }