package nsq
|
|
import (
|
"apsClient/conf"
|
"apsClient/constvar"
|
"apsClient/model"
|
"apsClient/model/common"
|
"apsClient/pkg/logx"
|
"apsClient/pkg/structx"
|
"apsClient/service/plc_address"
|
"apsClient/utils/file"
|
"encoding/json"
|
"fmt"
|
"github.com/jinzhu/gorm"
|
"github.com/mitchellh/mapstructure"
|
"github.com/spf13/cast"
|
"strings"
|
)
|
|
type ReceivedMessage struct {
|
Topic string
|
Message []byte
|
}
|
|
var ReceivedMessageChan chan *ReceivedMessage
|
|
func init() {
|
ReceivedMessageChan = make(chan *ReceivedMessage, 1000)
|
}
|
|
type MsgHandler interface {
|
HandleMessage(data []byte) (err error)
|
}
|
|
type ScheduleTask struct {
|
}
|
|
func (slf *ScheduleTask) HandleMessage(data []byte) (err error) {
|
logx.Infof("get an message :%s", data)
|
var tasks = make([]*common.DeliverScheduleTask, 0)
|
err = json.Unmarshal(data, &tasks)
|
if err != nil {
|
logx.Errorf("ScheduleTask HandleMessage Unmarshal json err: %v", err.Error())
|
return err
|
}
|
for _, task := range tasks {
|
oldWorkOrder, err := model.NewOrderSearch(nil).SetWorkOrderId(task.WorkOrder.WorkOrderID).First()
|
if oldWorkOrder != nil && oldWorkOrder.WorkOrderID != "" && oldWorkOrder.Status != model.OrderStatusWaitProcess { //已开始的工序不可以修改
|
continue
|
}
|
|
procedureRecords := make([]*model.Procedures, 0, len(task.Procedures))
|
for _, procedure := range task.Procedures {
|
procedureRecord := model.Procedures{
|
WorkOrderID: task.WorkOrder.WorkOrderID,
|
OrderID: task.WorkOrder.OrderID,
|
DeviceID: procedure.DeviceID,
|
ProcedureID: procedure.ProcedureID,
|
StartTime: procedure.StartTime,
|
EndTime: procedure.EndTime,
|
Status: model.ProcedureStatusWaitProcess,
|
Channel: procedure.Channel,
|
ProceduresInfo: common.ProductProcedure{},
|
}
|
procedureData, err := json.Marshal(procedure)
|
if err != nil {
|
return err
|
}
|
procedureRecord.ProcedureData = string(procedureData)
|
procedureRecords = append(procedureRecords, &procedureRecord)
|
}
|
|
var orderRecord model.Order
|
err = structx.AssignTo(task.WorkOrder, &orderRecord)
|
orderRecord.Status = model.OrderStatusWaitProcess
|
if err != nil {
|
logx.Errorf(" structx.Assign task.Order to orderRecord err: %v", err.Error())
|
return err
|
}
|
|
err = model.WithTransaction(func(db *gorm.DB) error {
|
var err error
|
if oldWorkOrder != nil && oldWorkOrder.WorkOrderID != "" {
|
if err = model.NewOrderSearch(db).SetId(oldWorkOrder.ID).Updates(&orderRecord); err != nil {
|
return err
|
}
|
for _, procedure := range procedureRecords {
|
err = model.NewProceduresSearch(db).SetWorkOrderId(procedure.WorkOrderID).SetProcedureId(procedure.ProcedureID).Upsert(procedure)
|
if err != nil {
|
return err
|
}
|
}
|
} else {
|
if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil {
|
return err
|
}
|
if err = model.NewProceduresSearch(db).CreateBatch(procedureRecords); err != nil {
|
return err
|
}
|
}
|
return nil
|
})
|
if err != nil {
|
logx.Errorf(" save task message error err: %v", err)
|
return err
|
}
|
}
|
return nil
|
}
|
|
type PlcAddress struct {
|
Topic string
|
}
|
|
func (slf *PlcAddress) HandleMessage(data []byte) (err error) {
|
logx.Infof("get an PlcAddress message :%s", data)
|
var resp = new(common.ResponsePlcAddress)
|
err = json.Unmarshal(data, &resp)
|
if err != nil {
|
logx.Errorf("ScheduleTask HandleMessage Unmarshal json err: %v", err.Error())
|
return nil
|
}
|
if resp.DeviceId != conf.Conf.System.DeviceId {
|
return nil
|
}
|
//通知回复收到
|
ReceivedMessageChan <- &ReceivedMessage{
|
Topic: slf.Topic,
|
Message: data,
|
}
|
if len(resp.KeyData) != 0 && len(resp.AddressData) != 0 {
|
//写入到文件
|
err = file.WriteFile(fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataKeyFileName), resp.KeyData)
|
if err != nil {
|
return err
|
}
|
err = file.WriteFile(fmt.Sprintf("%s%s", constvar.PlcAddressDataPath, constvar.PlcAddressDataValueFileName), resp.AddressData)
|
if err != nil {
|
return err
|
}
|
//写入到内存
|
keyString := string(resp.KeyData)
|
addressString := string(resp.AddressData)
|
|
keys := strings.Split(keyString, "\n")
|
addresses := strings.Split(addressString, "\n")
|
if len(keys) != len(addresses) {
|
logx.Error("plc address message error: key length not equal address length")
|
return nil
|
}
|
for i := 0; i < len(keys); i++ {
|
key := strings.ReplaceAll(keys[i], "\r", "")
|
address := cast.ToInt(strings.ReplaceAll(addresses[i], "\r", ""))
|
plc_address.Set(key, address)
|
logx.Infof("plc address set ok: key:%v, address:%v", key, address)
|
}
|
}
|
|
if resp.PlcConfig.Method == "" {
|
return nil
|
}
|
|
resp.PlcConfig.Method.ConvertToEnglish()
|
|
var record model.DevicePlc
|
err = mapstructure.Decode(resp.PlcConfig, &record)
|
if err != nil {
|
logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err)
|
}
|
err = mapstructure.Decode(resp.PlcAddressList, &record.Details)
|
if err != nil {
|
logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err)
|
}
|
detail, err := json.Marshal(record.Details)
|
record.Detail = string(detail)
|
record.ID = 1
|
err = model.NewDevicePlcSearch().SetId(record.ID).Save(&record)
|
if err != nil {
|
return err
|
}
|
return nil
|
}
|
|
type ProcessParams struct {
|
Topic string
|
}
|
|
func (slf *ProcessParams) HandleMessage(data []byte) (err error) {
|
logx.Infof("get an process params message :%s", data)
|
var processModel model.ProcessModel
|
err = json.Unmarshal(data, &processModel)
|
if err != nil {
|
logx.Infof("unmarshal process params sync err :%s", err)
|
return err
|
}
|
if processModel.DeviceId != conf.Conf.System.DeviceId { //不是发给本设备的消息
|
return nil
|
}
|
if processModel.Params != "" {
|
err = model.NewProcessModelSearch().Create(&processModel)
|
if err != nil {
|
logx.Infof("save process params err :%s", err)
|
}
|
}
|
//通知回复收到
|
ReceivedMessageChan <- &ReceivedMessage{
|
Topic: slf.Topic,
|
Message: data,
|
}
|
return nil
|
}
|
|
type ProcessParamsSync struct {
|
Topic string
|
}
|
|
func (slf *ProcessParamsSync) HandleMessage(data []byte) (err error) {
|
logx.Infof("get an process params sync message :%s", data)
|
var processModelMsg common.ResponseProcessParams
|
err = json.Unmarshal(data, &processModelMsg)
|
if err != nil {
|
logx.Infof("unmarshal process params sync err :%s", err)
|
return err
|
}
|
processModel := model.ProcessModel{
|
Number: processModelMsg.Number,
|
Product: processModelMsg.Product,
|
Procedure: processModelMsg.Procedure,
|
Params: processModelMsg.Params,
|
IsNew: true,
|
}
|
err = model.WithTransaction(func(db *gorm.DB) error {
|
err = model.NewProcessModelSearch().SetOrm(db).SetProduct(processModel.Product).SetProcedure(processModel.Procedure).
|
UpdateByMap(map[string]interface{}{"is_new": 0})
|
if err != nil {
|
return err
|
}
|
return model.NewProcessModelSearch().SetOrm(db).Create(&processModel)
|
})
|
|
if err != nil {
|
logx.Infof("save process params sync err :%s", err)
|
return err
|
}
|
return nil
|
}
|
|
type DeviceUpdate struct {
|
Topic string
|
}
|
|
func (slf *DeviceUpdate) HandleMessage(data []byte) (err error) {
|
logx.Infof("get a device update message :%s", data)
|
var msg common.DeviceMsg
|
err = json.Unmarshal(data, &msg)
|
if err != nil {
|
logx.Infof("unmarshal device update msg err :%s", err)
|
return err
|
}
|
|
procedures := make([]string, 0, len(msg.DeviceProcedureAttr))
|
for _, attr := range msg.DeviceProcedureAttr {
|
procedures = append(procedures, attr.ProcedureName)
|
}
|
|
deviceRecord := &model.Device{
|
DeviceID: msg.ID,
|
Procedures: strings.Join(procedures, ","),
|
ExtChannelAmount: msg.ExtChannelAmount,
|
}
|
|
oldRecord, err := model.NewDeviceSearch().SetDeviceId(msg.ID).First()
|
if err == gorm.ErrRecordNotFound {
|
err = model.NewDeviceSearch().Create(deviceRecord)
|
} else if oldRecord.ExtChannelAmount != deviceRecord.ExtChannelAmount || oldRecord.Procedures != deviceRecord.Procedures {
|
err = model.NewDeviceSearch().SetDeviceId(msg.ID).Save(deviceRecord)
|
}
|
|
if err != nil {
|
logx.Infof("save device record err :%s", err)
|
return err
|
}
|
|
if msg.PlcConfig.Method == "" {
|
return nil
|
}
|
|
msg.PlcConfig.Method.ConvertToEnglish()
|
var record model.DevicePlc
|
err = mapstructure.Decode(msg.PlcConfig, &record)
|
if err != nil {
|
logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err)
|
}
|
err = mapstructure.Decode(msg.PlcAddressList, &record.Details)
|
if err != nil {
|
logx.Infof("decode from PlcAddressList to DevicePlc Details err:%v", err)
|
}
|
detail, err := json.Marshal(record.Details)
|
record.Detail = string(detail)
|
record.ID = 1
|
err = model.NewDevicePlcSearch().SetId(record.ID).Save(&record)
|
if err != nil {
|
return err
|
}
|
|
return nil
|
}
|
|
type PullDataResponse struct {
|
Topic string
|
}
|
|
func (slf *PullDataResponse) HandleMessage(data []byte) (err error) {
|
logx.Infof("get a pull data response message :%s", data)
|
var pullDataResponse common.MsgPullDataResponse
|
err = json.Unmarshal(data, &pullDataResponse)
|
if err != nil {
|
logx.Infof("unmarshal msg err :%s", err)
|
return err
|
}
|
switch pullDataResponse.DataType {
|
case common.PullDataTypeProcessModel:
|
err = slf.DealProcessModelData(pullDataResponse.Data)
|
case common.PullDataTypeDevice:
|
err = slf.DealDeviceData(pullDataResponse.Data)
|
}
|
if err != nil {
|
logx.Infof("process pull data err :%s", err)
|
return err
|
}
|
return nil
|
}
|
|
func (slf *PullDataResponse) DealProcessModelData(data interface{}) error {
|
var processModels []*model.ProcessModel
|
err := mapstructure.Decode(data, &processModels)
|
if err != nil {
|
return err
|
}
|
numbers := make([]string, 0, len(processModels))
|
for _, processModel := range processModels {
|
numbers = append(numbers, processModel.Number)
|
}
|
existsProcessModels, err := model.NewProcessModelSearch().SetIsNew(true).SetNumbers(numbers).FindNotTotal()
|
if err != nil {
|
return err
|
}
|
|
existsProcessModelsMap := make(map[string]struct{}, len(existsProcessModels))
|
for _, processModel := range existsProcessModels {
|
existsProcessModelsMap[processModel.Number] = struct{}{}
|
}
|
|
for _, processModel := range processModels {
|
if _, exists := existsProcessModelsMap[processModel.Number]; exists {
|
continue
|
}
|
err = model.WithTransaction(func(db *gorm.DB) error {
|
err = model.NewProcessModelSearch().SetOrm(db).SetProcedure(processModel.Procedure).SetProduct(processModel.Product).SetIsNew(true).UpdateByMap(map[string]interface{}{"is_new": 0})
|
if err != nil {
|
return err
|
}
|
processModel.IsNew = true
|
return model.NewProcessModelSearch().SetOrm(db).Create(processModel)
|
})
|
if err != nil {
|
return err
|
}
|
}
|
return nil
|
}
|
|
func (slf *PullDataResponse) DealDeviceData(data interface{}) error {
|
var devices []*model.Device
|
err := mapstructure.Decode(data, &devices)
|
if err != nil {
|
return err
|
}
|
numbers := make([]string, 0, len(devices))
|
for _, item := range devices {
|
numbers = append(numbers, item.DeviceID)
|
}
|
existsDevices, err := model.NewDeviceSearch().SetDeviceIds(numbers).FindNotTotal()
|
if err != nil {
|
return err
|
}
|
|
existsDeviceMap := make(map[string]*model.Device, len(existsDevices))
|
for _, device := range existsDevices {
|
existsDeviceMap[device.DeviceID] = device
|
}
|
|
for _, device := range devices {
|
if oldDevice, exists := existsDeviceMap[device.DeviceID]; exists {
|
if oldDevice.ExtChannelAmount != device.ExtChannelAmount || //todo to be continued
|
oldDevice.Procedures != device.Procedures {
|
err = model.NewDeviceSearch().SetDeviceId(device.DeviceID).Save(device)
|
}
|
} else {
|
err = model.NewDeviceSearch().Create(device)
|
}
|
}
|
return err
|
}
|
