diff --git a/tyr-server/src/main/java/cn/axzo/tyr/server/controller/PrivateRoleController.java b/tyr-server/src/main/java/cn/axzo/tyr/server/controller/PrivateRoleController.java index ea737ac6..18e7a9db 100644 --- a/tyr-server/src/main/java/cn/axzo/tyr/server/controller/PrivateRoleController.java +++ b/tyr-server/src/main/java/cn/axzo/tyr/server/controller/PrivateRoleController.java @@ -20,6 +20,7 @@ import cn.axzo.tyr.client.model.vo.DeleteRoleVO; import cn.axzo.tyr.client.model.vo.SaasRoleGroupVO; import cn.axzo.tyr.client.model.vo.SaveOrUpdateRoleVO; import cn.axzo.tyr.server.config.MqProducer; +import cn.axzo.tyr.server.event.inner.CacheRoleSaasFeatureResourceHandler; import cn.axzo.tyr.server.event.payload.RolePermissionCreatedPayload; import cn.axzo.tyr.server.repository.dao.SaasPermissionGroupDao; import cn.axzo.tyr.server.repository.dao.SaasPgroupRoleRelationDao; @@ -109,6 +110,8 @@ public class PrivateRoleController { private MqProducer mqProducer; @Autowired private SaasPermissionGroupDao saasPermissionGroupDao; + @Autowired + private CacheRoleSaasFeatureResourceHandler cacheRoleSaasFeatureResourceHandler; private static final String TARGET_TYPE = "saasFeatureResourceId"; @@ -671,12 +674,10 @@ public class PrivateRoleController { @PostMapping("/api/private/superAdmin/pgroupPermission/init") public Object createSuperAdminPgroupPermission() { ListRoleReq listRoleReq = ListRoleReq.builder() - .roleTypes(Lists.newArrayList(RoleTypeEnum.SUPER_ADMIN.getValue())) + .roleTypes(Lists.newArrayList(RoleTypeEnum.SUPER_ADMIN.getValue(), RoleTypeEnum.ADMIN.getValue())) .build(); - List allSuperAdminRoles = roleService.list(listRoleReq).stream() - .filter(e -> e.getWorkspaceId() == 0L) - .collect(Collectors.toList()); + List allSuperAdminRoles = roleService.list(listRoleReq); if (CollectionUtils.isEmpty(allSuperAdminRoles)) { return "ok"; } @@ -756,7 +757,15 @@ public class PrivateRoleController { return "ok"; } - @Data + @PostMapping("/api/private/test") + public Object test(@RequestBody @Validated RolePermissionCreatedPayload payload) { + cacheRoleSaasFeatureResourceHandler.onRolePermissionUpsert(Event.builder() + .data(payload) + .build(), null); + return "ok"; + } + + @Data @Builder @NoArgsConstructor @AllArgsConstructor diff --git a/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRolePermissionHandler.java b/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRolePermissionHandler.java index 20bc1f7a..1f4afeed 100644 --- a/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRolePermissionHandler.java +++ b/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRolePermissionHandler.java @@ -2,7 +2,6 @@ package cn.axzo.tyr.server.event.inner; import cn.axzo.framework.rocketmq.Event; import cn.axzo.framework.rocketmq.EventConsumer; -import cn.axzo.tyr.client.common.enums.RoleTypeEnum; import cn.axzo.tyr.client.model.req.ListRoleReq; import cn.axzo.tyr.client.model.res.SaasRoleRes; import cn.axzo.tyr.server.event.payload.PageElementFeatureResourceUpsertPayload; @@ -49,7 +48,6 @@ public class CacheRolePermissionHandler implements InitializingBean { .roleIds(Optional.ofNullable(payload.getRoleIds()) .map(Lists::newArrayList) .orElse(null)) - .roleTypesNotIn(RoleTypeEnum.listAdmin()) .build(); Set roleIds = roleService.list(listSaasRoleParam).stream() .map(SaasRoleRes::getId) @@ -74,7 +72,6 @@ public class CacheRolePermissionHandler implements InitializingBean { } ListRoleReq listSaasRoleParam = ListRoleReq.builder() - .roleTypesNotIn(RoleTypeEnum.listAdmin()) .build(); Set roleIds = roleService.list(listSaasRoleParam).stream() .map(SaasRoleRes::getId) diff --git a/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRoleSaasFeatureResourceHandler.java b/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRoleSaasFeatureResourceHandler.java index 39e283b4..b3bd8e32 100644 --- a/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRoleSaasFeatureResourceHandler.java +++ b/tyr-server/src/main/java/cn/axzo/tyr/server/event/inner/CacheRoleSaasFeatureResourceHandler.java @@ -2,7 +2,6 @@ package cn.axzo.tyr.server.event.inner; import cn.axzo.framework.rocketmq.Event; import cn.axzo.framework.rocketmq.EventConsumer; -import cn.axzo.tyr.client.common.enums.RoleTypeEnum; import cn.axzo.tyr.client.model.req.ListRoleReq; import cn.axzo.tyr.client.model.res.SaasRoleRes; import cn.axzo.tyr.server.event.payload.PageElementFeatureResourceUpsertPayload; @@ -44,7 +43,6 @@ public class CacheRoleSaasFeatureResourceHandler implements InitializingBean { .roleIds(Optional.ofNullable(payload.getRoleIds()) .map(Lists::newArrayList) .orElse(null)) - .roleTypesNotIn(RoleTypeEnum.listAdmin()) .build(); Set roleIds = roleService.list(listSaasRoleParam).stream() .map(SaasRoleRes::getId) @@ -68,9 +66,7 @@ public class CacheRoleSaasFeatureResourceHandler implements InitializingBean { return; } - ListRoleReq listSaasRoleParam = ListRoleReq.builder() - .roleTypesNotIn(RoleTypeEnum.listAdmin()) - .build(); + ListRoleReq listSaasRoleParam = ListRoleReq.builder().build(); Set roleIds = roleService.list(listSaasRoleParam).stream() .map(SaasRoleRes::getId)