test orgcreate

src/test/java/com/gct/tools/etlcamelhuge/EtlCamelHugeApplicationTests.java

@@ -2,6 +2,7 @@ package com.gct.tools.etlcamelhuge;
 
 import com.gct.tools.etlcamelhuge.routeconfig.CamelJDBCConfiguration;
 import lombok.Data;
+import org.json.JSONObject;
 import org.junit.jupiter.api.Test;
 import org.junit.runner.RunWith;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -15,10 +16,7 @@ import java.lang.reflect.Array;
 import java.lang.reflect.InvocationTargetException;
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 @RunWith(SpringRunner.class)
 @SpringBootTest
@@ -27,6 +25,9 @@ class EtlCamelHugeApplicationTests {
     @Resource(name = "oracle")
     DataSource oracle;
 
+    @Resource(name="centralbase")
+    DataSource centralbase;
+
     @Test
     public void testOracle(){
         JdbcTemplate template = new JdbcTemplate(oracle);
@@ -38,4 +39,40 @@ class EtlCamelHugeApplicationTests {
         });
     }
 
+    @Test
+    public void orgCreate(){
+        JdbcTemplate oracleTemplate = new JdbcTemplate(oracle);
+        final JdbcTemplate centralbaseTemplate = new JdbcTemplate(centralbase);
+        String sql = String.format("select * from zy_dxgw.V_PC_ORGANIZATION_T where CANTON = '重油公司'");
+        List<Map<String, Object>> results = oracleTemplate.queryForList(sql);
+
+        Map<String,Map<String,Object>> parentObjMap = new HashMap<>();
+        Map<String, Map<String,Object>> orgIdMap = new HashMap<>();
+        results.forEach(rowMap->{
+            orgIdMap.put(rowMap.get("ORG_ID").toString(),rowMap);
+        });
+        results.forEach(rowMap->{
+            parentObjMap.put(rowMap.get("ORG_ID").toString(),orgIdMap.get(rowMap.get("PARENT_ID")));
+        });
+
+        results.forEach(rowMap->{
+            Stack<String> orgPreStack = new Stack<>();
+            Map<String,Object> parentObj = null;
+            while((parentObj=parentObjMap.get(rowMap.get("PARENT_ID").toString()))!=null){
+                orgPreStack.push(parentObj.get("ORG_NAME").toString());
+            }
+            StringBuilder orgIdPre = new StringBuilder();
+            while (!orgPreStack.isEmpty()){
+                orgIdPre.append("@").append(orgPreStack.pop());
+            }
+            String orgIdPreStr = orgIdPre.delete(0, 1).toString();
+            String sqlInsertOrg = String.format("insert into centralbase.cb_pc_organization(" +
+                    "org_id,org_name,org_code,org_level,parent_id,org_id_pre) " +
+                    "values(%s,%s,%s,%s,%s,%s)",rowMap.get("ORG_ID"),rowMap.get("ORG_NAME"),rowMap.get("ORG_CODE"),rowMap.get("ORG_LEVEL")
+                    ,rowMap.get("PARENT_ID"),orgIdPreStr);
+            System.out.println(sqlInsertOrg);
+//            centralbaseTemplate.execute(sqlInsertOrg);
+
+        });
+    }
 }