From 8c2d879c9d3e2208b2cd4d9401b3f6756857d0e9 Mon Sep 17 00:00:00 2001
From: gongshangguo <gongshangguo@admin.com>
Date: 星期一, 28 二月 2022 10:39:30 +0800
Subject: [PATCH] 设备注册信息

---
 server/server.go |   16 ++++++++++------
 1 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/server/server.go b/server/server.go
index 205cbc7..6c10b79 100644
--- a/server/server.go
+++ b/server/server.go
@@ -4,6 +4,7 @@
 	"basic.com/valib/go-aiot.git/aiotProto/aiot"
 	"basic.com/valib/go-aiot.git/client"
 	"basic.com/valib/go-aiot.git/util"
+	"encoding/json"
 	uuid "github.com/satori/go.uuid"
 	"go.uber.org/zap"
 	"net"
@@ -160,8 +161,9 @@
 }
 
 // 娉ㄥ唽璁惧淇℃伅
-func (s *Server) SetDeviceList(masterId string, nodeIds []string) bool {
-	if len(nodeIds) == 0 {
+func (s *Server) SetDeviceList(masterId string, registerData *aiot.DeviceRegister) bool {
+
+	if len(registerData.DeviceList) == 0 {
 		return true
 	}
 	// 閿�
@@ -174,9 +176,9 @@
 	}
 
 	// 娣诲姞璁惧ID
-	for _, nodeId := range nodeIds{
-		s.ClusterDevice[masterId][nodeId] = struct{}{}
-		s.Devices[nodeId] = struct{}{}
+	for _, node := range registerData.DeviceList{
+		s.ClusterDevice[masterId][node.DeviceId] = struct{}{}
+		s.Devices[node.DeviceId] = struct{}{}
 	}
 	return true
 }
@@ -265,5 +267,7 @@
 	// 娣诲姞闆嗙兢ID
 	s.SetCluster(msg.SenderId, cli)
 	// 璁剧疆闆嗙兢
-	s.SetDeviceList(msg.SenderId, msg.DeviceProto.DeviceIds)
+	registerData := &aiot.DeviceRegister{}
+	json.Unmarshal(msg.Data, registerData)
+	s.SetDeviceList(msg.SenderId, registerData)
 }

--
Gitblit v1.8.0