BaseController.java
8.07 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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
package com.bsth.controller;
import com.bsth.common.ResponseCode;
import com.bsth.service.BaseService;
import com.bsth.service.schedule.utils.DataImportExportService;
import com.google.common.base.Splitter;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Sort;
import org.springframework.data.domain.Sort.Direction;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
*
* @ClassName: BaseController
* @Description: TODO(基础的Controller实现)
* @author PanZhao
* @date 2016年3月17日 下午12:44:06
*
* @param <T>
* @param <ID> 主键类型
*/
public class BaseController<T, ID extends Serializable> {
@Autowired
protected BaseService<T, ID> baseService;
@Autowired
DataImportExportService dataImportExportService;
/**
*
* @Title: list
* @Description: TODO(多条件分页查询)
* @param @param map 查询条件
* @param @param page 页码
* @param @param size 每页显示数量
* @throws
*/
@RequestMapping(method = RequestMethod.GET)
public Page<T> list(@RequestParam Map<String, Object> map,
@RequestParam(defaultValue = "0") int page,
@RequestParam(defaultValue = "10") int size,
@RequestParam(defaultValue = "id") String order,
@RequestParam(defaultValue = "DESC") String direction){
// 允许多个字段排序，order可以写单个字段，也可以写多个字段
// 多个字段格式：{col1},{col2},{col3},....,{coln}
List<String> order_columns = Splitter.on(",").trimResults().splitToList(order);
// 多字段排序：DESC,ASC...
List<String> order_dirs = Splitter.on(",").trimResults().splitToList(direction);
if (order_dirs.size() == 1) { // 所有字段采用一种排序
if (null != order_dirs.get(0) && order_dirs.get(0).equals("ASC")) {
return baseService.list(map, new PageRequest(page, size, new Sort(Direction.ASC, order_columns)));
} else {
return baseService.list(map, new PageRequest(page, size, new Sort(Direction.DESC, order_columns)));
}
} else if (order_columns.size() == order_dirs.size()) {
List<Sort.Order> orderList = new ArrayList<>();
for (int i = 0; i < order_columns.size(); i++) {
if (null != order_dirs.get(i) && order_dirs.get(i).equals("ASC")) {
orderList.add(new Sort.Order(Direction.ASC, order_columns.get(i)));
} else {
orderList.add(new Sort.Order(Direction.DESC, order_columns.get(i)));
}
}
return baseService.list(map, new PageRequest(page, size, new Sort(orderList)));
} else {
throw new RuntimeException("多字段排序参数格式问题，排序顺序和字段数不一致");
}
}
/**
*
* @Title: list
* @Description: TODO(多条件查询)
* @param @param map
* @throws
*/
@RequestMapping(value = "/all", method = RequestMethod.GET)
public Iterable<T> list(@RequestParam Map<String, Object> map){
return baseService.list(map);
}
/**
*
* @Title: save
* @Description: TODO(持久化对象)
* @param @param t
* @param @return 设定文件
* @return Map<String,Object> {status: 1（成功），-1（失败）}
* @throws
*/
@RequestMapping(method = RequestMethod.POST)
public Map<String, Object> save(T t){
return baseService.save(t);
}
/**
*
* @Title: findById
* @Description: TODO(根据主键获取单个对象)
* @param @param id
* @throws
*/
@RequestMapping(value="/{id}",method = RequestMethod.GET)
public T findById(@PathVariable("id") ID id){
return baseService.findById(id);
}
/**
*
* @Title: delete
* @Description: TODO(根据主键删除对象)
* @param @param id
* @throws
*/
@RequestMapping(value="/{id}",method = RequestMethod.DELETE)
public Map<String, Object> delete(@PathVariable("id") ID id){
return baseService.delete(id);
}
/**
* 上传数据文件，并使用ktr转换文件导入数据。
* @param file
* @return
* @throws Exception
*/
@RequestMapping(value = "/dataImport", method = RequestMethod.POST)
public Map<String, Object> uploadDataAndImport(MultipartFile file) throws Exception {
Map<String, Object> resultMap = new HashMap<>();
try {
// 获取ktr转换文件绝对路径
File ktrfile = new File(this.getClass().getResource(getDataImportKtrClasspath()).toURI());
System.out.println(ktrfile.getAbsolutePath());
// 导入数据
dataImportExportService.fileDataImport(file, ktrfile);
resultMap.put("status", ResponseCode.SUCCESS);
resultMap.put("msg", "导入成功");
} catch (Exception exp) {
exp.printStackTrace();
resultMap.put("status", ResponseCode.ERROR);
resultMap.put("msg", exp.getLocalizedMessage());
}
return resultMap;
}
/**
* 使用ktr导出数据。
* @param response
* @throws Exception
*/
@RequestMapping(value = "/dataExport", method = RequestMethod.GET)
public void dataExport(HttpServletResponse response) throws Exception {
// 1、使用ktr转换获取输出文件
File ktrfile = new File(this.getClass().getResource(getDataExportKtrClasspath()).toURI());
File outputfile = dataImportExportService.fileDataOutput(
getDataExportFilename(),
ktrfile);
System.out.println(outputfile.getName());
String filePath = outputfile.getAbsolutePath();
String fp[] = filePath.split(File.separator);
String fileName = fp[fp.length - 1];
// TODO：使用ktr获取导出数据
response.setHeader("conent-type", "application/octet-stream");
response.setContentType("application/octet-stream");
response.setHeader("Content-Disposition", "attachment; filename=" + "东东");
OutputStream os = response.getOutputStream();
BufferedOutputStream bos = new BufferedOutputStream(os);
InputStream is = null;
is = new FileInputStream(filePath);
BufferedInputStream bis = new BufferedInputStream(is);
int length = 0;
byte[] temp = new byte[1 * 1024 * 10];
while ((length = bis.read(temp)) != -1) {
bos.write(temp, 0, length);
}
bos.flush();
bis.close();
bos.close();
is.close();
}
/**
* @return 数据导出的ktr转换文件类路径。
*/
protected String getDataExportKtrClasspath() {
// 默认返回异常，子类如果要使用导出功能，必须覆写此方法，指定ktr文件类路径
throw new RuntimeException("必须override，并指定ktr classpath");
}
/**
* @return 导出文件名。
*/
protected String getDataExportFilename() {
// 默认返回异常，子类如果要使用导出功能，必须覆写此方法，指定导出的文件路径名
throw new RuntimeException("必须override，并指定导出文件名");
}
/**
* @return 数据导入的ktr转换文件类路径。
*/
protected String getDataImportKtrClasspath() {
// 默认返回异常，子类如果要使用导入功能，必须覆写此方法，指定ktr文件类路径
throw new RuntimeException("必须override，并指定ktr classpath");
}
}