From 13b9eb612f28bed4df5fc0241e27733f539bc2e3 Mon Sep 17 00:00:00 2001
From: wangpengfei <274878379@qq.com>
Date: 星期一, 31 七月 2023 15:32:59 +0800
Subject: [PATCH] add

---
 service/assign.go |   13 ++++++++++++-
 1 files changed, 12 insertions(+), 1 deletions(-)

diff --git a/service/assign.go b/service/assign.go
index 28d7a05..e05f20a 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{}
 
@@ -9,10 +12,18 @@
 }
 
 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
 	}

--
Gitblit v1.8.0