Merge branch '20230804' into 'master_dev'

fix: 新建租户数据无法处理问题修复

See merge request yunteng/thingskit!220

@@ -205,17 +205,26 @@ public class TkAdminController extends BaseController {
 
 
   @PutMapping("/menu/assign/{tenantId}")
   @PutMapping("/menu/assign/{tenantId}")
   public void assignMenuToTenant(
   public void assignMenuToTenant(
-      @RequestBody String[] menuIds, @PathVariable("tenantId") String tenantId){
+      @RequestBody String[] menuIds, @PathVariable("tenantId") String tenantId) {
     menuService.assignMenuToTenant(tenantId, menuIds);
     menuService.assignMenuToTenant(tenantId, menuIds);
   }
   }
 
 
   @PostMapping("tenant/updateOrCreateTenant")
   @PostMapping("tenant/updateOrCreateTenant")
-  @Transactional
   public TenantDTO updateOrCreateTenant(
   public TenantDTO updateOrCreateTenant(
       @Validated(UpdateGroup.class) @RequestBody TenantReqDTO tenantReqDTO)
       @Validated(UpdateGroup.class) @RequestBody TenantReqDTO tenantReqDTO)
-      throws ThingsboardException, IOException {
-    updateOrSaveTenant(tenantReqDTO);
-    return tkTenantService.updateOrCreateTenant(tenantReqDTO);
+      throws ThingsboardException {
+    boolean isCreate = StringUtils.isBlank(tenantReqDTO.getTenantId());
+    TenantDTO result = null;
+    try {
+      updateOrSaveTenant(tenantReqDTO, isCreate);
+      result = tkTenantService.updateOrCreateTenant(tenantReqDTO, isCreate);
+    } catch (Exception e) {
+      if (isCreate) {
+        deleteTenant(tenantReqDTO.getTenantId());
+        throw handleException(e);
+      }
+    }
+    return result;
   }
   }
 
 
   @GetMapping("all/tenants")
   @GetMapping("all/tenants")
@@ -253,34 +262,40 @@ public class TkAdminController extends BaseController {
     }
     }
   }
   }
 
 
-  private void updateOrSaveTenant(TenantReqDTO tenantReqDTO) throws IOException {
-      boolean isCreate = tenantReqDTO.getTenantId() == null;
-      Tenant tbTenant;
-      if (isCreate) {
-        // 增加TB的租户创建
-        tbTenant = new Tenant();
-      } else {
-        tbTenant = new Tenant(TenantId.fromUUID(UUID.fromString(tenantReqDTO.getTenantId())));
-      }
-      tbTenant.setTenantProfileId(tenantReqDTO.getTenantProfileId());
-      tbTenant.setTitle(tenantReqDTO.getName());
-      tbTenant = tenantService.saveTenant(tbTenant);
-      if (isCreate) {
-        installScripts.createDefaultRuleChains(tbTenant.getId());
-        installScripts.createDefaultEdgeRuleChains(tbTenant.getId());
-      }
-      tenantProfileCache.evict(tbTenant.getId());
-      tbClusterService.onTenantChange(tbTenant, null);
-      tbClusterService.broadcastEntityStateChangeEvent(
-          tbTenant.getId(),
-          tbTenant.getId(),
-          isCreate ? ComponentLifecycleEvent.CREATED : ComponentLifecycleEvent.UPDATED);
-      tenantReqDTO.setTenantId(tbTenant.getTenantId().toString());
+  /**
+   * 实现TB创建租户的原生逻辑
+   *
+   * @param tenantReqDTO
+   * @param isCreate
+   * @throws IOException
+   */
+  private void updateOrSaveTenant(TenantReqDTO tenantReqDTO, boolean isCreate) throws IOException {
+    Tenant tbTenant;
+    if (isCreate) {
+      // 增加TB的租户创建
+      tbTenant = new Tenant();
+    } else {
+      tbTenant = new Tenant(TenantId.fromUUID(UUID.fromString(tenantReqDTO.getTenantId())));
+    }
+    tbTenant.setTenantProfileId(tenantReqDTO.getTenantProfileId());
+    tbTenant.setTitle(tenantReqDTO.getName());
+    tbTenant = tenantService.saveTenant(tbTenant);
+    if (isCreate) {
+      installScripts.createDefaultRuleChains(tbTenant.getId());
+      installScripts.createDefaultEdgeRuleChains(tbTenant.getId());
+    }
+    tenantProfileCache.evict(tbTenant.getId());
+    tbClusterService.onTenantChange(tbTenant, null);
+    tbClusterService.broadcastEntityStateChangeEvent(
+        tbTenant.getId(),
+        tbTenant.getId(),
+        isCreate ? ComponentLifecycleEvent.CREATED : ComponentLifecycleEvent.UPDATED);
+    tenantReqDTO.setTenantId(tbTenant.getTenantId().toString());
   }
   }
 
 
-  private void deleteTenant(String strTenantId) throws ThingsboardException {
+  private void deleteTenant(String tbTenantId) throws ThingsboardException {
     try {
     try {
-      TenantId tenantId = TenantId.fromUUID(toUUID(strTenantId));
+      TenantId tenantId = TenantId.fromUUID(toUUID(tbTenantId));
       Tenant tenant = checkTenantId(tenantId, Operation.DELETE);
       Tenant tenant = checkTenantId(tenantId, Operation.DELETE);
       tenantService.deleteTenant(tenantId);
       tenantService.deleteTenant(tenantId);
       tenantProfileCache.evict(tenantId);
       tenantProfileCache.evict(tenantId);

@@ -265,8 +265,10 @@ public class TkTenantServiceImpl extends AbstractBaseService<TenantMapper, SysTe
 
 
   @Override
   @Override
   @Transactional
   @Transactional
-  public TenantDTO updateOrCreateTenant(TenantReqDTO tenantReqDTO) {
-    if (StringUtils.isNotBlank(tenantReqDTO.getId())) {
+  public TenantDTO updateOrCreateTenant(TenantReqDTO tenantReqDTO,boolean isCreate) {
+    if (isCreate) {
+      return createNewTenant(tenantReqDTO);
+    } else {
       SysTenantEntity tenant;
       SysTenantEntity tenant;
       tenant = baseMapper.selectById(tenantReqDTO.getId());
       tenant = baseMapper.selectById(tenantReqDTO.getId());
       if (tenant != null) {
       if (tenant != null) {
@@ -286,8 +288,6 @@ public class TkTenantServiceImpl extends AbstractBaseService<TenantMapper, SysTe
         baseMapper.updateById(tenant);
         baseMapper.updateById(tenant);
       }
       }
       return tenant.getDTO(TenantDTO.class);
       return tenant.getDTO(TenantDTO.class);
-    } else {
-      return createNewTenant(tenantReqDTO);
     }
     }
   }
   }
 
 

@@ -25,7 +25,7 @@ public interface TkTenantService {
 
 
   TkPageData<TenantDTO> page(Map<String, Object> queryMap);
   TkPageData<TenantDTO> page(Map<String, Object> queryMap);
 
 
-  TenantDTO updateOrCreateTenant(TenantReqDTO tenantReqDTO);
+  TenantDTO updateOrCreateTenant(TenantReqDTO tenantReqDTO,boolean isCreate);
 
 
   List<String> getTenantRolesByTenantId(String tenantId);
   List<String> getTenantRolesByTenantId(String tenantId);