refactor: user remove is ready for console

api/go/core/app_server.go

@@ -6,7 +6,6 @@ import (
 	"chatplus/utils"
 	"chatplus/utils/resp"
 	"context"
-	"encoding/gob"
 	"github.com/gin-contrib/sessions"
 	"github.com/gin-contrib/sessions/cookie"
 	"github.com/gin-gonic/gin"
@@ -51,7 +50,7 @@ func (s *AppServer) Init(debug bool) {
 	s.Engine.Use(sessionMiddleware(s.AppConfig))
 	s.Engine.Use(authorizeMiddleware(s))
 	s.Engine.Use(errorHandler)
-	gob.Register(model.User{})
+	//gob.Register(model.User{})
 }
 
 func (s *AppServer) Run(db *gorm.DB) error {

api/go/handler/admin/user_handler.go

@@ -65,15 +65,16 @@ func (h *UserHandler) Update(c *gin.Context) {
 		resp.ERROR(c, types.InvalidArgs)
 		return
 	}
-	var user = model.User{
-		Nickname:    data.Nickname,
-		Calls:       data.Calls,
-		Status:      data.Status,
-		ChatRoles:   utils.JsonEncode(data.ChatRoles),
-		ExpiredTime: utils.Str2stamp(data.ExpiredTime),
-	}
+	var user = model.User{}
 	user.Id = data.Id
-	res := h.db.Updates(&user)
+	// 此处需要用 map 更新，用结构体无法更新 0 值
+	res := h.db.Model(&user).Updates(map[string]interface{}{
+		"nickname":        data.Nickname,
+		"calls":           data.Calls,
+		"status":          data.Status,
+		"chat_roles_json": utils.JsonEncode(data.ChatRoles),
+		"expired_time":    utils.Str2stamp(data.ExpiredTime),
+	})
 	if res.Error != nil {
 		resp.ERROR(c, "更新数据库失败")
 		return
@@ -82,6 +83,41 @@ func (h *UserHandler) Update(c *gin.Context) {
 	resp.SUCCESS(c)
 }
 
+func (h *UserHandler) Remove(c *gin.Context) {
+	id := h.GetInt(c, "id", 0)
+	if id > 0 {
+		tx := h.db.Begin()
+		res := h.db.Where("id = ?", id).Delete(&model.User{})
+		if res.Error != nil {
+			resp.ERROR(c, "删除失败")
+			return
+		}
+		// 删除聊天记录
+		res = h.db.Where("user_id = ?", id).Delete(&model.ChatItem{})
+		if res.Error != nil {
+			tx.Rollback()
+			resp.ERROR(c, "删除失败")
+			return
+		}
+		// 删除聊天历史记录
+		res = h.db.Where("user_id = ?", id).Delete(&model.HistoryMessage{})
+		if res.Error != nil {
+			tx.Rollback()
+			resp.ERROR(c, "删除失败")
+			return
+		}
+		// 删除登录日志
+		res = h.db.Where("user_id = ?", id).Delete(&model.UserLoginLog{})
+		if res.Error != nil {
+			tx.Rollback()
+			resp.ERROR(c, "删除失败")
+			return
+		}
+		tx.Commit()
+	}
+	resp.SUCCESS(c)
+}
+
 func (h *UserHandler) InitUser(c *gin.Context) {
 	var users []model.User
 	h.db.Find(&users)

api/go/main.go

@@ -135,6 +135,7 @@ func main() {
 			group := s.Engine.Group("/api/admin/user/")
 			group.GET("list", h.List)
 			group.POST("update", h.Update)
+			group.GET("remove", h.Remove)
 			group.GET("test", h.InitUser)
 		}),
 		fx.Invoke(func(s *core.AppServer, h *admin.ChatRoleHandler) {

api/go/store/model/user.go

@@ -12,7 +12,7 @@ type User struct {
 	ChatConfig  string `gorm:"column:chat_config_json"` // 聊天配置 json
 	ChatRoles   string `gorm:"column:chat_roles_json"`  // 聊天角色
 	ExpiredTime int64  // 账户到期时间
-	Status      bool   // 当前状态
+	Status      bool   `gorm:"default:true"` // 当前状态
 	LastLoginAt int64  // 最后登录时间
 	LastLoginIp string // 最后登录 IP
 }