From 8d2a95fc0eeabe1b13d0a914c9ec2845d42c0be3 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期四, 19 十月 2023 11:32:57 +0800
Subject: [PATCH] 添加主从serf切换事件
---
nsq/msg_handler.go | 325 +++++++++++++++++++++++++++++++++++++++++++++++++++---
1 files changed, 307 insertions(+), 18 deletions(-)
diff --git a/nsq/msg_handler.go b/nsq/msg_handler.go
index 495b6ef..1db651e 100644
--- a/nsq/msg_handler.go
+++ b/nsq/msg_handler.go
@@ -1,11 +1,32 @@
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)
@@ -15,35 +36,303 @@
}
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))
+ 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("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).Upsert(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)
+ 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 processModel model.ProcessModel
+ err = json.Unmarshal(data, &processModel)
+ if err != nil {
+ logx.Infof("unmarshal process params sync err :%s", err)
+ return err
+ }
+ 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
+ }
+ processModel.IsNew = true
+ 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 device common.Device
+ err = json.Unmarshal(data, &device)
+ if err != nil {
+ logx.Infof("unmarshal device update msg err :%s", err)
+ return err
+ }
+
+ procedures := make([]string, 0, len(device.DeviceProcedureAttr))
+ for _, attr := range device.DeviceProcedureAttr {
+ procedures = append(procedures, attr.ProcedureName)
+ }
+
+ deviceRecord := &model.Device{
+ DeviceID: device.ID,
+ Procedures: strings.Join(procedures, ","),
+ ExtChannelAmount: device.ExtChannelAmount,
+ }
+
+ oldRecord, err := model.NewDeviceSearch().SetDeviceId(device.ID).First()
+ if err == gorm.ErrRecordNotFound {
+ err = model.NewDeviceSearch().Create(deviceRecord)
+ } else {
+ deviceRecord.ID = oldRecord.ID
+ err = model.NewDeviceSearch().Save(deviceRecord)
+ }
+
+ if err != nil {
+ logx.Infof("save device record err :%s", err)
+ 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)
+
+ }
+ 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
}
}
--
Gitblit v1.8.0