package com.highdatas.srs.web;
|
|
import com.alibaba.fastjson.JSONObject;
|
import com.highdatas.srs.entity.ProjectDeal;
|
import com.highdatas.srs.entity.Scheme;
|
import com.highdatas.srs.entity.SchemeDetail;
|
import com.highdatas.srs.entity.Topic;
|
import com.highdatas.srs.pojo.CodeMsg;
|
import com.highdatas.srs.pojo.Result;
|
import com.highdatas.srs.service.ISchemeDetailService;
|
import com.highdatas.srs.service.ISchemeService;
|
import com.highdatas.srs.service.ITopicService;
|
import com.highdatas.srs.util.DbUtils;
|
import com.highdatas.srs.util.FileUtils;
|
import lombok.extern.java.Log;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URLDecoder;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
/**
|
* @author kimi
|
* @description
|
* @date 2020-01-17 12:05
|
*/
|
|
@RestController
|
@Log
|
@RequestMapping("file")
|
public class FileController {
|
@Autowired
|
ISchemeDetailService schemeDetailService;
|
@Autowired
|
ISchemeService schemeService;
|
@Autowired
|
ITopicService topicService;
|
@Value("${base.file.path}")
|
String basePath;
|
@Value("${base.topic.path}")
|
String topicPath;
|
|
@RequestMapping(value = "child", method = RequestMethod.GET)
|
public Result children(@RequestParam String path) throws UnsupportedEncodingException {
|
path = URLDecoder.decode(path, "UTF-8");
|
ArrayList<JSONObject> childrenFiles = FileUtils.getChildrenFiles(path);
|
return Result.success(childrenFiles);
|
}
|
@RequestMapping(value = "downloadPath", method = RequestMethod.GET)
|
public Result downloadPath(@RequestParam String path, HttpServletResponse response) throws UnsupportedEncodingException {
|
path = URLDecoder.decode(path, "UTF-8");
|
File file = new File(path);
|
if(file.exists()){ //判断文件父目录是否存在
|
response.setContentType("application/vnd.ms-excel;charset=UTF-8");
|
response.setCharacterEncoding("UTF-8");
|
// response.setContentType("application/force-download");
|
response.setHeader("Content-Disposition", "attachment;fileName=" + java.net.URLEncoder.encode(file.getName(),"UTF-8"));
|
byte[] buffer = new byte[1024];
|
FileInputStream fis = null; //文件输入流
|
BufferedInputStream bis = null;
|
|
OutputStream os = null; //输出流
|
try {
|
os = response.getOutputStream();
|
fis = new FileInputStream(file);
|
bis = new BufferedInputStream(fis);
|
int i = bis.read(buffer);
|
while(i != -1){
|
os.write(buffer);
|
i = bis.read(buffer);
|
}
|
|
} catch (Exception e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
log.info("----------file download---" + file.getName());
|
try {
|
bis.close();
|
fis.close();
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}}
|
return null;
|
}
|
@RequestMapping("{detailId}/download/{fileName}")
|
public Result downLoad(@PathVariable String detailId, @PathVariable String fileName, HttpServletResponse response) throws UnsupportedEncodingException {
|
SchemeDetail detail = schemeDetailService.selectById(detailId);
|
if (detail == null) {
|
return Result.error(CodeMsg.SELECT_ERROR);
|
}
|
|
String parentId = detail.getParentId();
|
Scheme scheme = schemeService.selectById(parentId);
|
if (scheme == null) {
|
return Result.error(CodeMsg.SELECT_ERROR);
|
}
|
|
String path = FileUtils.createFile(basePath, scheme.getName(), detail.getType().covert());
|
File file = new File(path);
|
if(file.exists()){ //判断文件父目录是否存在
|
response.setContentType("application/vnd.ms-excel;charset=UTF-8");
|
response.setCharacterEncoding("UTF-8");
|
// response.setContentType("application/force-download");
|
response.setHeader("Content-Disposition", "attachment;fileName=" + java.net.URLEncoder.encode(fileName,"UTF-8"));
|
byte[] buffer = new byte[1024];
|
FileInputStream fis = null; //文件输入流
|
BufferedInputStream bis = null;
|
|
OutputStream os = null; //输出流
|
try {
|
os = response.getOutputStream();
|
fis = new FileInputStream(file);
|
bis = new BufferedInputStream(fis);
|
int i = bis.read(buffer);
|
while(i != -1){
|
os.write(buffer);
|
i = bis.read(buffer);
|
}
|
|
} catch (Exception e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
log.info("----------file download---" + fileName);
|
try {
|
bis.close();
|
fis.close();
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
}
|
return null;
|
}
|
@RequestMapping("/download/{topicId}")
|
public Result downLoadTopic(@PathVariable String topicId, HttpServletResponse response) throws UnsupportedEncodingException {
|
Topic topic = topicService.selectById(topicId);
|
String path = topic.getAttachment();
|
File file = new File(path);
|
if(file.exists()){ //判断文件父目录是否存在
|
response.setContentType("application/vnd.ms-excel;charset=UTF-8");
|
response.setCharacterEncoding("UTF-8");
|
// response.setContentType("application/force-download");
|
response.setHeader("Content-Disposition", "attachment;fileName=" + java.net.URLEncoder.encode(file.getName(),"UTF-8"));
|
byte[] buffer = new byte[1024];
|
FileInputStream fis = null; //文件输入流
|
BufferedInputStream bis = null;
|
|
OutputStream os = null; //输出流
|
try {
|
os = response.getOutputStream();
|
fis = new FileInputStream(file);
|
bis = new BufferedInputStream(fis);
|
int i = bis.read(buffer);
|
while(i != -1){
|
os.write(buffer);
|
i = bis.read(buffer);
|
}
|
|
} catch (Exception e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
log.info("----------file download---" + file.getName());
|
try {
|
bis.close();
|
fis.close();
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
}
|
return null;
|
}
|
|
@RequestMapping("/downloadPath/{path}")
|
public Result downLoadPath(@PathVariable String path, HttpServletResponse response) throws UnsupportedEncodingException {
|
path = URLDecoder.decode(path, "UTF-8");
|
File file = new File(path);
|
if(file.exists()){ //判断文件父目录是否存在
|
response.setContentType("application/vnd.ms-excel;charset=UTF-8");
|
response.setCharacterEncoding("UTF-8");
|
// response.setContentType("application/force-download");
|
response.setHeader("Content-Disposition", "attachment;fileName=" + java.net.URLEncoder.encode(file.getName(),"UTF-8"));
|
byte[] buffer = new byte[1024];
|
FileInputStream fis = null; //文件输入流
|
BufferedInputStream bis = null;
|
|
OutputStream os = null; //输出流
|
try {
|
os = response.getOutputStream();
|
fis = new FileInputStream(file);
|
bis = new BufferedInputStream(fis);
|
int i = bis.read(buffer);
|
while(i != -1){
|
os.write(buffer);
|
i = bis.read(buffer);
|
}
|
|
} catch (Exception e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
log.info("----------file download---" + file.getName());
|
try {
|
bis.close();
|
fis.close();
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
}
|
}
|
return null;
|
}
|
|
/**
|
* 实现文件上传
|
* */
|
@RequestMapping("fileUpload/{detailId}")
|
@ResponseBody
|
public Result fileUpload(@PathVariable String detailId, @RequestParam("file") MultipartFile file, HttpServletRequest request){
|
String desp = request.getParameter("desp");
|
SchemeDetail detail = schemeDetailService.selectById(detailId);
|
if (detail == null) {
|
return Result.error(CodeMsg.SELECT_ERROR);
|
}
|
if (!detail.getMustAttach()) {
|
ProjectDeal projectDeal = new ProjectDeal();
|
projectDeal.setCreateTime(new Date()).setSchemeId(detail.getParentId())
|
.setSchemeDetailId(detailId).setDesp(desp)
|
.setId(DbUtils.getUUID()).insert();
|
|
detail.setFinish(true);
|
detail.setUpdateTime(new Date());
|
detail.updateById();
|
return Result.success(CodeMsg.SUCCESS);
|
}
|
if(file.isEmpty()){
|
return Result.error(CodeMsg.ERROR_PARAMS_NOT_MATHED);
|
}
|
|
String parentId = detail.getParentId();
|
Scheme scheme = schemeService.selectById(parentId);
|
if (scheme == null) {
|
return Result.error(CodeMsg.SELECT_ERROR);
|
}
|
|
String fileName = file.getOriginalFilename();
|
int size = (int) file.getSize();
|
log.info(fileName + "-->" + size);
|
|
String path = FileUtils.createFile(basePath, scheme.getName(), detail.getType().covert());
|
File dest = new File(path + "/" + fileName);
|
if(!dest.getParentFile().exists()){ //判断文件父目录是否存在
|
dest.getParentFile().mkdirs();
|
}
|
dest.deleteOnExit();
|
|
try {
|
dest.createNewFile();
|
file.transferTo(dest); //保存文件
|
|
ProjectDeal projectDeal = new ProjectDeal();
|
projectDeal.setCreateTime(new Date()).setSchemeId(detail.getParentId())
|
.setAttachment(path + "/" + fileName)
|
.setSchemeDetailId(detailId).setDesp(desp)
|
.setId(DbUtils.getUUID()).insert();
|
|
ProjectDeal projectDeal2 = new ProjectDeal();
|
projectDeal2.setCreateTime(new Date()).setSchemeId(detail.getParentId())
|
.setSchemeDetailId(detailId).setDesp("处理任务:" + detail.getType().covert())
|
.setId(DbUtils.getUUID()).insert();
|
detail.setFinish(true);
|
detail.setAttachment(path + "/" + fileName);
|
detail.setUpdateTime(new Date());
|
detail.updateById();
|
return Result.success(CodeMsg.SUCCESS);
|
} catch (IllegalStateException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
return Result.error(CodeMsg.OPERATR_ERROR);
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
return Result.error(CodeMsg.OPERATR_ERROR);
|
}
|
}
|
|
@RequestMapping("topicUpload")
|
@ResponseBody
|
public Result topicUpload(@RequestParam("file") MultipartFile file, HttpServletRequest request){
|
String id = request.getParameter("id");
|
String title = request.getParameter("title");
|
String desp = request.getParameter("desp");
|
String userId = request.getParameter("userId");
|
Topic topic;
|
if (StringUtils.isEmpty(id)) {
|
topic = new Topic().setId(DbUtils.getUUID()).setTitle(title).setDesp(desp).setUserId(userId).setCreateTime(new Date());
|
} else {
|
topic = topicService.selectById(id).setTitle(title).setDesp(desp);
|
}
|
|
String fileName = file.getOriginalFilename();
|
int size = (int) file.getSize();
|
log.info(fileName + "-->" + size);
|
String path = topicPath + "/"+ topic.getId() + fileName;
|
File dest = new File(path);
|
if(!dest.getParentFile().exists()){ //判断文件父目录是否存在
|
dest.getParentFile().mkdirs();
|
}
|
dest.deleteOnExit();
|
|
try {
|
dest.createNewFile();
|
file.transferTo(dest); //保存文件
|
topic.setAttachment(path);
|
topic.insertOrUpdate();
|
return Result.success(CodeMsg.SUCCESS);
|
} catch (IllegalStateException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
return Result.error(CodeMsg.OPERATR_ERROR);
|
} catch (IOException e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
return Result.error(CodeMsg.OPERATR_ERROR);
|
}
|
}
|
|
/**
|
* 实现多文件上传
|
* */
|
@RequestMapping(value="multifileUpload",method= RequestMethod.POST)
|
public @ResponseBody String multifileUpload(HttpServletRequest request){
|
|
List<MultipartFile> files = ((MultipartHttpServletRequest)request).getFiles("fileName");
|
|
if(files.isEmpty()){
|
return "false";
|
}
|
|
String path = "F:/test" ;
|
|
for(MultipartFile file:files){
|
String fileName = file.getOriginalFilename();
|
int size = (int) file.getSize();
|
log.info(fileName + "-->" + size);
|
|
if(file.isEmpty()){
|
return "false";
|
}else{
|
File dest = new File(path + "/" + fileName);
|
if(!dest.getParentFile().exists()){ //判断文件父目录是否存在
|
dest.getParentFile().mkdir();
|
}
|
try {
|
file.transferTo(dest);
|
}catch (Exception e) {
|
// TODO Auto-generated catch block
|
e.printStackTrace();
|
return "false";
|
}
|
}
|
}
|
return "true";
|
}
|
|
}
|
