EmployeeConfigInfoDataToolsImpl.java
5.08 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
126
127
128
129
package com.bsth.service.schedule.datatools;
import com.bsth.service.schedule.exception.ScheduleException;
import com.bsth.service.schedule.utils.DataToolsProperties;
import com.bsth.service.schedule.utils.DataToolsService;
import jxl.Cell;
import jxl.Sheet;
import jxl.Workbook;
import jxl.write.Label;
import jxl.write.WritableSheet;
import jxl.write.WritableWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.stereotype.Service;
import java.io.File;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;
/**
* Created by xu on 17/5/16.
*/
@EnableConfigurationProperties(DataToolsProperties.class)
@Service(value = "employeeConfig_dataTool")
public class EmployeeConfigInfoDataToolsImpl implements DataToolsService {
/** 日志记录器 */
private static final Logger LOGGER = LoggerFactory.getLogger(EmployeeConfigInfoDataToolsImpl.class);
@Autowired
@Qualifier(value = "dataToolsServiceImpl")
private DataToolsService dataToolsService;
@Autowired
private DataToolsProperties dataToolsProperties;
@Override
public File uploadFile(String filename, byte[] filedata) throws ScheduleException {
try {
// 对上传的excel文件做处理，将第一个sheet名字设定为工作表1
File file = dataToolsService.uploadFile(filename, filedata);
Workbook workbook = Workbook.getWorkbook(file);
Sheet sheet = workbook.getSheet(0);
File fileCal = new File(file.getAbsolutePath() + "_sheetChange.xls");
WritableWorkbook writableWorkbook = Workbook.createWorkbook(fileCal);
WritableSheet writableSheet = writableWorkbook.createSheet("工作表1", 0);
for (int i = 0; i < sheet.getRows(); i++) {
Cell[] cells = sheet.getRow(i);
for (int j = 0; j < cells.length; j++) {
writableSheet.addCell(new Label(j, i, cells[j].getContents()));
}
}
writableWorkbook.write();
writableWorkbook.close();
return fileCal;
} catch (Exception exp) {
throw new ScheduleException(exp);
}
}
@Override
public void importData(File file, Map<String, Object> params) throws ScheduleException {
try {
LOGGER.info("//---------------- 导入人员配置信息 start... ----------------//");
// 创建ktr转换所需参数
Map<String, Object> ktrParms = new HashMap<>();
File ktrFile = new File(this.getClass().getResource(
dataToolsProperties.getEmployeesconfigDatainputktr()).toURI());
// 通用参数，转换文件路径，excel输入文件路径，错误输出文件路径
ktrParms.put("transpath", ktrFile.getAbsolutePath());
ktrParms.put("filepath", file.getAbsolutePath());
ktrParms.put("erroroutputdir", dataToolsProperties.getTransErrordir());
ktrParms.putAll(params);
dataToolsService.importData(file, ktrParms);
LOGGER.info("//---------------- 导入人员配置信息 success... ----------------//");
} catch (Exception exp) {
LOGGER.info("//---------------- 导入人员配置信息 failed... ----------------//");
StringWriter sw = new StringWriter();
exp.printStackTrace(new PrintWriter(sw));
LOGGER.info(sw.toString());
throw new ScheduleException(exp.getMessage());
}
}
@Override
public File exportData(Map<String, Object> params) throws ScheduleException {
try {
LOGGER.info("//---------------- 导出人员配置信息 start... ----------------//");
// 创建ktr转换所需参数
Map<String, Object> ktrParms = new HashMap<>();
File ktrFile = new File(this.getClass().getResource(
dataToolsProperties.getEmployeesconfigDataoutputktr()).toURI());
// 通用参数，转换文件路径，excel输出文件名
ktrParms.put("transpath", ktrFile.getAbsolutePath());
ktrParms.put("filename", "人员配置信息_download-");
ktrParms.putAll(params);
File file = dataToolsService.exportData(ktrParms);
LOGGER.info("//---------------- 导出人员配置信息 success... ----------------//");
return file;
} catch (Exception exp) {
LOGGER.info("//---------------- 导出人员配置信息 failed... ----------------//");
StringWriter sw = new StringWriter();
exp.printStackTrace(new PrintWriter(sw));
LOGGER.info(sw.toString());
throw new ScheduleException(exp.getMessage());
}
}
}