xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-06 01:03:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

synchronize license every 10 secs

Browse Source
This commit is contained in:
RockYang
2024-04-26 14:35:01 +08:00
parent 74b1b07b31
commit 1966e69460
5 changed files with 113 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/handler/admin/user_handler.go
View File

@@ -4,6 +4,7 @@ import (
"chatplus/core"
"chatplus/core/types"
"chatplus/handler"
"chatplus/service"
"chatplus/store/model"
"chatplus/store/vo"
"chatplus/utils"
@@ -17,10 +18,11 @@ import (
type UserHandler struct {
handler.BaseHandler
licenseService *service.LicenseService
}
func NewUserHandler(app *core.AppServer, db *gorm.DB) *UserHandler {
return &UserHandler{BaseHandler: handler.BaseHandler{App: app, DB: db}}
func NewUserHandler(app *core.AppServer, db *gorm.DB, licenseService *service.LicenseService) *UserHandler {
return &UserHandler{BaseHandler: handler.BaseHandler{App: app, DB: db}, licenseService: licenseService}
}
// List 用户列表
@@ -75,6 +77,13 @@ func (h *UserHandler) Save(c *gin.Context) {
resp.ERROR(c, types.InvalidArgs)
return
}
// 检测最大注册人数
var totalUser int64
h.DB.Model(&model.User{}).Count(&totalUser)
if int(totalUser) >= h.licenseService.GetLicense().UserNum {
resp.ERROR(c, "当前注册用户数已达上限，请请升级 License")
return
}
var user = model.User{}
var res *gorm.DB
var userVo vo.User