时刻表数据带线路版本v2.3

1、修正导入bug，支持只有站名的形式，内部自动加{->数字}格式 2、修正导出格式，站名不带{->数字}格式

时刻表数据带线路版本v2.3
1、修正导入bug，支持只有站名的形式，内部自动加{->数字}格式 2、修正导出格式，站名不带{->数字}格式
徐烜
1 parent 217aeefa
Showing 1 changed file with 46 additions and 1 deletions
src/main/java/com/bsth/service/schedule/datatools/TTInfoDetailDataToolsImpl.java
@@ -128,7 +128,8 @@ public class TTInfoDetailDataToolsImpl implements DataToolsService, TTInfoDetail
                 if (i == 0) {
                     colList.add(PoiUtils.getStringValueFromCell(cell).trim());
                 } else {
-                    colList.add(PoiUtils.getStringValueFromCell(cell) + i);
+                    // 站点名字后加->数字形式标示成不同的字段
+                    colList.add(PoiUtils.getStringValueFromCell(cell).trim().replaceAll("(->\\d+)", "") + "->" + i);
                 }
             }
  
@@ -229,6 +230,50 @@ public class TTInfoDetailDataToolsImpl implements DataToolsService, TTInfoDetail
  
             DataToolsFile file = dataToolsService.exportData(ktrParms);
  
+            // 将导出的数据表头重新处理一遍，祛除->数字
+
+            // poi api，并读取第一行数据，组合成站点列表，逗号分隔
+            org.apache.poi.ss.usermodel.Workbook poi_workbook;
+            org.apache.poi.ss.usermodel.Sheet poi_sheet;
+            if (DataToolsFileType.XLS.isThisType(file.getFile())) {
+                poi_workbook = DataToolsFileType.XLS.getWorkBook(file.getFile());
+            } else if (DataToolsFileType.XLSX.isThisType(file.getFile())) {
+                poi_workbook = DataToolsFileType.XLSX.getWorkBook(file.getFile());
+            } else {
+                throw new Exception("不是xls xlsx文件!");
+            }
+            poi_sheet = poi_workbook.getSheet("Sheet1");
+            List<String> colList = new ArrayList<>();
+            int rownums = poi_sheet.getLastRowNum() + 1;
+            int colnums = poi_sheet.getRow(0).getLastCellNum();
+            Row firstrow = poi_sheet.getRow(0);
+            for (int i = 0; i < colnums; i++) {
+                org.apache.poi.ss.usermodel.Cell cell = firstrow.getCell(i);
+                colList.add(PoiUtils.getStringValueFromCell(cell).trim().replaceAll("(->\\d+)", ""));
+            }
+
+            // jxl api
+            File fileCal = new File(file.getFile().getAbsolutePath() + "_rowHeadReplace.xls");
+            WritableWorkbook writableWorkbook = Workbook.createWorkbook(fileCal);
+            WritableSheet sheet1 = writableWorkbook.createSheet("Sheet1", 0);
+            for (int i = 0; i < colnums; i++) { // 第一行数据
+                sheet1.addCell(new Label(i, 0, colList.get(i)));
+            }
+            for (int i = 1; i < rownums; i++) { // 第二行开始
+                for (int j = 0; j < colnums; j++) {
+                    // poi读
+                    String cellContent = PoiUtils.getStringValueFromCell(
+                            poi_sheet.getRow(i).getCell(j)
+                    ).replaceAll("\\s*", "");
+                    // jxl写
+                    sheet1.addCell(new Label(j, i, cellContent));
+                }
+            }
+            writableWorkbook.write();
+            writableWorkbook.close();
+
+            file.setFile(fileCal);
+
             LOGGER.info("//---------------- 导出时刻表明细 success... ----------------//");
  
             return file;