From 7438f17c0b3a9f79ee5890c554bedc64c362859b Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 13 十月 2023 14:04:59 +0800
Subject: [PATCH] 销售计划简单数据权限
---
service/subOrder.go | 76 +++++++++++++++++++++++++++----------
1 files changed, 55 insertions(+), 21 deletions(-)
diff --git a/service/subOrder.go b/service/subOrder.go
index cf6df59..ce0a337 100644
--- a/service/subOrder.go
+++ b/service/subOrder.go
@@ -3,12 +3,13 @@
import (
"aps_crm/model"
"aps_crm/pkg/ecode"
+ "aps_crm/pkg/mysqlx"
)
type SubOrderService struct{}
func (SubOrderService) AddSubOrder(subOrder *model.SubOrder) int {
- err := model.NewSubOrderSearch().Create(subOrder)
+ err := model.NewSubOrderSearch(nil).Create(subOrder)
if err != nil {
return ecode.SubOrderExist
}
@@ -16,39 +17,72 @@
return ecode.OK
}
-func (SubOrderService) DeleteSubOrder(id int) int {
- _, err := model.NewSubOrderSearch().SetId(id).Find()
+func (SubOrderService) UpdateSubOrder(subOrder *model.SubOrder) int {
+ // check subOrder exist
+ tmp, err := model.NewSubOrderSearch(nil).SetId(subOrder.Id).Find()
if err != nil {
return ecode.SubOrderNotExist
}
- err = model.NewSubOrderSearch().SetId(id).Delete()
- if err != nil {
- return ecode.SubOrderNotExist
+ if len(subOrder.Products) == 0 {
+ err = model.NewSubOrderSearch(nil).SetId(subOrder.Id).Update(subOrder)
+ if err != nil {
+ return ecode.SubOrderSetErr
+ }
+ } else {
+ tx := mysqlx.GetDB().Begin()
+ err = tx.Model(tmp).Association("Products").Clear()
+ if err != nil {
+ tx.Rollback()
+ return ecode.SubOrderSetErr
+ }
+
+ err = model.NewSubOrderSearch(tx).SetId(subOrder.Id).Update(subOrder)
+ if err != nil {
+ tx.Rollback()
+ return ecode.SubOrderSetErr
+ }
+
+ for _, product := range subOrder.Products {
+ if product.Id == 0 {
+ err = model.NewProductSearch(tx).Create(&product)
+ if err != nil {
+ tx.Rollback()
+ return ecode.SubOrderSetErr
+ }
+ } else {
+ err = model.NewProductSearch(tx).SetId(product.Id).Update(&product)
+ if err != nil {
+ tx.Rollback()
+ return ecode.SubOrderSetErr
+ }
+ }
+ err = tx.Model(&tmp).Association("Products").Append(&product)
+ if err != nil {
+ tx.Rollback()
+ return ecode.SubOrderSetErr
+ }
+ }
+ tx.Commit()
}
+
return ecode.OK
}
-func (SubOrderService) GetSubOrderList() ([]*model.SubOrder, int) {
- list, err := model.NewSubOrderSearch().FindAll()
+func (SubOrderService) GetSubOrderList(page, pageSize int, data map[string]interface{}) ([]*model.SubOrder, int64, int) {
+ // get contact list
+ contacts, total, err := model.NewSubOrderSearch(nil).SetSearchMap(data).SetPage(page, pageSize).SetSearchMap(data).FindAll()
if err != nil {
- return nil, ecode.SubOrderListErr
+ return nil, 0, ecode.SubOrderListErr
}
-
- return list, ecode.OK
+ return contacts, total, ecode.OK
}
-func (SubOrderService) UpdateSubOrder(subOrder *model.SubOrder) int {
- // check subOrder exist
- _, err := model.NewSubOrderSearch().SetId(subOrder.Id).Find()
+func (SubOrderService) DeleteSubOrder(ids []int) int {
+ // delete client
+ err := model.NewSubOrderSearch(nil).SetIds(ids).Delete()
if err != nil {
- return ecode.SubOrderNotExist
+ return ecode.SubOrderDeleteErr
}
-
- err = model.NewSubOrderSearch().SetId(subOrder.Id).Update(subOrder)
- if err != nil {
- return ecode.SubOrderSetErr
- }
-
return ecode.OK
}
--
Gitblit v1.8.0