|
|
package com.platform.controller;
|
|
|
|
|
|
import java.io.File;
|
|
|
import java.io.IOException;
|
|
|
import java.io.InputStream;
|
|
|
import java.io.PrintWriter;
|
|
|
import java.io.RandomAccessFile;
|
|
|
import java.io.UnsupportedEncodingException;
|
|
|
import java.util.ArrayList;
|
|
|
import java.util.Arrays;
|
|
|
import java.util.Collection;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.Iterator;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
|
import java.util.Set;
|
|
|
import java.util.regex.Matcher;
|
|
|
import java.util.regex.Pattern;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
import javax.servlet.ServletException;
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
|
|
|
import net.sf.json.JSONArray;
|
|
|
|
|
|
import org.apache.commons.io.FileUtils;
|
|
|
import org.apache.commons.lang.StringUtils;
|
|
|
import org.apache.log4j.Logger;
|
|
|
import org.springframework.http.HttpHeaders;
|
|
|
import org.springframework.http.HttpStatus;
|
|
|
import org.springframework.http.MediaType;
|
|
|
import org.springframework.http.ResponseEntity;
|
|
|
import org.springframework.stereotype.Controller;
|
|
|
import org.springframework.ui.ModelMap;
|
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
|
import org.springframework.web.bind.annotation.RequestMethod;
|
|
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
|
|
import org.springframework.web.multipart.commons.CommonsMultipartResolver;
|
|
|
|
|
|
import com.base.BaseController;
|
|
|
import com.platform.entities.PagerOptions;
|
|
|
import com.platform.entities.PreDataInfo;
|
|
|
import com.platform.entities.ResumableInfo;
|
|
|
import com.platform.entities.ResumableInfoStorage;
|
|
|
import com.platform.service.IPreDataInfoService;
|
|
|
import com.platform.service.IScriptMakeService;
|
|
|
import com.platform.utils.Configs;
|
|
|
import com.platform.utils.HttpUtils;
|
|
|
import com.platform.utils.ThreadRemoveFile;
|
|
|
import com.platform.utils.UtilsHelper;
|
|
|
|
|
|
/** 信息系统--excel操作
|
|
|
* @author chen
|
|
|
*
|
|
|
*/
|
|
|
@Controller
|
|
|
@RequestMapping("/fileOperation")
|
|
|
public class ExcelController extends BaseController{
|
|
|
|
|
|
public static Logger log = Configs.DAILY_ROLLING_LOGGER.getLogger(ExcelController.class);
|
|
|
|
|
|
@Resource(name = "preDataInfoService")
|
|
|
private IPreDataInfoService preDataInfoService;
|
|
|
|
|
|
|
|
|
// 文件上传处理函数
|
|
|
@RequestMapping(value = "/file/upload", method = RequestMethod.POST)
|
|
|
public void upload(HttpServletRequest request, HttpServletResponse response)
|
|
|
throws ServletException, IOException {
|
|
|
|
|
|
int resumableChunkNumber = HttpUtils.toInt(
|
|
|
request.getParameter("resumableChunkNumber"), -1);
|
|
|
ResumableInfo info = getResumableInfo(request);
|
|
|
|
|
|
String requestMethod = request.getMethod();
|
|
|
// 处理GET请求
|
|
|
if ("GET".equals(requestMethod)) {
|
|
|
if (info.uploadedChunks
|
|
|
.contains(new ResumableInfo.ResumableChunkNumber(
|
|
|
resumableChunkNumber))) {
|
|
|
response.getWriter().print("Uploaded."); // This Chunk has been
|
|
|
// Uploaded.
|
|
|
} else {
|
|
|
response.setStatus(HttpServletResponse.SC_NOT_FOUND);
|
|
|
}
|
|
|
} else if ("POST".equals(requestMethod)) { // 处理POST请求
|
|
|
RandomAccessFile raf = new RandomAccessFile(info.resumableFilePath,
|
|
|
"rw");
|
|
|
// Seek to position
|
|
|
raf.seek((resumableChunkNumber - 1)
|
|
|
* (long) info.resumableChunkSize);
|
|
|
|
|
|
// Save to file
|
|
|
InputStream is = request.getInputStream();
|
|
|
long readed = 0;
|
|
|
long content_length = request.getContentLength();
|
|
|
byte[] bytes = new byte[1024 * 100];
|
|
|
while (readed < content_length) {
|
|
|
int r = is.read(bytes);
|
|
|
if (r < 0) {
|
|
|
break;
|
|
|
}
|
|
|
raf.write(bytes, 0, r);
|
|
|
readed += r;
|
|
|
}
|
|
|
raf.close();
|
|
|
|
|
|
// Mark as uploaded.
|
|
|
info.uploadedChunks.add(new ResumableInfo.ResumableChunkNumber(
|
|
|
resumableChunkNumber));
|
|
|
if (info.checkIfUploadFinished()) { // Check if all chunks uploaded,
|
|
|
// and change filename
|
|
|
log.info(info.resumableFilename);
|
|
|
ResumableInfoStorage.getInstance().remove(info);
|
|
|
response.getWriter().print("All finished.");
|
|
|
//开始导入excel
|
|
|
|
|
|
} else {
|
|
|
response.getWriter().print("Upload");
|
|
|
}
|
|
|
} else { // 不处理非GET/POST请求
|
|
|
throw new IllegalStateException("只接受 POST或GET请求");
|
|
|
}
|
|
|
log.info("----");
|
|
|
}
|
|
|
|
|
|
// 文件下载处理函数
|
|
|
@RequestMapping(value = "/file/download")
|
|
|
public ResponseEntity<byte[]> download(HttpServletRequest request,
|
|
|
HttpServletResponse response) throws Exception {
|
|
|
//先导出
|
|
|
preDataInfoService.exportExcel(Configs.FILE_DOWNLOAD_PATH);
|
|
|
|
|
|
File file = new File(Configs.FILE_DOWNLOAD_PATH);
|
|
|
System.out.println(Configs.FILE_DOWNLOAD_PATH);
|
|
|
if (file.exists()) {
|
|
|
HttpHeaders headers = new HttpHeaders();
|
|
|
String fileName = new String(file.getName().getBytes("UTF-8"),
|
|
|
"iso-8859-1");// 为了解决中文名称乱码问题
|
|
|
headers.setContentDispositionFormData("attachment", fileName);
|
|
|
headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
|
|
|
return new ResponseEntity<byte[]>(
|
|
|
FileUtils.readFileToByteArray(file), headers,
|
|
|
HttpStatus.CREATED);
|
|
|
} else {
|
|
|
response.setStatus(500);
|
|
|
return null;
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// 文件上传处理函数
|
|
|
private ResumableInfo getResumableInfo(HttpServletRequest request)
|
|
|
throws ServletException, IOException {
|
|
|
String base_dir = Configs.FILE_UPLOAD_PATH;
|
|
|
|
|
|
int resumableChunkSize = HttpUtils.toInt(
|
|
|
request.getParameter("resumableChunkSize"), -1);
|
|
|
long resumableTotalSize = HttpUtils.toLong(
|
|
|
request.getParameter("resumableTotalSize"), -1);
|
|
|
String resumableIdentifier = request
|
|
|
.getParameter("resumableIdentifier");
|
|
|
String fileName = new String(request.getParameter("resumableFilename").getBytes("iso-8859-1"),
|
|
|
"UTF-8");// 为了解决中文名称乱码问题
|
|
|
String resumableFilename = fileName /*request.getParameter("resumableFilename")*/;
|
|
|
String resumableRelativePath = request
|
|
|
.getParameter("resumableRelativePath");
|
|
|
// Here we add a ".temp" to every upload file to indicate NON-FINISHED
|
|
|
new File(base_dir).mkdir();
|
|
|
String resumableFilePath = new File(base_dir, resumableFilename)
|
|
|
.getAbsolutePath() + ".temp";
|
|
|
|
|
|
ResumableInfoStorage storage = ResumableInfoStorage.getInstance();
|
|
|
|
|
|
ResumableInfo info = storage.get(resumableChunkSize,
|
|
|
resumableTotalSize, resumableIdentifier, resumableFilename,
|
|
|
resumableRelativePath, resumableFilePath);
|
|
|
if (!info.vaild()) {
|
|
|
storage.remove(info);
|
|
|
throw new ServletException("Invalid request params.");
|
|
|
}
|
|
|
return info;
|
|
|
}
|
|
|
|
|
|
@ResponseBody
|
|
|
@RequestMapping("/findByParam")
|
|
|
public ModelMap findByParam(HttpServletRequest res, HttpServletResponse req) throws Exception {
|
|
|
ModelMap modelMap = new ModelMap();
|
|
|
|
|
|
res.setCharacterEncoding("UTF-8");
|
|
|
Map<String, String[]> paramMap = res.getParameterMap();
|
|
|
Set<String> keySet = paramMap.keySet();
|
|
|
Map<String, String> params = new HashMap<String, String>();
|
|
|
StringBuffer sb = new StringBuffer().append("当前的请求参数:{");
|
|
|
for (String str : keySet) {
|
|
|
String value = paramMap.get(str)[0];
|
|
|
if (StringUtils.isNotEmpty(value)) {
|
|
|
params.put(str, value);
|
|
|
sb.append(str).append(":").append(value).append(",");
|
|
|
} else {
|
|
|
sb.append(str).append(":").append("null").append(",");
|
|
|
}
|
|
|
}
|
|
|
Configs.CONSOLE_LOGGER.info(sb.deleteCharAt(sb.length() - 1)
|
|
|
.append("}").toString());
|
|
|
PagerOptions pagerOptions = (PagerOptions) UtilsHelper
|
|
|
.newObjAndSetAttrsByClass(PagerOptions.class, params);
|
|
|
|
|
|
List<PreDataInfo> result = preDataInfoService.findByParam(pagerOptions);
|
|
|
modelMap.addAttribute("data", result);
|
|
|
modelMap.addAttribute("length", result.size());
|
|
|
return modelMap;
|
|
|
}
|
|
|
|
|
|
@ResponseBody
|
|
|
@RequestMapping("/findAll")
|
|
|
public ModelMap findAll() throws Exception {
|
|
|
ModelMap modelMap = new ModelMap();
|
|
|
List<PreDataInfo> result = preDataInfoService.findAll();
|
|
|
modelMap.addAttribute("data", result);
|
|
|
modelMap.addAttribute("length", result.size());
|
|
|
return modelMap;
|
|
|
}
|
|
|
|
|
|
@ResponseBody
|
|
|
@RequestMapping(produces = "application/json", value = "/importExcel", method = RequestMethod.POST)
|
|
|
public ModelMap importExcel(HttpServletRequest request,
|
|
|
HttpServletResponse response, @RequestBody String fileNameList) throws Exception {
|
|
|
log.info("---importExcel---" + fileNameList);
|
|
|
JSONArray jsonArray = JSONArray.fromObject(fileNameList);
|
|
|
List<String> jsonlist = (List<String>) JSONArray.toCollection(jsonArray, String.class);
|
|
|
|
|
|
ModelMap modelMap = new ModelMap();
|
|
|
if (null == jsonlist && jsonlist.size() > 0) {
|
|
|
System.err.println(request.getParameter("fileNameList"));
|
|
|
modelMap.addAttribute("returncode","3002");
|
|
|
return modelMap;
|
|
|
}
|
|
|
log.info(jsonlist.size());
|
|
|
List<String> listPath = new ArrayList<String>();
|
|
|
// 末尾 含有 /(linux)
|
|
|
Pattern pattern2 = Pattern.compile("\\/$");
|
|
|
Matcher matcher2 = pattern2.matcher(Configs.FILE_UPLOAD_PATH);
|
|
|
// 加上 最后 的 /
|
|
|
String dirPath = Configs.FILE_UPLOAD_PATH;
|
|
|
if (!matcher2.find()) {
|
|
|
dirPath = dirPath + File.separator;
|
|
|
}
|
|
|
for (String p : jsonlist) {
|
|
|
listPath.add(dirPath + p);
|
|
|
}
|
|
|
Map<String, List> result = new HashMap<String, List>();
|
|
|
if (listPath.size() > 0) {
|
|
|
//导入
|
|
|
result = preDataInfoService.importExcel(listPath);
|
|
|
//失败
|
|
|
if (result.containsKey("fileUnExist") || result.containsKey("areaUnImport")) {
|
|
|
modelMap.addAttribute("data", result);
|
|
|
modelMap.addAttribute("code","3001");
|
|
|
response.setStatus(500);
|
|
|
}
|
|
|
//成功
|
|
|
else {
|
|
|
List<PreDataInfo> datalist = preDataInfoService.findAll();
|
|
|
modelMap.addAttribute("data", datalist);
|
|
|
modelMap.addAttribute("length", datalist.size());
|
|
|
modelMap.addAttribute("code","1001");
|
|
|
}
|
|
|
//删除文件
|
|
|
List<String> failedDelete = new ArrayList<String>();
|
|
|
for (String filePath : listPath) {
|
|
|
//读取完后删除源文件（不管是否成功导入）
|
|
|
File tmpf = new File(filePath + ".temp");
|
|
|
File f = new File(filePath);
|
|
|
if(!tmpf.delete()){
|
|
|
failedDelete.add(tmpf.getAbsolutePath());
|
|
|
}
|
|
|
if(!f.delete()){
|
|
|
failedDelete.add(f.getAbsolutePath());
|
|
|
}
|
|
|
}
|
|
|
//删除失败时--启用线程去删除
|
|
|
new ThreadRemoveFile(failedDelete).start();
|
|
|
}
|
|
|
return modelMap;
|
|
|
}
|
|
|
|
|
|
public void uploadSpring(HttpServletRequest request, HttpServletResponse respone){
|
|
|
System.out.println(request.getAttribute("file"));
|
|
|
System.out.println(request.getParameter("file"));
|
|
|
//解析器解析request的上下文
|
|
|
CommonsMultipartResolver multipartResolver =
|
|
|
new CommonsMultipartResolver(request.getSession().getServletContext());
|
|
|
//先判断request中是否包涵multipart类型的数据，
|
|
|
if(multipartResolver.isMultipart(request)){
|
|
|
//再将request中的数据转化成multipart类型的数据
|
|
|
MultipartHttpServletRequest multiRequest = (MultipartHttpServletRequest) request;
|
|
|
Iterator iter = multiRequest.getFileNames();
|
|
|
while(iter.hasNext()){
|
|
|
//这里的name为fileItem的alias属性值，相当于form表单中name
|
|
|
String name=(String)iter.next();
|
|
|
System.out.println("name:"+name);
|
|
|
//根据name值拿取文件
|
|
|
MultipartFile file = multiRequest.getFile(name);
|
|
|
if(file != null){
|
|
|
String fileName = file.getOriginalFilename();
|
|
|
String path = "D:/test/" + fileName;
|
|
|
File localFile = new File(path);
|
|
|
if(!localFile.getParentFile().exists()) {
|
|
|
//如果目标文件所在的目录不存在，则创建父目录
|
|
|
localFile.getParentFile().mkdirs();
|
|
|
System.out.println("parent:"+localFile.getParentFile().getPath());
|
|
|
}
|
|
|
//写文件到本地
|
|
|
try {
|
|
|
file.transferTo(localFile);
|
|
|
} catch (IOException e) {
|
|
|
// TODO Auto-generated catch block
|
|
|
e.printStackTrace();
|
|
|
}
|
|
|
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
