From 7b3074cf1f5c7af8319ccb02fac335d18f779a48 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 08 九月 2023 09:49:07 +0800
Subject: [PATCH] fix
---
nsq/msg_handler.go | 25 ++++++++++++++++---------
1 files changed, 16 insertions(+), 9 deletions(-)
diff --git a/nsq/msg_handler.go b/nsq/msg_handler.go
index 6e8df60..d3b08ec 100644
--- a/nsq/msg_handler.go
+++ b/nsq/msg_handler.go
@@ -79,16 +79,23 @@
}
err = model.WithTransaction(func(db *gorm.DB) error {
- if err = model.NewOrderSearch(db).SetWorkOrderId(task.WorkOrder.WorkOrderID).Delete(); err != nil {
- return err
+ if oldWorkOrder != nil {
+ 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)
}
- if err = model.NewOrderSearch(db).Save(&orderRecord); err != nil {
- return err
- }
- if err = model.NewProceduresSearch(db).SetProcedureIds(procedureIds).Delete(); err != nil {
- return err
- }
- return model.NewProceduresSearch(db).CreateBatch(procedureRecords)
+ return nil
})
if err != nil {
logx.Errorf(" save task message error err: %v", err.Error())
--
Gitblit v1.8.0