feat: admin 使用固定 ID 并补全全角色权限和演示数据
This commit is contained in:
+35
-11
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user