feat: admin 使用固定 ID 并补全全角色权限和演示数据

This commit is contained in:
zetaloop
2026-05-03 08:09:43 +08:00
parent 776ecc479f
commit 9a32850030
6 changed files with 89 additions and 30 deletions
+35 -11
View File
@@ -6,14 +6,28 @@ import (
"strings"
"time"
"juwan-backend/app/snowflake/rpc/snowflake"
"juwan-backend/app/users/rpc/internal/models/schema"
"juwan-backend/app/users/rpc/internal/models/users"
"juwan-backend/app/users/rpc/internal/svc"
"juwan-backend/common/utils/pwdUtils"
"juwan-backend/pkg/types"
"github.com/zeromicro/go-zero/core/logx"
)
const adminUserID int64 = 100000
var adminVerifiedRoles = types.TextArray{
Elements: []string{"consumer", "player", "owner", "admin"},
Valid: true,
}
var adminVerificationStatus = schema.VerificationStatusStruct{
Consumer: "approved",
Player: "approved",
Owner: "approved",
}
func initAdmin(svcCtx *svc.ServiceContext) {
username := strings.TrimSpace(os.Getenv("ADMIN_USERNAME"))
password := strings.TrimSpace(os.Getenv("ADMIN_PASSWORD"))
@@ -24,8 +38,9 @@ func initAdmin(svcCtx *svc.ServiceContext) {
go func() {
ctx := context.Background()
var existing bool
for i := range 30 {
exists, err := svcCtx.UsersModelRW.Users.Query().Where(users.UsernameEQ(username)).Exist(ctx)
ok, err := svcCtx.UsersModelRW.Users.Query().Where(users.UsernameEQ(username)).Exist(ctx)
if err != nil {
if i < 29 {
time.Sleep(time.Second)
@@ -34,10 +49,23 @@ func initAdmin(svcCtx *svc.ServiceContext) {
logx.Errorf("check admin user: %v", err)
return
}
if exists {
existing = ok
break
}
if existing {
err := svcCtx.UsersModelRW.Users.Update().
Where(users.UsernameEQ(username)).
SetIsAdmin(true).
SetVerifiedRoles(adminVerifiedRoles).
SetVerificationStatus(adminVerificationStatus).
Exec(ctx)
if err != nil {
logx.Errorf("reconcile admin user: %v", err)
return
}
break
logx.Infof("reconciled admin user: %s", username)
return
}
hashedPassword, err := pwdUtils.HashPassword(password)
@@ -46,14 +74,8 @@ func initAdmin(svcCtx *svc.ServiceContext) {
return
}
resp, err := svcCtx.Snowflake.NextId(ctx, &snowflake.NextIdReq{})
if err != nil {
logx.Errorf("generate admin user ID: %v", err)
return
}
_, err = svcCtx.UsersModelRW.Users.Create().
SetID(resp.Id).
SetID(adminUserID).
SetUsername(username).
SetPasswordHash(hashedPassword).
SetEmail(email).
@@ -63,6 +85,8 @@ func initAdmin(svcCtx *svc.ServiceContext) {
SetCurrentRole("consumer").
SetNickname(username).
SetIsAdmin(true).
SetVerifiedRoles(adminVerifiedRoles).
SetVerificationStatus(adminVerificationStatus).
Save(ctx)
if err != nil {
logx.Errorf("create admin user: %v", err)