liuyc 2 жил өмнө
parent
commit
911f915f93

+ 4 - 0
blade-service/blade-manager/src/main/java/org/springblade/manager/mapper/WbsTreePrivateMapper.java

@@ -4,6 +4,7 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
 import org.apache.ibatis.annotations.Param;
 import org.springblade.manager.dto.WbsTreePrivateDTO2;
 import org.springblade.manager.dto.WbsTreePrivateDTO3;
+import org.springblade.manager.entity.TableInfo;
 import org.springblade.manager.entity.WbsTree;
 import org.springblade.manager.entity.WbsTreePrivate;
 import org.springblade.manager.injector.EasyBaseMapper;
@@ -89,4 +90,7 @@ public interface WbsTreePrivateMapper extends EasyBaseMapper<WbsTreePrivate> {
 
     //删除表单信息
     void delTableById(String pKeyId);
+
+    List<TableInfo> selectTabInfoAll();
+
 }

+ 4 - 0
blade-service/blade-manager/src/main/java/org/springblade/manager/mapper/WbsTreePrivateMapper.xml

@@ -658,6 +658,10 @@
         ORDER BY sort,node_name,create_time
     </select>
 
+    <select id="selectTabInfoAll" resultType="org.springblade.manager.entity.TableInfo">
+        select * from m_table_info
+    </select>
+
     <delete id="delTabProjectById">
         delete from m_wbs_tree_private WHERE p_key_id = #{pKeyId} and project_id=#{projectId}AND type = 10
     </delete>

+ 20 - 0
blade-service/blade-manager/src/main/java/org/springblade/manager/service/impl/WbsTreePrivateServiceImpl.java

@@ -31,6 +31,7 @@ import org.springblade.manager.service.ITableInfoService;
 import org.springblade.manager.service.IWbsTreePrivateService;
 import org.springblade.manager.vo.*;
 import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -1514,4 +1515,23 @@ public class WbsTreePrivateServiceImpl extends BaseServiceImpl<WbsTreePrivateMap
         }
         this.insertBatch(wbsTreePrivatesNew, 1000);
     }
+
+    /**
+     * 修改initTableId
+     */
+    @Scheduled(cron = "0 0 3 * * ?")
+    public void updateInitTableId() {
+        List<WbsTreePrivate> wbsTreePrivates = baseMapper.selectList(Wrappers.<WbsTreePrivate>lambdaQuery().isNull(WbsTreePrivate::getInitTableId).isNotNull(WbsTreePrivate::getInitTableName).ne(WbsTreePrivate::getType, 1));
+        List<TableInfo> list = baseMapper.selectTabInfoAll();
+        for (TableInfo tableInfo : list) {
+            for (WbsTreePrivate wbsTreePrivate : wbsTreePrivates) {
+                if (tableInfo.getTabEnName().equals(wbsTreePrivate.getInitTableName()) || tableInfo.getTabChName().equals(wbsTreePrivate.getInitTableName())) {
+                    if (ObjectUtil.isEmpty(wbsTreePrivate.getInitTableId())) {
+                        this.update(Wrappers.<WbsTreePrivate>lambdaUpdate().set(WbsTreePrivate::getInitTableId, String.valueOf(tableInfo.getId())).eq(WbsTreePrivate::getPKeyId, wbsTreePrivate.getPKeyId()));
+                    }
+                }
+            }
+        }
+    }
+
 }