三菱协议更改

zhangqian

2023-09-12 b555874608be2e4bb40cceba60b497c04caef842

 nsq/msg_handler.go

@@ -1,11 +1,31 @@
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/spf13/cast"
   "gorm.io/gorm"
   "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)
@@ -15,37 +35,172 @@
}

func (slf *ScheduleTask) HandleMessage(data []byte) (err error) {
   fmt.Println(string(data))

   var tasks = make([]*DeliverScheduleTask, 0)

   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 {
      taskRecord := model.ScheduleTask{
         Id:          0,
         OrderId:     task.Order.OrderID,
         ProductId:   task.Order.ProductID,
         ProductName: task.Order.ProductName,
         Amount:      task.Order.Amount,
         Unit:        task.Order.Unit,
         StartTime:   task.Order.StartTime,
         EndTime:     task.Order.EndTime,
      oldWorkOrder, err := model.NewOrderSearch(nil).SetWorkOrderId(task.WorkOrder.WorkOrderID).First()
      if oldWorkOrder != nil && oldWorkOrder.WorkOrderID != "" && oldWorkOrder.Status != model.OrderStatusWaitProcess { //已开始的工序不可以修改
         continue
      }
      jsonStr, err := json.Marshal(task)

      procedureRecords := make([]*model.Procedures, 0, len(task.Procedures))
      procedureIds := make([]string, 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,
            ProceduresInfo: common.ProductProcedure{},
         }
         procedureData, err := json.Marshal(procedure)
         if err != nil {
            return err
         }
         procedureRecord.ProcedureData = string(procedureData)
         procedureRecords = append(procedureRecords, &procedureRecord)
         procedureIds = append(procedureIds, procedure.ProcedureID)
      }

      var orderRecord model.Order
      err = structx.AssignTo(task.WorkOrder, &orderRecord)
      orderRecord.Status = model.OrderStatusWaitProcess
      if err != nil {
         logx.Errorf("ScheduleTask HandleMessage Marshal err: %v, old: %#v", err.Error(), task)
         logx.Errorf(" structx.Assign task.Order to orderRecord err: %v", err.Error())
         return err
      }
      taskRecord.Data = string(jsonStr)
      err = model.NewScheduleTaskSearch(nil).Create(&taskRecord)

      err = model.WithTransaction(func(db *gorm.DB) 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).Updates(procedure)
               if err != nil {
                  return err
               }
            }
         } else {
            if err = model.NewOrderSearch(db).Create(&orderRecord); err != nil {
               return err
            }
            return model.NewProceduresSearch(db).CreateBatch(procedureRecords)
         }
         return nil
      })
      if err != nil {
         logx.Errorf("ScheduleTask HandleMessage Create taskRecord err: %v,  record: %#v", err.Error(), taskRecord)
         logx.Errorf(" save task message error err: %v", err.Error())
         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 {
      return nil
   }
   //写入到文件
   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)
   }
   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 processModel model.ProcessModel
   err = json.Unmarshal(data, &processModel)
   if err != nil {
      logx.Infof("unmarshal process params sync err :%s", err)
      return err
   }
   err = model.NewProcessModelSearch().Save(&processModel)
   if err != nil {
      logx.Infof("save process params sync err :%s", err)
      return err
   }
   return nil
}