well source创建完成

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

@@ -17,6 +17,7 @@ import java.lang.reflect.InvocationTargetException;
 import java.time.LocalDate;
 import java.time.format.DateTimeFormatter;
 import java.util.*;
+import java.util.concurrent.atomic.AtomicInteger;
 
 @RunWith(SpringRunner.class)
 @SpringBootTest(classes = EtlCamelHugeApplication.class)
@@ -48,7 +49,9 @@ class EtlCamelHugeApplicationTests {
 
         Map<String, Map<String, Object>> parentObjMap = new HashMap<>();
         Map<String, Map<String, Object>> orgIdMap = new HashMap<>();
+        AtomicInteger a = new AtomicInteger(1);
         results.forEach(rowMap -> {
+            rowMap.put("org_id_map_int",a.getAndIncrement());
             orgIdMap.put(rowMap.get("ORG_ID").toString(), rowMap);
         });
         results.forEach(rowMap -> {
@@ -72,12 +75,15 @@ class EtlCamelHugeApplicationTests {
 
             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);
+                            "values('%s','%s','%s','%s','%s','%s')", rowMap.get("org_id_map_int"), rowMap.get("ORG_NAME"), rowMap.get("ORG_CODE"),
+                    orgIdPreStr.split("@").length
+                    , Objects.nonNull(orgIdMap.get(rowMap.get("PARENT_ID").toString()))?orgIdMap.get(rowMap.get("PARENT_ID").toString()).get("org_id_map_int"):"0"
+                    , orgIdPreStr);
 //            System.out.println(sqlInsertOrg);
-//            centralbaseTemplate.execute(sqlInsertOrg);   //excute finish, unused
+            centralbaseTemplate.execute(sqlInsertOrg);   //excute finish, unused
         });
 
+        if(a.get()>1)return;
         //excute wellsource
         sql = String.format("select WELL_ID,well_common_name,org_id from " +
                 "zy_dxgw.pc_dev_well_attr_info where well_purpose_name='采油井'");
@@ -87,11 +93,13 @@ class EtlCamelHugeApplicationTests {
             String wellCommonName = rowMap.get("WELL_COMMON_NAME").toString();
             String wellLegalName = rowMap.get("WELL_COMMON_NAME").toString();
             String orgId = rowMap.get("ORG_ID").toString();
+            String orgIdMapInt = orgIdMap.get(orgId).get("org_id_map_int").toString();
             String stationId = orgIdMap.get(orgId).get("ORG_ID_PRE").toString();
-           String  sqlInsertWellSource = String.format("insert into centralbase.cb_pc_well_source(well_id,well_common_name," +
+           String  sqlInsertWellSource = String.format("insert into centralbase.cb_cd_well_source(well_id,well_common_name," +
                    "well_legal_name,org_id,station_id) values('%s','%s','%s','%s','%s')",
-                   wellId, wellCommonName, wellLegalName, orgId, stationId);
-            System.out.println(sqlInsertWellSource);
+                   wellId, wellCommonName, wellLegalName, orgIdMapInt, stationId);
+//            System.out.println(sqlInsertWellSource);
+            centralbaseTemplate.execute(sqlInsertWellSource);
         });
 
     }