diff --git a/internal/db/models/user_dao.go b/internal/db/models/user_dao.go index 618dfec6..8f7d90b1 100644 --- a/internal/db/models/user_dao.go +++ b/internal/db/models/user_dao.go @@ -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(). diff --git a/internal/rpc/services/service_admin.go b/internal/rpc/services/service_admin.go index c07cb31d..97ea7ee2 100644 --- a/internal/rpc/services/service_admin.go +++ b/internal/rpc/services/service_admin.go @@ -251,9 +251,10 @@ func (this *AdminService) FindAllAdminModules(ctx context.Context, req *pb.FindA } list := &pb.AdminModuleList{ - AdminId: int64(admin.Id), - IsSuper: admin.IsSuper == 1, - Modules: pbModules, + AdminId: int64(admin.Id), + IsSuper: admin.IsSuper == 1, + Fullname: admin.Fullname, + Modules: pbModules, } result = append(result, list) } diff --git a/internal/rpc/services/service_server.go b/internal/rpc/services/service_server.go index ade2a442..95dd1af0 100644 --- a/internal/rpc/services/service_server.go +++ b/internal/rpc/services/service_server.go @@ -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, }) }