From e8d604b28abd319c63c8efe398f602eff66aa5ef Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期一, 16 十月 2023 11:12:51 +0800
Subject: [PATCH] 添加一级页面的记录时。负责人如果为空，且为普通用户，取当前用户id

---
 api/v1/client.go          |    7 +++
 api/v1/subOrder.go        |    7 +++
 api/v1/masterOrder.go     |    7 +++
 api/v1/contact.go         |    7 +++
 api/v1/contract.go        |    7 +++
 api/v1/serviceOrder.go    |    6 +++
 api/v1/quotation.go       |    7 +++
 api/v1/salesReturn.go     |    7 +++
 api/v1/saleChance.go      |    7 +++
 api/v1/salesRefund.go     |    7 +++
 api/v1/salesLeads.go      |    7 +++
 api/v1/serviceContract.go |    7 +++
 api/v1/salesDetails.go    |    7 +++
 api/v1/followRecord.go    |    7 +++
 api/v1/serviceFollowup.go |    6 +++
 15 files changed, 103 insertions(+), 0 deletions(-)

diff --git a/api/v1/client.go b/api/v1/client.go
index 40375c9..bac72a8 100644
--- a/api/v1/client.go
+++ b/api/v1/client.go
@@ -28,6 +28,13 @@
 		return
 	}
 
+	if params.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			params.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode, client := checkClientParams(params.Client)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/contact.go b/api/v1/contact.go
index 8b4c016..6c60e46 100644
--- a/api/v1/contact.go
+++ b/api/v1/contact.go
@@ -34,6 +34,13 @@
 		return
 	}
 
+	if params.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			params.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = contactService.AddContact(&contact)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/contract.go b/api/v1/contract.go
index f1702f7..8e05fa5 100644
--- a/api/v1/contract.go
+++ b/api/v1/contract.go
@@ -34,6 +34,13 @@
 		return
 	}
 
+	if contract.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			contract.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = contractService.AddContract(&contract)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/followRecord.go b/api/v1/followRecord.go
index 0fe3939..d64b85a 100644
--- a/api/v1/followRecord.go
+++ b/api/v1/followRecord.go
@@ -36,6 +36,13 @@
 		return
 	}
 
+	if followRecord.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			followRecord.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = followRecordService.AddFollowRecord(followRecord)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/masterOrder.go b/api/v1/masterOrder.go
index 246e9b8..373078f 100644
--- a/api/v1/masterOrder.go
+++ b/api/v1/masterOrder.go
@@ -34,6 +34,13 @@
 		return
 	}
 
+	if masterOrder.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			masterOrder.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = masterOrderService.AddMasterOrder(&masterOrder)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/quotation.go b/api/v1/quotation.go
index a6f8790..7beeb3b 100644
--- a/api/v1/quotation.go
+++ b/api/v1/quotation.go
@@ -34,6 +34,13 @@
 		return
 	}
 
+	if quotation.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			quotation.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = quotationService.AddQuotation(&quotation)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/saleChance.go b/api/v1/saleChance.go
index f5e57f2..791ce98 100644
--- a/api/v1/saleChance.go
+++ b/api/v1/saleChance.go
@@ -45,6 +45,13 @@
 		}
 	}
 
+	if saleChance.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			saleChance.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = saleChanceService.AddSaleChance(&saleChance)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/salesDetails.go b/api/v1/salesDetails.go
index 89831f6..350e9c2 100644
--- a/api/v1/salesDetails.go
+++ b/api/v1/salesDetails.go
@@ -35,6 +35,13 @@
 		return
 	}
 
+	if salesDetails.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			salesDetails.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = salesDetailsService.AddSalesDetails(&salesDetails)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/salesLeads.go b/api/v1/salesLeads.go
index 0812a5a..e35a85d 100644
--- a/api/v1/salesLeads.go
+++ b/api/v1/salesLeads.go
@@ -28,6 +28,13 @@
 		return
 	}
 
+	if params.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			params.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode, salesLeads := checkSalesLeadsParams(params.SalesLeads)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/salesRefund.go b/api/v1/salesRefund.go
index 1d630b3..cbce386 100644
--- a/api/v1/salesRefund.go
+++ b/api/v1/salesRefund.go
@@ -36,6 +36,13 @@
 		return
 	}
 
+	if salesRefund.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			salesRefund.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = salesRefundService.AddSalesRefund(&salesRefund)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/salesReturn.go b/api/v1/salesReturn.go
index f53260f..bf1fdbd 100644
--- a/api/v1/salesReturn.go
+++ b/api/v1/salesReturn.go
@@ -37,6 +37,13 @@
 		return
 	}
 
+	if salesReturn.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			salesReturn.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode = salesReturnService.AddSalesReturn(&salesReturn)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/serviceContract.go b/api/v1/serviceContract.go
index dd32f74..d6b13c1 100644
--- a/api/v1/serviceContract.go
+++ b/api/v1/serviceContract.go
@@ -35,6 +35,13 @@
 		return
 	}
 
+	if params.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			params.MemberId = userInfo.CrmUserId
+		}
+	}
+
 	errCode := serviceContractService.AddServiceContract(serviceContract)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/serviceFollowup.go b/api/v1/serviceFollowup.go
index 116acf7..39af767 100644
--- a/api/v1/serviceFollowup.go
+++ b/api/v1/serviceFollowup.go
@@ -28,6 +28,12 @@
 		return
 	}
 
+	if params.MemberId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			params.MemberId = userInfo.CrmUserId
+		}
+	}
 	errCode, serviceFollowup := checkServiceFollowupParams(params.ServiceFollowup)
 	if errCode != ecode.OK {
 		ctx.Fail(errCode)
diff --git a/api/v1/serviceOrder.go b/api/v1/serviceOrder.go
index aef80c2..faef2d6 100644
--- a/api/v1/serviceOrder.go
+++ b/api/v1/serviceOrder.go
@@ -30,6 +30,12 @@
 		return
 	}
 
+	if params.ServiceManId == 0 {
+		userInfo := utils.GetUserInfo(c)
+		if userInfo.UserType == constvar.UserTypeSub {
+			params.ServiceManId = userInfo.CrmUserId
+		}
+	}
 	serviceOrder := new(model.ServiceOrder)
 	if err := structx.AssignTo(params, serviceOrder); err != nil {
 		ctx.Fail(ecode.ParamsErr)
diff --git a/api/v1/subOrder.go b/api/v1/subOrder.go
index 17372fc..b4595ee 100644
--- a/api/v1/subOrder.go
+++ b/api/v1/subOrder.go
@@ -34,6 +34,13 @@
 		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)

--
Gitblit v1.8.0