修改上报的数据结构, 添加plc查询接口

zhangzengfei

2023-08-11 bc0b7e914a378b2c40f9d2ec2470b61a19c18288

 collector/device.go

@@ -10,7 +10,8 @@
)

func getDeviceList() ([]msg.PLCDevice, error) {
   responseBody, err := util.HttpPost(config.Options.ApsDeviceWebApi, nil)
   token, _ := util.CreateToken()
   responseBody, err := util.HttpPost(config.Options.DeviceListWebApi, nil, token)
   if err != nil {
      logger.Warn("get device list from aps error:%s", err.Error())
      return nil, err
@@ -32,16 +33,16 @@
func HandleDeviceUpdate(message []byte) error {
   var device msg.PLCDevice

   logger.Debug("receive device update msg: %s", string(message))

   err := json.Unmarshal(message, &device)
   if err != nil {
      logger.Error("unmarshal device update msg error:%s", err.Error())
      return err
   }

   if task, ok := mapTask.Load(device.DeviceID); ok {
      // 存在的任务, 先停止掉, 然后重新开启一个
      task.(collectorProc).cancel()
   }
   // 先停掉正在执行的任务
   stopTask(&device)

   // 判断是否是重新启动的状态, 启动一个新的任务
   if device.Status == 0 {
@@ -50,3 +51,33 @@

   return nil
}

func HandleDeviceReadRequest(device *msg.PLCDevice) *msg.PLCResponse {
   procCache, ok := tasksStore.Load(device.DeviceID)
   if ok {
      if procCache.(*collectorProc).device.DeviceIP == device.DeviceIP {
         return plc4xRequest(procCache.(*collectorProc))
      }
   }

   plcConnection, err := NewModbusConnection(device.DeviceIP)
   if err != nil {
      logger.Warn("error connecting to PLC: %s, ip: %s", device.DeviceName, device.DeviceIP)

      // 上报设备离线
      return &msg.PLCResponse{
         DeviceID:   device.DeviceID,
         DeviceName: device.DeviceName,
         DeviceIP:   device.DeviceIP,
         Online:     false,
      }
   }

   proc := collectorProc{
      device:  device,
      cancel:  nil,
      plcConn: &plcConnection,
   }

   return plc4xRequest(&proc)
}