mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-07 02:20:24 +08:00
优化系统用户登录校验
This commit is contained in:
@@ -63,6 +63,19 @@ func (this *UserDAO) FindEnabledUser(id int64) (*User, error) {
|
||||
return result.(*User), err
|
||||
}
|
||||
|
||||
// 查找用户基本信息
|
||||
func (this *UserDAO) FindEnabledBasicUser(id int64) (*User, error) {
|
||||
result, err := this.Query().
|
||||
Pk(id).
|
||||
Attr("state", UserStateEnabled).
|
||||
Result("id", "fullname", "username").
|
||||
Find()
|
||||
if result == nil {
|
||||
return nil, err
|
||||
}
|
||||
return result.(*User), err
|
||||
}
|
||||
|
||||
// 获取管理员名称
|
||||
func (this *UserDAO) FindUserFullname(userId int64) (string, error) {
|
||||
return this.Query().
|
||||
|
||||
@@ -253,6 +253,7 @@ func (this *AdminService) FindAllAdminModules(ctx context.Context, req *pb.FindA
|
||||
list := &pb.AdminModuleList{
|
||||
AdminId: int64(admin.Id),
|
||||
IsSuper: admin.IsSuper == 1,
|
||||
Fullname: admin.Fullname,
|
||||
Modules: pbModules,
|
||||
}
|
||||
result = append(result, list)
|
||||
|
||||
@@ -430,6 +430,19 @@ func (this *ServerService) ListEnabledServersMatch(ctx context.Context, req *pb.
|
||||
}
|
||||
}
|
||||
|
||||
// 用户
|
||||
user, err := models.SharedUserDAO.FindEnabledBasicUser(int64(server.UserId))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var pbUser *pb.User = nil
|
||||
if user != nil {
|
||||
pbUser = &pb.User{
|
||||
Id: int64(user.Id),
|
||||
Fullname: user.Fullname,
|
||||
}
|
||||
}
|
||||
|
||||
result = append(result, &pb.Server{
|
||||
Id: int64(server.Id),
|
||||
IsOn: server.IsOn == 1,
|
||||
@@ -453,6 +466,7 @@ func (this *ServerService) ListEnabledServersMatch(ctx context.Context, req *pb.
|
||||
Name: clusterName,
|
||||
},
|
||||
Groups: pbGroups,
|
||||
User: pbUser,
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user