From 4b86cf6bbe5a7204a801169398fa836e8e06c109 Mon Sep 17 00:00:00 2001
From: guoshilong <123456>
Date: Wed, 14 Dec 2022 17:04:36 +0800
Subject: [PATCH] Merge remote-tracking branch 'refs/remotes/origin/master'
---
src/main/java/org/springblade/modules/system/service/impl/TenantServiceImpl.java | 239 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 239 insertions(+), 0 deletions(-)
diff --git a/src/main/java/org/springblade/modules/system/service/impl/TenantServiceImpl.java b/src/main/java/org/springblade/modules/system/service/impl/TenantServiceImpl.java
new file mode 100644
index 0000000..7cf537e
--- /dev/null
+++ b/src/main/java/org/springblade/modules/system/service/impl/TenantServiceImpl.java
@@ -0,0 +1,239 @@
+/*
+ * Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * Redistributions of source code must retain the above copyright notice,
+ * this list of conditions and the following disclaimer.
+ * Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ * Neither the name of the dreamlu.net developer nor the names of its
+ * contributors may be used to endorse or promote products derived from
+ * this software without specific prior written permission.
+ * Author: Chill 庄骞 (smallchill@163.com)
+ */
+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 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;
+import org.springblade.core.tenant.TenantId;
+import org.springblade.core.tool.constant.BladeConstant;
+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.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.core.cache.constant.CacheConstant.SYS_CACHE;
+
+/**
+ * 服务实现类
+ *
+ * @author Chill
+ */
+@Service
+@AllArgsConstructor
+public class TenantServiceImpl extends BaseServiceImpl<TenantMapper, Tenant> implements ITenantService {
+
+ private final TenantId tenantId;
+ private final IRoleService roleService;
+ private final IMenuService menuService;
+ private final IDeptService deptService;
+ private final IPostService postService;
+ private final IRoleMenuService roleMenuService;
+ private final IDictBizService dictBizService;
+ private final IUserService userService;
+
+ @Override
+ public IPage<Tenant> selectTenantPage(IPage<Tenant> page, Tenant tenant) {
+ return page.setRecords(baseMapper.selectTenantPage(page, tenant));
+ }
+
+ @Override
+ public Tenant getByTenantId(String tenantId) {
+ return getOne(Wrappers.<Tenant>query().lambda().eq(Tenant::getTenantId, tenantId));
+ }
+
+ @Override
+ @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);
+ }
+ CacheUtil.clear(SYS_CACHE, tenant.getTenantId());
+ return super.saveOrUpdate(tenant);
+ }
+
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public boolean removeTenant(List<Long> ids) {
+ List<String> tenantIds = this.list(Wrappers.<Tenant>query().lambda().in(Tenant::getId, ids))
+ .stream().map(tenant -> Func.toStr(tenant.getTenantId())).distinct().collect(Collectors.toList());
+ CacheUtil.clear(SYS_CACHE, tenantIds);
+ if (tenantIds.contains(BladeConstant.ADMIN_TENANT_ID)) {
+ throw new ServiceException("不可删除管理租户!");
+ }
+ boolean tenantTemp = this.deleteLogic(ids);
+ boolean userTemp = userService.remove(Wrappers.<User>query().lambda().in(User::getTenantId, tenantIds));
+ return tenantTemp && userTemp;
+ }
+
+ @Override
+ public boolean setting(Integer accountNumber, Date expireTime, String ids) {
+ List<String> tenantIds = this.list(Wrappers.<Tenant>query().lambda().in(Tenant::getId, ids))
+ .stream().map(tenant -> Func.toStr(tenant.getTenantId())).distinct().collect(Collectors.toList());
+ CacheUtil.clear(SYS_CACHE, tenantIds);
+ Func.toLongList(ids).forEach(id -> {
+ Kv kv = Kv.create().set("accountNumber", accountNumber).set("expireTime", expireTime).set("id", id);
+ String licenseKey = DesUtil.encryptToHex(JsonUtil.toJson(kv), DES_KEY);
+ update(
+ Wrappers.<Tenant>update().lambda()
+ .set(Tenant::getAccountNumber, accountNumber)
+ .set(Tenant::getExpireTime, expireTime)
+ .set(Tenant::getLicenseKey, licenseKey)
+ .eq(Tenant::getId, id)
+ );
+ });
+ 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);
+ });
+ }
+
+}
--
Gitblit v1.9.3