package collector
|
|
import (
|
"encoding/json"
|
|
"plc-recorder/config"
|
"plc-recorder/logger"
|
"plc-recorder/msg"
|
"plc-recorder/util"
|
)
|
|
func getDeviceList() ([]msg.PLCDevice, error) {
|
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
|
}
|
|
var response msg.ApsDeviceApiResponse
|
err = json.Unmarshal(responseBody, &response)
|
if err != nil {
|
logger.Warn("unmarshal aps response error:%s", err.Error())
|
return nil, err
|
}
|
|
logger.Debug("get device list total:%d", response.Total)
|
|
return response.Data, nil
|
}
|
|
// 设备信息发生改变时更新采集的任务, 删除, 修改, 或者重新启动
|
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
|
}
|
|
// 先停掉正在执行的任务
|
stopTask(&device)
|
|
// 判断是否是重新启动的状态, 启动一个新的任务
|
if device.Status == 0 {
|
addTask(&device)
|
}
|
|
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)
|
}
|
