|
|
@@ -17,10 +17,10 @@
|
|
|
package org.springblade.modules.system.service.impl;
|
|
|
|
|
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
|
-import com.baomidou.mybatisplus.core.toolkit.IdWorker;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
+import com.yomahub.liteflow.core.FlowExecutor;
|
|
|
+import com.yomahub.liteflow.flow.LiteflowResponse;
|
|
|
import lombok.AllArgsConstructor;
|
|
|
-import org.springblade.common.cache.ParamCache;
|
|
|
import org.springblade.core.cache.utils.CacheUtil;
|
|
|
import org.springblade.core.log.exception.ServiceException;
|
|
|
import org.springblade.core.mp.base.BaseServiceImpl;
|
|
|
@@ -30,20 +30,18 @@ import org.springblade.core.tool.jackson.JsonUtil;
|
|
|
import org.springblade.core.tool.support.Kv;
|
|
|
import org.springblade.core.tool.utils.DesUtil;
|
|
|
import org.springblade.core.tool.utils.Func;
|
|
|
-import org.springblade.modules.auth.enums.UserEnum;
|
|
|
import org.springblade.modules.system.entity.*;
|
|
|
import org.springblade.modules.system.mapper.TenantMapper;
|
|
|
+import org.springblade.modules.system.rule.TenantContext;
|
|
|
import org.springblade.modules.system.service.*;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
|
-import java.util.ArrayList;
|
|
|
import java.util.Date;
|
|
|
-import java.util.LinkedList;
|
|
|
import java.util.List;
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
-import static org.springblade.common.constant.TenantConstant.*;
|
|
|
+import static org.springblade.common.constant.TenantConstant.DES_KEY;
|
|
|
import static org.springblade.core.cache.constant.CacheConstant.SYS_CACHE;
|
|
|
|
|
|
/**
|
|
|
@@ -55,7 +53,7 @@ import static org.springblade.core.cache.constant.CacheConstant.SYS_CACHE;
|
|
|
@AllArgsConstructor
|
|
|
public class TenantServiceImpl extends BaseServiceImpl<TenantMapper, Tenant> implements ITenantService {
|
|
|
|
|
|
- private final TenantId tenantId;
|
|
|
+ private final TenantId tenantIdGenerator;
|
|
|
private final IRoleService roleService;
|
|
|
private final IMenuService menuService;
|
|
|
private final IDeptService deptService;
|
|
|
@@ -63,6 +61,7 @@ public class TenantServiceImpl extends BaseServiceImpl<TenantMapper, Tenant> imp
|
|
|
private final IRoleMenuService roleMenuService;
|
|
|
private final IDictBizService dictBizService;
|
|
|
private final IUserService userService;
|
|
|
+ private final FlowExecutor flowExecutor;
|
|
|
|
|
|
@Override
|
|
|
public IPage<Tenant> selectTenantPage(IPage<Tenant> page, Tenant tenant) {
|
|
|
@@ -78,75 +77,40 @@ public class TenantServiceImpl extends BaseServiceImpl<TenantMapper, Tenant> imp
|
|
|
@Transactional(rollbackFor = Exception.class)
|
|
|
public boolean submitTenant(Tenant tenant) {
|
|
|
if (Func.isEmpty(tenant.getId())) {
|
|
|
- List<Tenant> tenants = baseMapper.selectList(Wrappers.<Tenant>query().lambda().eq(Tenant::getIsDeleted, BladeConstant.DB_NOT_DELETED));
|
|
|
- List<String> codes = tenants.stream().map(Tenant::getTenantId).collect(Collectors.toList());
|
|
|
- String tenantId = getTenantId(codes);
|
|
|
- tenant.setTenantId(tenantId);
|
|
|
- // 获取参数配置的账号额度
|
|
|
- int accountNumber = Func.toInt(ParamCache.getValue(ACCOUNT_NUMBER_KEY), DEFAULT_ACCOUNT_NUMBER);
|
|
|
- tenant.setAccountNumber(accountNumber);
|
|
|
- // 新建租户对应的默认角色
|
|
|
- Role role = new Role();
|
|
|
- role.setTenantId(tenantId);
|
|
|
- role.setParentId(BladeConstant.TOP_PARENT_ID);
|
|
|
- role.setRoleName("管理员");
|
|
|
- role.setRoleAlias("admin");
|
|
|
- role.setSort(2);
|
|
|
- role.setIsDeleted(BladeConstant.DB_NOT_DELETED);
|
|
|
- roleService.save(role);
|
|
|
- // 新建租户对应的角色菜单权限
|
|
|
- LinkedList<Menu> userMenus = new LinkedList<>();
|
|
|
- // 获取参数配置的默认菜单集合,逗号隔开
|
|
|
- List<String> menuCodes = Func.toStrList(ParamCache.getValue(ACCOUNT_MENU_CODE_KEY));
|
|
|
- List<Menu> menus = getMenus((menuCodes.size() > 0 ? menuCodes : MENU_CODES), userMenus);
|
|
|
- List<RoleMenu> roleMenus = new ArrayList<>();
|
|
|
- menus.forEach(menu -> {
|
|
|
- RoleMenu roleMenu = new RoleMenu();
|
|
|
- roleMenu.setMenuId(menu.getId());
|
|
|
- roleMenu.setRoleId(role.getId());
|
|
|
- roleMenus.add(roleMenu);
|
|
|
- });
|
|
|
- roleMenuService.saveBatch(roleMenus);
|
|
|
- // 新建租户对应的默认部门
|
|
|
- Dept dept = new Dept();
|
|
|
- dept.setTenantId(tenantId);
|
|
|
- dept.setParentId(BladeConstant.TOP_PARENT_ID);
|
|
|
- dept.setAncestors(String.valueOf(BladeConstant.TOP_PARENT_ID));
|
|
|
- dept.setDeptName(tenant.getTenantName());
|
|
|
- dept.setFullName(tenant.getTenantName());
|
|
|
- dept.setDeptCategory(1);
|
|
|
- dept.setSort(2);
|
|
|
- dept.setIsDeleted(BladeConstant.DB_NOT_DELETED);
|
|
|
- deptService.save(dept);
|
|
|
- // 新建租户对应的默认岗位
|
|
|
- Post post = new Post();
|
|
|
- post.setTenantId(tenantId);
|
|
|
- post.setCategory(1);
|
|
|
- post.setPostCode("ceo");
|
|
|
- post.setPostName("首席执行官");
|
|
|
- post.setSort(1);
|
|
|
- postService.save(post);
|
|
|
- // 新建租户对应的默认业务字典
|
|
|
- LinkedList<DictBiz> dictBizs = new LinkedList<>();
|
|
|
- List<DictBiz> dictBizList = getDictBizs(tenantId, dictBizs);
|
|
|
- dictBizService.saveBatch(dictBizList);
|
|
|
- // 新建租户对应的默认管理用户
|
|
|
- User user = new User();
|
|
|
- user.setTenantId(tenantId);
|
|
|
- user.setName("admin");
|
|
|
- user.setRealName("admin");
|
|
|
- user.setAccount("admin");
|
|
|
- // 获取参数配置的密码
|
|
|
- String password = Func.toStr(ParamCache.getValue(PASSWORD_KEY), DEFAULT_PASSWORD);
|
|
|
- user.setPassword(password);
|
|
|
- user.setRoleId(String.valueOf(role.getId()));
|
|
|
- user.setDeptId(String.valueOf(dept.getId()));
|
|
|
- user.setPostId(String.valueOf(post.getId()));
|
|
|
- user.setBirthday(new Date());
|
|
|
- user.setSex(1);
|
|
|
- user.setUserType(UserEnum.WEB.getCategory());
|
|
|
- user.setIsDeleted(BladeConstant.DB_NOT_DELETED);
|
|
|
- userService.submit(user);
|
|
|
+ TenantContext tenantContext = new TenantContext();
|
|
|
+ tenantContext.setTenantIdGenerator(tenantIdGenerator);
|
|
|
+ tenantContext.setTenant(tenant);
|
|
|
+ tenantContext.setMenuService(menuService);
|
|
|
+ tenantContext.setDictBizService(dictBizService);
|
|
|
+ tenantContext.setTenantService(this);
|
|
|
+
|
|
|
+ LiteflowResponse resp = flowExecutor.execute2Resp("tenantChain", null, tenantContext);
|
|
|
+ if (resp.isSuccess()) {
|
|
|
+ Role role = tenantContext.getRole();
|
|
|
+ roleService.save(role);
|
|
|
+
|
|
|
+ Long roleId = role.getId();
|
|
|
+ List<RoleMenu> roleMenuList = tenantContext.getRoleMenuList();
|
|
|
+ roleMenuList.forEach(roleMenu -> roleMenu.setRoleId(roleId));
|
|
|
+ roleMenuService.saveBatch(roleMenuList);
|
|
|
+
|
|
|
+ Dept dept = tenantContext.getDept();
|
|
|
+ deptService.save(dept);
|
|
|
+
|
|
|
+ Post post = tenantContext.getPost();
|
|
|
+ postService.save(post);
|
|
|
+
|
|
|
+ List<DictBiz> dictBizList = tenantContext.getDictBizList();
|
|
|
+ dictBizService.saveBatch(dictBizList);
|
|
|
+
|
|
|
+ User user = tenantContext.getUser();
|
|
|
+ user.setRoleId(String.valueOf(role.getId()));
|
|
|
+ user.setDeptId(String.valueOf(dept.getId()));
|
|
|
+ user.setPostId(String.valueOf(post.getId()));
|
|
|
+ userService.submit(user);
|
|
|
+ } else {
|
|
|
+ throw new ServiceException("租户业务数据构建异常");
|
|
|
+ }
|
|
|
}
|
|
|
CacheUtil.clear(SYS_CACHE, tenant.getTenantId());
|
|
|
return super.saveOrUpdate(tenant);
|
|
|
@@ -185,55 +149,4 @@ public class TenantServiceImpl extends BaseServiceImpl<TenantMapper, Tenant> imp
|
|
|
return true;
|
|
|
}
|
|
|
|
|
|
- private String getTenantId(List<String> codes) {
|
|
|
- String code = tenantId.generate();
|
|
|
- if (codes.contains(code)) {
|
|
|
- return getTenantId(codes);
|
|
|
- }
|
|
|
- return code;
|
|
|
- }
|
|
|
-
|
|
|
- private List<Menu> getMenus(List<String> codes, LinkedList<Menu> menus) {
|
|
|
- codes.forEach(code -> {
|
|
|
- Menu menu = menuService.getOne(Wrappers.<Menu>query().lambda().eq(Menu::getCode, code).eq(Menu::getIsDeleted, BladeConstant.DB_NOT_DELETED));
|
|
|
- if (menu != null) {
|
|
|
- menus.add(menu);
|
|
|
- recursionMenu(menu.getId(), menus);
|
|
|
- }
|
|
|
- });
|
|
|
- return menus;
|
|
|
- }
|
|
|
-
|
|
|
- private void recursionMenu(Long parentId, LinkedList<Menu> menus) {
|
|
|
- List<Menu> menuList = menuService.list(Wrappers.<Menu>query().lambda().eq(Menu::getParentId, parentId).eq(Menu::getIsDeleted, BladeConstant.DB_NOT_DELETED));
|
|
|
- menus.addAll(menuList);
|
|
|
- menuList.forEach(menu -> recursionMenu(menu.getId(), menus));
|
|
|
- }
|
|
|
-
|
|
|
- private List<DictBiz> getDictBizs(String tenantId, LinkedList<DictBiz> dictBizs) {
|
|
|
- List<DictBiz> dictBizList = dictBizService.list(Wrappers.<DictBiz>query().lambda().eq(DictBiz::getParentId, BladeConstant.TOP_PARENT_ID).eq(DictBiz::getIsDeleted, BladeConstant.DB_NOT_DELETED));
|
|
|
- dictBizList.forEach(dictBiz -> {
|
|
|
- Long oldParentId = dictBiz.getId();
|
|
|
- Long newParentId = IdWorker.getId();
|
|
|
- dictBiz.setId(newParentId);
|
|
|
- dictBiz.setTenantId(tenantId);
|
|
|
- dictBizs.add(dictBiz);
|
|
|
- recursionDictBiz(tenantId, oldParentId, newParentId, dictBizs);
|
|
|
- });
|
|
|
- return dictBizs;
|
|
|
- }
|
|
|
-
|
|
|
- private void recursionDictBiz(String tenantId, Long oldParentId, Long newParentId, LinkedList<DictBiz> dictBizs) {
|
|
|
- List<DictBiz> dictBizList = dictBizService.list(Wrappers.<DictBiz>query().lambda().eq(DictBiz::getParentId, oldParentId).eq(DictBiz::getIsDeleted, BladeConstant.DB_NOT_DELETED));
|
|
|
- dictBizList.forEach(dictBiz -> {
|
|
|
- Long oldSubParentId = dictBiz.getId();
|
|
|
- Long newSubParentId = IdWorker.getId();
|
|
|
- dictBiz.setId(newSubParentId);
|
|
|
- dictBiz.setTenantId(tenantId);
|
|
|
- dictBiz.setParentId(newParentId);
|
|
|
- dictBizs.add(dictBiz);
|
|
|
- recursionDictBiz(tenantId, oldSubParentId, newSubParentId, dictBizs);
|
|
|
- });
|
|
|
- }
|
|
|
-
|
|
|
}
|