From 4b92b607f77e4e99612c30c4459603585d01c9e1 Mon Sep 17 00:00:00 2001 From: Ryan SU Date: Wed, 8 Nov 2023 23:00:04 +0800 Subject: [PATCH] fix: add index for menu --- api/internal/logic/base/init_database_logic.go | 6 ++++++ api/internal/svc/service_context.go | 5 +++++ 2 files changed, 11 insertions(+) diff --git a/api/internal/logic/base/init_database_logic.go b/api/internal/logic/base/init_database_logic.go index bda2f6aa..012239c4 100644 --- a/api/internal/logic/base/init_database_logic.go +++ b/api/internal/logic/base/init_database_logic.go @@ -72,5 +72,11 @@ func (l *InitDatabaseLogic) InitDatabase() (resp *types.BaseMsgResp, err error) return nil, errorx.NewCodeInternalError(i18n.DatabaseError) } + // load role ban data + err = l.svcCtx.LoadBanRoleData() + if err != nil { + return nil, err + } + return &types.BaseMsgResp{Msg: l.svcCtx.Trans.Trans(l.ctx, result.Msg)}, nil } diff --git a/api/internal/svc/service_context.go b/api/internal/svc/service_context.go index 7090c995..d42bbf80 100644 --- a/api/internal/svc/service_context.go +++ b/api/internal/svc/service_context.go @@ -12,6 +12,7 @@ import ( "github.com/suyuan32/simple-admin-message-center/mcmsclient" "github.com/zeromicro/go-zero/core/errorx" "github.com/zeromicro/go-zero/core/logx" + "strings" "github.com/suyuan32/simple-admin-job/jobclient" @@ -84,6 +85,10 @@ func (l *ServiceContext) LoadBanRoleData() error { }) if err != nil { + if strings.Contains(err.Error(), i18n.DatabaseError) { + return nil + } + logx.Error("failed to load role data, please check if initialize the database") return errorx.NewCodeInternalError("failed to load role data") }