TTInfoDetailServiceImpl.java
5.88 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package com.bsth.service.schedule;
import com.bsth.entity.schedule.TTInfoDetail;
import com.bsth.service.impl.BaseServiceImpl;
import com.bsth.service.schedule.utils.DataImportExportService;
import com.bsth.service.schedule.utils.DataToolsProperties;
import jxl.Sheet;
import jxl.Workbook;
import org.apache.commons.lang3.StringUtils;
import org.pentaho.di.trans.Trans;
import org.pentaho.di.trans.TransMeta;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
/**
* Created by xu on 16/7/2.
*/
@Service
@EnableConfigurationProperties(DataToolsProperties.class)
public class TTInfoDetailServiceImpl extends BaseServiceImpl<TTInfoDetail, Long> implements TTInfoDetailService {
@Autowired
private DataImportExportService dataImportExportService;
@Autowired
private DataToolsProperties dataToolsProperties;
/**
* 获取待编辑的数据。
* @param ttid 时刻表id
* @return
*/
public Object getEditInfo(Long ttid) throws Exception {
// 1、使用ktr转换获取输出文件
// 1.1、获取转换用ktr
File ktrFile = new File(this.getClass().getResource(
dataToolsProperties.getTtinfodetailForeditktr()).toURI());
TransMeta transMeta = new TransMeta(ktrFile.getAbsolutePath());
Trans trans = new Trans(transMeta);
// 1.2、设定命名参数，TODO：之后还要添加其他命名参数
String outputFilePath = "ttinfodetail_" + (new Date()).getTime();
trans.setParameterValue("tempfilepath", dataToolsProperties.getTransTempdir() + File.separator + outputFilePath); // 数据输出文件路径
// 1.3、执行转换
trans.execute(null);
// 1.4、等待转换结束
trans.waitUntilFinished();
// 3、判定ktr错误数，注意这种错误代表部分数据错误，不会终止转换执行，一般设计ktr的时候，会有错误输出文件，TODO：以后考虑使用日志实时输出
if (trans.getErrors() > 0) {
throw new Exception("转换数据部分错误，请查看相关错误输出文件！");
}
// TODO：
return new Object();
}
/**
* 上传并导入数据，和DataImportExportService的同名方法有差别。
* @param datafile form上传文件
* @param xlmc 线路名称
* @param ttinfoname 时刻表名字
* @param tccname 停车场名字
* @throws Exception
*/
public void fileDataImport(MultipartFile datafile,
String xlmc,
String ttinfoname,
String tccname) throws Exception {
// 1、上传数据文件
File uploadFile = dataImportExportService.uploadFile(datafile);
System.out.println("线路名称：" + xlmc);
System.out.println("时刻表名称：" + ttinfoname);
System.out.println("停车场名字：" + tccname);
System.out.println("时刻表明细上传文件：" + uploadFile);
// 2、jexcelapi读取excel文件
Workbook book = Workbook.getWorkbook(uploadFile);
Sheet sheet = book.getSheet(0);
List<String> columnames = new ArrayList<>();
for (int i = 0; i < sheet.getColumns(); i++) { // 获取第一行，数据，作为列名
columnames.add(sheet.getCell(i, 0).getContents());
}
System.out.println("表头1：" + StringUtils.join(columnames.toArray(), ","));
// 2、使用kettle运行封装数据导入逻辑的ktr转换文件
// 2.1、初始化kettle（组件初始化已经做了）
// 2.2、创建转换元数据，转换
File ktrFile = new File(this.getClass().getResource(
dataToolsProperties.getTtinfodetailMetadatainputktr()).toURI());
File ktrFile2 = new File(this.getClass().getResource(
dataToolsProperties.getTtinfodetailDatainputktr()).toURI());
TransMeta transMeta = new TransMeta(ktrFile.getAbsolutePath());
Trans trans = new Trans(transMeta);
// 2.3、设定命名参数，用于指定数据文件，注意每个ktr必须都有以下指定的命名参数
trans.setParameterValue("injectktrfile", ktrFile2.getAbsolutePath()); // 注入元数据的ktr文件
trans.setParameterValue("filepath", uploadFile.getAbsolutePath()); // 指定导入数据文件的位置
trans.setParameterValue("erroroutputdir", dataToolsProperties.getTransErrordir()); // ktr转换错误输出目录
trans.setParameterValue("xlname", xlmc); // 线路名称
trans.setParameterValue("ttinfoname", ttinfoname); // 时刻表名称
trans.setParameterValue("tccname", tccname); // 停车场名字
trans.setParameterValue("excelfieldnames", StringUtils.join(columnames.toArray(), ",")); // 时刻表excel输入字段名，以逗号连接
columnames.remove(0);
trans.setParameterValue("normalizefieldnames", StringUtils.join(columnames.toArray(), ",")); // 数据范式化字段名，以逗号连接
// TODO：可以考虑设定日志输出
// 2.4、执行转换
trans.execute(null);
// 2.5、等待转换结束
trans.waitUntilFinished();
// 3、判定ktr错误数，注意这种错误代表部分数据错误，不会终止转换执行，一般设计ktr的时候，会有错误输出文件，TODO：以后考虑使用日志实时输出
if (trans.getErrors() > 0) {
throw new Exception("转换数据部分错误，请查看相关错误输出文件！");
}
}
}