EmployeeDataToolsImpl.java
5.74 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
130
131
132
133
134
135
136
137
138
139
140
141
142
package com.bsth.service.schedule.datatools;
import com.bsth.service.schedule.exception.ScheduleException;
import com.bsth.service.schedule.utils.*;
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 = "employee_dataTool")
public class EmployeeDataToolsImpl implements DataToolsService {
/** 日志记录器 */
private final static Logger LOGGER = LoggerFactory.getLogger(EmployeeDataToolsImpl.class);
@Autowired
@Qualifier(value = "dataToolsServiceImpl")
private DataToolsService dataToolsService;
@Autowired
private DataToolsProperties dataToolsProperties;
@Override
public DataToolsFile uploadFile(String filename, byte[] filedata) throws ScheduleException {
try {
// 对上传的excel文件做处理
// 将第一个sheet保存成一个xls文件
DataToolsFile dataToolsFile = dataToolsService.uploadFile(filename, filedata);
File file = dataToolsFile.getFile();
// poi api
org.apache.poi.ss.usermodel.Workbook poi_workbook = dataToolsFile.getFileType().getWorkBook(file);
org.apache.poi.ss.usermodel.Sheet poi_sheet = poi_workbook.getSheetAt(0); // 第一个sheet
int rowNum = poi_sheet.getLastRowNum(); // 获取总共多少行
if (rowNum < 0) {
throw new RuntimeException("表格内容为空！");
}
int colNum = poi_sheet.getRow(0).getLastCellNum(); // 获取总共多少列，以第一行为主
// jxl api
File fileCal = new File(file.getAbsolutePath() + "_sheetChange.xls");
WritableWorkbook writableWorkbook = jxl.Workbook.createWorkbook(fileCal);
WritableSheet writableSheet = writableWorkbook.createSheet("工作表1", 0);
for (int i = 0; i <= rowNum; i++) {
for (int j = 0; j <= colNum; j++) {
// poi读
String cellContent = PoiUtils.getStringValueFromCell(poi_sheet.getRow(i).getCell(j));
// jxl写
writableSheet.addCell(new Label(j, i, cellContent));
}
}
writableWorkbook.write();
writableWorkbook.close();
poi_workbook.close();
DataToolsFile dataToolsFile1 = new DataToolsFile();
dataToolsFile1.setFile(fileCal);
dataToolsFile1.setFileType(DataToolsFileType.XLS);
return dataToolsFile1;
} 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.getEmployeesDatainputktr()).toURI());
// 通用参数，转换文件路径，excel输入文件路径，错误输出文件路径
ktrParms.put("transpath", ktrFile.getAbsolutePath());
ktrParms.put("filepath", file.getAbsolutePath());
ktrParms.put("erroroutputdir", dataToolsProperties.getTransErrordir());
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 DataToolsFile 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.getEmployeesDataoutputktr()).toURI());
// 通用参数，转换文件路径，excel输出文件名
ktrParms.put("transpath", ktrFile.getAbsolutePath());
ktrParms.put("filename", "人员基础信息_download-");
ktrParms.putAll(params);
DataToolsFile 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());
}
}
}