设备注册信息

gongshangguo

2022-02-28 8c2d879c9d3e2208b2cd4d9401b3f6756857d0e9

 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)
}