mirror of
https://gitee.com/hhyykk/ipms-sjy.git
synced 2025-07-17 12:35:07 +08:00
基于 Guava 实现 tenant 租户数据的本地缓存
This commit is contained in:
@ -1,10 +1,10 @@
|
||||
package cn.iocoder.yudao.module.system.service.tenant;
|
||||
|
||||
import cn.hutool.core.util.ReflectUtil;
|
||||
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
||||
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
||||
import cn.iocoder.yudao.framework.tenant.config.TenantProperties;
|
||||
import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder;
|
||||
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
||||
import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantCreateReqVO;
|
||||
import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantExportReqVO;
|
||||
import cn.iocoder.yudao.module.system.controller.admin.tenant.vo.tenant.TenantPageReqVO;
|
||||
@ -16,14 +16,12 @@ import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantPackageDO;
|
||||
import cn.iocoder.yudao.module.system.dal.mysql.tenant.TenantMapper;
|
||||
import cn.iocoder.yudao.module.system.enums.permission.RoleCodeEnum;
|
||||
import cn.iocoder.yudao.module.system.enums.permission.RoleTypeEnum;
|
||||
import cn.iocoder.yudao.module.system.mq.producer.tenant.TenantProducer;
|
||||
import cn.iocoder.yudao.module.system.service.permission.MenuService;
|
||||
import cn.iocoder.yudao.module.system.service.permission.PermissionService;
|
||||
import cn.iocoder.yudao.module.system.service.permission.RoleService;
|
||||
import cn.iocoder.yudao.module.system.service.tenant.handler.TenantInfoHandler;
|
||||
import cn.iocoder.yudao.module.system.service.tenant.handler.TenantMenuHandler;
|
||||
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
|
||||
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.boot.test.mock.mockito.MockBean;
|
||||
@ -34,13 +32,11 @@ import java.time.Duration;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import static cn.iocoder.yudao.framework.common.util.collection.SetUtils.asSet;
|
||||
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.addTime;
|
||||
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.buildTime;
|
||||
import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.cloneIgnoreId;
|
||||
import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.max;
|
||||
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertPojoEquals;
|
||||
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
|
||||
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
|
||||
@ -78,43 +74,18 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
||||
private MenuService menuService;
|
||||
@MockBean
|
||||
private PermissionService permissionService;
|
||||
@MockBean
|
||||
private TenantProducer tenantProducer;
|
||||
|
||||
@BeforeEach
|
||||
public void setUp() {
|
||||
// 清理缓存
|
||||
ReflectUtil.setFieldValue(tenantService, "tenantCache", Collections.emptyMap());
|
||||
ReflectUtil.setFieldValue(tenantService, "maxUpdateTime", null);
|
||||
// 清理租户上下文
|
||||
TenantContextHolder.clear();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInitLocalCache() {
|
||||
// mock 数据
|
||||
TenantDO tenantDO1 = randomPojo(TenantDO.class);
|
||||
tenantMapper.insert(tenantDO1);
|
||||
TenantDO tenantDO2 = randomPojo(TenantDO.class);
|
||||
tenantMapper.insert(tenantDO2);
|
||||
|
||||
// 调用
|
||||
tenantService.initLocalCache();
|
||||
// 断言 tenantCache 缓存
|
||||
Map<Long, TenantDO> tenantCache = tenantService.getTenantCache();
|
||||
assertEquals(2, tenantCache.size());
|
||||
assertPojoEquals(tenantDO1, tenantCache.get(tenantDO1.getId()));
|
||||
assertPojoEquals(tenantDO2, tenantCache.get(tenantDO2.getId()));
|
||||
// 断言 maxUpdateTime 缓存
|
||||
assertEquals(max(tenantDO1.getUpdateTime(), tenantDO2.getUpdateTime()), tenantService.getMaxUpdateTime());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testGetTenantIds() {
|
||||
// mock 数据
|
||||
TenantDO tenant = randomPojo(TenantDO.class, o -> o.setId(1L));
|
||||
tenantMapper.insert(tenant);
|
||||
tenantService.initLocalCache();
|
||||
|
||||
// 调用,并断言业务异常
|
||||
List<Long> result = tenantService.getTenantIds();
|
||||
@ -131,7 +102,6 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
||||
// mock 数据
|
||||
TenantDO tenant = randomPojo(TenantDO.class, o -> o.setId(1L).setStatus(CommonStatusEnum.DISABLE.getStatus()));
|
||||
tenantMapper.insert(tenant);
|
||||
tenantService.initLocalCache();
|
||||
|
||||
// 调用,并断言业务异常
|
||||
assertServiceException(() -> tenantService.validTenant(1L), TENANT_DISABLE, tenant.getName());
|
||||
@ -143,7 +113,6 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
||||
TenantDO tenant = randomPojo(TenantDO.class, o -> o.setId(1L).setStatus(CommonStatusEnum.ENABLE.getStatus())
|
||||
.setExpireTime(buildTime(2020, 2, 2)));
|
||||
tenantMapper.insert(tenant);
|
||||
tenantService.initLocalCache();
|
||||
|
||||
// 调用,并断言业务异常
|
||||
assertServiceException(() -> tenantService.validTenant(1L), TENANT_EXPIRE, tenant.getName());
|
||||
@ -155,7 +124,6 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
||||
TenantDO tenant = randomPojo(TenantDO.class, o -> o.setId(1L).setStatus(CommonStatusEnum.ENABLE.getStatus())
|
||||
.setExpireTime(addTime(Duration.ofDays(1))));
|
||||
tenantMapper.insert(tenant);
|
||||
tenantService.initLocalCache();
|
||||
|
||||
// 调用,并断言业务异常
|
||||
tenantService.validTenant(1L);
|
||||
@ -206,8 +174,6 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
||||
verify(permissionService).assignRoleMenu(eq(200L), same(tenantPackage.getMenuIds()));
|
||||
// verify 分配角色
|
||||
verify(permissionService).assignUserRole(eq(300L), eq(singleton(200L)));
|
||||
// verify 发送刷新消息
|
||||
verify(tenantProducer).sendTenantRefreshMessage();
|
||||
}
|
||||
|
||||
@Test
|
||||
@ -240,8 +206,6 @@ public class TenantServiceImplTest extends BaseDbUnitTest {
|
||||
// 校验是否更新正确
|
||||
TenantDO tenant = tenantMapper.selectById(reqVO.getId()); // 获取最新的
|
||||
assertPojoEquals(reqVO, tenant);
|
||||
// verify 发送刷新消息
|
||||
verify(tenantProducer).sendTenantRefreshMessage();
|
||||
// verify 设置角色权限
|
||||
verify(permissionService).assignRoleMenu(eq(100L), eq(asSet(200L, 201L)));
|
||||
verify(permissionService).assignRoleMenu(eq(101L), eq(asSet(201L)));
|
||||
|
Reference in New Issue
Block a user