From 6e15fb58d1530c5dc28be872f950b2d5758246f8 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期日, 27 十一月 2022 08:50:48 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/service/data/DataLoaderService.java |   31 +++++++++++++++++--------------
 1 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/src/main/java/com/lf/server/service/data/DataLoaderService.java b/src/main/java/com/lf/server/service/data/DataLoaderService.java
index efe5704..6229f84 100644
--- a/src/main/java/com/lf/server/service/data/DataLoaderService.java
+++ b/src/main/java/com/lf/server/service/data/DataLoaderService.java
@@ -242,11 +242,11 @@
                         case "shp":
                         case "gdb":
                         case "mdb":
-                            insertDb(type, temp, mf, tab);
+                            insertDb(type, temp, entity, mf, tab);
                             break;
                         // case "xls":
                         default:
-                            insertXls(temp, mf, tab);
+                            insertXls(temp, entity, mf, tab);
                             break;
                     }
                     if (0 == tab.getRows()) {
@@ -354,7 +354,7 @@
     /**
      * 鎻掑叆Excel
      */
-    private void insertXls(String root, MetaFileEntity mf, TabMapperEntity tab) {
+    private void insertXls(String root, MetaEntity me, MetaFileEntity mf, TabMapperEntity tab) {
         BasicMapper basicMapper = ClassHelper.getBasicMapper(tab.getEntity());
         if (null == basicMapper) {
             return;
@@ -371,7 +371,7 @@
         if (list.isEmpty()) {
             return;
         }
-        setCreateInfo(list, mf);
+        setCreateInfo(list, me);
 
         int rows = 0;
         for (int i = 0, c = list.size(); i < c; i++) {
@@ -388,16 +388,19 @@
     /**
      * 璁剧疆鍒涘缓淇℃伅
      */
-    private <T> void setCreateInfo(List<T> list, MetaFileEntity mf) {
+    private <T> void setCreateInfo(List<T> list, MetaEntity me) {
         try {
-            for (T t : list) {
-                Field cuField = t.getClass().getDeclaredField("createuser");
-                cuField.setAccessible(true);
-                cuField.set(t, mf.getCreateUser());
+            if (!(list.get(0) instanceof BaseEntity)) {
+                return;
+            }
 
-                Field ctField = t.getClass().getDeclaredField("createtime");
-                ctField.setAccessible(true);
-                ctField.set(t, mf.getCreateTime());
+            for (T t : list) {
+                BaseEntity be = (BaseEntity) t;
+                be.setCreateuser(me.getCreateUser());
+                be.setCreatetime(me.getCreateTime());
+                be.setDirid(me.getDirid());
+                be.setDepid(me.getDepid());
+                be.setVerid(me.getVerid());
             }
         } catch (Exception ex) {
             log.error(ex.getMessage(), ex);
@@ -433,7 +436,7 @@
     /**
      * 鎻掑叆DB
      */
-    private void insertDb(String type, String root, MetaFileEntity mf, TabMapperEntity tab) {
+    private void insertDb(String type, String root, MetaEntity me, MetaFileEntity mf, TabMapperEntity tab) {
         BasicMapper basicMapper = ClassHelper.getBasicMapper(tab.getEntity());
         if (null == basicMapper) {
             return;
@@ -464,7 +467,7 @@
         if (null == list || list.isEmpty()) {
             return;
         }
-        setCreateInfo(list, mf);
+        setCreateInfo(list, me);
 
         int rows = basicMapper.insertBatch(list);
         tab.setTab(tabName);

--
Gitblit v1.9.3