|
@@ -55,6 +55,7 @@ import org.springblade.core.boot.ctrl.BladeController;
|
|
|
import java.rmi.ServerException;
|
|
|
import java.util.Arrays;
|
|
|
import java.util.List;
|
|
|
+import java.util.Objects;
|
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
/**
|
|
@@ -137,7 +138,7 @@ public class TableFileController extends BladeController {
|
|
|
updateWrapper.set("tab_file_type", 1);
|
|
|
wbsTreeContractService.update(updateWrapper);
|
|
|
WbsTreeContract wbsTreeContract = wbsTreeContractService.getBaseMapper().selectOne(new QueryWrapper<WbsTreeContract>().eq("p_key_id", tableFile.getTabId()));
|
|
|
- if(wbsTreeContract.getNodeClass().equals(Integer.valueOf(2))){
|
|
|
+ if(Objects.equals(wbsTreeContract.getNodeClass(), 2)){
|
|
|
Integer classify =
|
|
|
Arrays.asList("1", "2", "3").contains(wbsTreeContract.getTableOwner()) ? 1 :
|
|
|
Arrays.asList("4", "5", "6").contains(wbsTreeContract.getTableOwner()) ? 2 : 1;
|