75 lines
2.4 KiB
Java
75 lines
2.4 KiB
Java
package com.ruoyi.project.common;
|
|
|
|
import java.io.UnsupportedEncodingException;
|
|
import java.net.URLEncoder;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
import org.springframework.stereotype.Controller;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import com.ruoyi.common.utils.file.FileUtils;
|
|
import com.ruoyi.framework.config.RuoYiConfig;
|
|
|
|
/**
|
|
* 通用请求处理
|
|
*
|
|
* @author ruoyi
|
|
*/
|
|
@Controller
|
|
public class CommonController
|
|
{
|
|
private static final Logger log = LoggerFactory.getLogger(CommonController.class);
|
|
|
|
@RequestMapping("common/download")
|
|
public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request)
|
|
{
|
|
String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1);
|
|
try
|
|
{
|
|
String filePath = RuoYiConfig.getDownloadPath() + fileName;
|
|
|
|
response.setCharacterEncoding("utf-8");
|
|
response.setContentType("multipart/form-data");
|
|
response.setHeader("Content-Disposition", "attachment;fileName=" + setFileDownloadHeader(request, realFileName));
|
|
FileUtils.writeBytes(filePath, response.getOutputStream());
|
|
if (delete)
|
|
{
|
|
FileUtils.deleteFile(filePath);
|
|
}
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
log.error("下载文件失败", e);
|
|
}
|
|
}
|
|
|
|
public String setFileDownloadHeader(HttpServletRequest request, String fileName) throws UnsupportedEncodingException
|
|
{
|
|
final String agent = request.getHeader("USER-AGENT");
|
|
String filename = fileName;
|
|
if (agent.contains("MSIE"))
|
|
{
|
|
// IE浏览器
|
|
filename = URLEncoder.encode(filename, "utf-8");
|
|
filename = filename.replace("+", " ");
|
|
}
|
|
else if (agent.contains("Firefox"))
|
|
{
|
|
// 火狐浏览器
|
|
filename = new String(fileName.getBytes(), "ISO8859-1");
|
|
}
|
|
else if (agent.contains("Chrome"))
|
|
{
|
|
// google浏览器
|
|
filename = URLEncoder.encode(filename, "utf-8");
|
|
}
|
|
else
|
|
{
|
|
// 其它浏览器
|
|
filename = URLEncoder.encode(filename, "utf-8");
|
|
}
|
|
return filename;
|
|
}
|
|
}
|