diff --git a/src/modules/network_data/controller/udm_sub.go b/src/modules/network_data/controller/udm_sub.go index e8617139..2272e9a5 100644 --- a/src/modules/network_data/controller/udm_sub.go +++ b/src/modules/network_data/controller/udm_sub.go @@ -54,7 +54,17 @@ func (s *UDMSubController) ResetData(c *gin.Context) { // GET /list func (s *UDMSubController) List(c *gin.Context) { querys := ctx.QueryMap(c) - querys["userName"] = ctx.LoginUserToUserName(c) + // querys["userName"] = ctx.LoginUserToUserName(c) + // 判断租户角色 + loginUser, _ := ctx.LoginUser(c) + if len(loginUser.User.Roles) > 0 { + for _, v := range loginUser.User.Roles { + if v.RoleKey == "tenant" { + querys["tenantName"] = loginUser.User.UserName + break + } + } + } data := s.udmSubService.SelectPage(querys) c.JSON(200, result.Ok(data)) } @@ -368,6 +378,17 @@ func (s *UDMSubController) Export(c *gin.Context) { return } + // 判断租户角色 + loginUser, _ := ctx.LoginUser(c) + if len(loginUser.User.Roles) > 0 { + for _, v := range loginUser.User.Roles { + if v.RoleKey == "tenant" { + querys["tenantName"] = loginUser.User.UserName + break + } + } + } + querys["pageNum"] = 1 querys["pageSize"] = 10000 data := s.udmSubService.SelectPage(querys)