From 957fcc4df076728f7f9ffefe97eeb8b46588253d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8C=83=E4=BF=8A=E6=88=90?= <gakkiismywife1@163.com>
Date: Tue, 26 Sep 2023 13:56:34 +0800
Subject: [PATCH] =?UTF-8?q?=E6=A0=87=E8=AE=B0=E7=94=A8=E6=88=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/admin/apis/usermanage/user.go | 22 ++++++++++++++++++++++
 app/admin/models/user.go          | 12 ++++++++++++
 app/admin/router/usermanage.go    |  1 +
 3 files changed, 35 insertions(+)

diff --git a/app/admin/apis/usermanage/user.go b/app/admin/apis/usermanage/user.go
index a072cbe..823b706 100644
--- a/app/admin/apis/usermanage/user.go
+++ b/app/admin/apis/usermanage/user.go
@@ -903,3 +903,25 @@ func ExpireMemberSmsSendRecordList(c *gin.Context) {
 	app.OK(c, resp, "")
 	return
 }
+
+type UserMarkRequest struct {
+	Uid  []uint32 `json:"uid"`
+	Mark int8     `json:"mark"`
+}
+
+func MarkUser(c *gin.Context) {
+	var req = new(UserMarkRequest)
+
+	err := c.ShouldBindJSON(req)
+	if err != nil || (req.Mark != models.UNMARK && req.Mark != models.MARK) {
+		app.Error(c, http.StatusBadRequest, err, "参数错误")
+		return
+	}
+
+	err = models.MarkUser(req.Uid, req.Mark)
+	if err != nil {
+		app.Error(c, http.StatusInternalServerError, err, "更新失败")
+		return
+	}
+	app.OK(c, nil, "更新成功")
+}
diff --git a/app/admin/models/user.go b/app/admin/models/user.go
index 2031415..985081f 100644
--- a/app/admin/models/user.go
+++ b/app/admin/models/user.go
@@ -58,6 +58,7 @@ type UserInfo struct {
 	ShopAssistantName     string    `json:"shop_assistant_name"` // 店员名称
 	RenewalTime           time.Time `json:"renewal_time"`
 	RenewalMemberLevel    uint32    `json:"renewal_member_level"`
+	Mark                  int8      `json:"mark"` //标记
 
 	Store          *Store      `json:"store" gorm:"-"`
 	OrderCards     []OrderCard `json:"order_cards" gorm:"-"`
@@ -73,6 +74,11 @@ const (
 	UserTypeShopAssistant = 2 // 店员
 )
 
+const (
+	UNMARK = iota
+	MARK
+)
+
 const (
 	OpenMemberChannelStorePromotion = "store_promotion" // 门店推荐
 	OpenMemberChannelUserInvite     = "user_invite"
@@ -2412,3 +2418,9 @@ func (m *ExpireMemberSmsSendListReq) List() (*ExpireMemberSmsSendListResp, error
 //	}
 //	return url + fileName, nil
 //}
+
+func MarkUser(uid []uint32, mark int8) error {
+	return orm.Eloquent.Table("user").
+		Where("uid in (?)", uid).
+		Update("mark", mark).Error
+}
diff --git a/app/admin/router/usermanage.go b/app/admin/router/usermanage.go
index e165510..426ead2 100644
--- a/app/admin/router/usermanage.go
+++ b/app/admin/router/usermanage.go
@@ -33,6 +33,7 @@ func registerUserManageRouter(v1 *gin.RouterGroup, authMiddleware *jwt.GinJWTMid
 	{
 		userInfo.POST("/list", usermanage.UserList)
 		userInfo.POST("/invite_derive", usermanage.ExportDataUser)
+		userInfo.POST("/mark", usermanage.MarkUser)
 	}
 }