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