diff --git a/api/go/core/app_server.go b/api/go/core/app_server.go index ab1d94c3..3cacbbcf 100644 --- a/api/go/core/app_server.go +++ b/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 { diff --git a/api/go/handler/admin/user_handler.go b/api/go/handler/admin/user_handler.go index 4b53987d..de9954d4 100644 --- a/api/go/handler/admin/user_handler.go +++ b/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) diff --git a/api/go/main.go b/api/go/main.go index b0b17f37..8b80793c 100644 --- a/api/go/main.go +++ b/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) { diff --git a/api/go/store/model/user.go b/api/go/store/model/user.go index df3ccb62..01abc6a8 100644 --- a/api/go/store/model/user.go +++ b/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 } diff --git a/web/src/views/admin/UserList.vue b/web/src/views/admin/UserList.vue index 0378895d..27fd92fc 100644 --- a/web/src/views/admin/UserList.vue +++ b/web/src/views/admin/UserList.vue @@ -1,7 +1,8 @@