|
@@ -58,10 +58,6 @@ func (a *Admin) Login(ctx *gin.Context, username string, password string) (*enti
|
|
|
user, err := a.QueryUserByUsername(ctx, username)
|
|
|
if err != nil {
|
|
|
mhayaLogger.Warnf("Login QueryUserByUsername error:", err)
|
|
|
- if errors.Is(err, mongo.ErrNoDocuments) {
|
|
|
- return nil, common.NewResult(code.AccountNotExistError)
|
|
|
- }
|
|
|
-
|
|
|
return nil, common.NewResult(code.InternalError)
|
|
|
}
|
|
|
|
|
@@ -188,11 +184,7 @@ func (a *Admin) loginAuthSetRoleRedis(roleID, generateToken string) error {
|
|
|
func (a *Admin) QueryUserByUsername(ctx context.Context, username string) (*model.Admin, error) {
|
|
|
admin := &model.Admin{}
|
|
|
err := mdb.MDB.Collection(a.GetDBName()).FindOne(ctx, bson.M{"username": username}).Decode(&admin)
|
|
|
- if errors.Is(err, mongo.ErrNoDocuments) {
|
|
|
- if username != "admin" {
|
|
|
- return nil, err
|
|
|
- }
|
|
|
-
|
|
|
+ if errors.Is(err, mongo.ErrNoDocuments) && username == "admin" {
|
|
|
// 如果是admin 登录的话 创建一个初始的admin并且存入数据库
|
|
|
pwd, err := HashPassword("123456")
|
|
|
if err != nil {
|
|
@@ -230,6 +222,7 @@ func (a *Admin) QueryUserByUsername(ctx context.Context, username string) (*mode
|
|
|
}
|
|
|
|
|
|
return admin, nil
|
|
|
+
|
|
|
}
|
|
|
|
|
|
// ChangePassword 修改管理员密码
|