From 7d80867c1d2d8340da8e106b4af2ceaba3fc9fc1 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 19 三月 2024 19:16:50 +0800
Subject: [PATCH] 增加发货完成按钮

---
 service/assign.go |   24 +++++++++++++++++++++---
 1 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/service/assign.go b/service/assign.go
index 28d7a05..88e789e 100644
--- a/service/assign.go
+++ b/service/assign.go
@@ -1,6 +1,9 @@
 package service
 
-import "aps_crm/pkg/ecode"
+import (
+	"aps_crm/model"
+	"aps_crm/pkg/ecode"
+)
 
 type AssignService struct{}
 
@@ -8,11 +11,26 @@
 	Assign(id int) error
 }
 
-func (AssignService) Assign(memberId, id int, modelType string) int {
+func (AssignService) Assign(memberId int, ids []int, modelType string) int {
+	// check member exist
+	if memberId != 0 {
+		_, err := model.NewUserSearch(nil).SetUserId(memberId).First()
+		if err != nil {
+			return ecode.UserNotExist
+		}
+	}
+
 	var errCode int
 	switch modelType {
 	case "client":
-		errCode = ClientService{}.Assign(id, memberId)
+		errCode = ClientService{}.Assign(ids, memberId)
+	case "salesLead":
+		errCode = SalesLeadsService{}.Assign(ids, memberId)
+	case "contact":
+		errCode = ContactService{}.Assign(ids, memberId)
+	case "followRecord":
+		errCode = FollowRecordService{}.Assign(ids, memberId)
+
 	default:
 		errCode = ecode.AssignWrongModelType
 	}

--
Gitblit v1.8.0