From 9a7e0a7da01a9f9625ceaca0c61a59c540c6438f Mon Sep 17 00:00:00 2001
From: wangpengfei <274878379@qq.com>
Date: 星期五, 18 八月 2023 17:32:10 +0800
Subject: [PATCH] fix

---
 service/assign.go |   77 ++++++++++++++++++++------------------
 1 files changed, 41 insertions(+), 36 deletions(-)

diff --git a/service/assign.go b/service/assign.go
index e05f20a..4f6d526 100644
--- a/service/assign.go
+++ b/service/assign.go
@@ -1,36 +1,41 @@
-package service
-
-import (
-	"aps_crm/model"
-	"aps_crm/pkg/ecode"
-)
-
-type AssignService struct{}
-
-type Assign interface {
-	Assign(id int) error
-}
-
-func (AssignService) Assign(memberId, id int, modelType string) int {
-	// check member exist
-	_, 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)
-	case "salesLead":
-		errCode = SalesLeadsService{}.Assign(id, memberId)
-	default:
-		errCode = ecode.AssignWrongModelType
-	}
-
-	if errCode != ecode.OK {
-		return errCode
-	}
-
-	return ecode.OK
-}
+package service
+
+import (
+	"aps_crm/model"
+	"aps_crm/pkg/ecode"
+)
+
+type AssignService struct{}
+
+type Assign interface {
+	Assign(id int) error
+}
+
+func (AssignService) Assign(memberId int, ids []int, modelType string) int {
+	// check member exist
+	_, err := model.NewUserSearch(nil).SetUserId(memberId).First()
+	if err != nil {
+		return ecode.UserNotExist
+	}
+
+	var errCode int
+	switch modelType {
+	case "client":
+		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
+	}
+
+	if errCode != ecode.OK {
+		return errCode
+	}
+
+	return ecode.OK
+}

--
Gitblit v1.8.0