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