From 63b531279a8ab1d609910e7db53add83bbc0b97a Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 17 十月 2023 20:21:01 +0800
Subject: [PATCH] 添加默认数据

---
 api/v1/subOrder.go |   25 +++++++++++++++++++++++++
 1 files changed, 25 insertions(+), 0 deletions(-)

diff --git a/api/v1/subOrder.go b/api/v1/subOrder.go
index d6bdf6c..b4595ee 100644
--- a/api/v1/subOrder.go
+++ b/api/v1/subOrder.go
@@ -1,11 +1,13 @@
 package v1
 
 import (
+	"aps_crm/constvar"
 	"aps_crm/model"
 	"aps_crm/model/request"
 	"aps_crm/model/response"
 	"aps_crm/pkg/contextx"
 	"aps_crm/pkg/ecode"
+	"aps_crm/utils"
 	"github.com/gin-gonic/gin"
 )
 
@@ -32,12 +34,26 @@
 		return
 	}
 
+	if subOrder.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			subOrder.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = subOrderService.AddSubOrder(&subOrder)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
 		return
 	}
 
+	if params.CodeRule.Method == 1 {
+		autoCode := model.GetAutoCode(subOrder.Id, &params.CodeRule)
+		m := map[string]interface{}{
+			"number": autoCode,
+		}
+		_ = model.NewSubOrderSearch(nil).SetId(subOrder.Id).UpdateByMap(m)
+	}
 	ctx.Ok()
 }
 
@@ -113,6 +129,7 @@
 	newSubOrder.MemberId = subOrder.MemberId
 
 	newSubOrder.Products = subOrder.Product
+	newSubOrder.CodeStandID = subOrder.CodeStandID
 
 	return ecode.OK, newSubOrder
 }
@@ -132,6 +149,14 @@
 		return
 	}
 
+	userInfo := utils.GetUserInfo(c)
+	if userInfo.UserType == constvar.UserTypeSub {
+		if params.SearchMap == nil {
+			params.SearchMap = make(map[string]interface{}, 0)
+		}
+		params.SearchMap["member_id"] = userInfo.CrmUserId
+	}
+
 	subOrders, total, errCode := subOrderService.GetSubOrderList(params.Page, params.PageSize, params.SearchMap)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)

--
Gitblit v1.8.0