package com.vci.ubcs.starter.web.util;
|
|
import org.apache.commons.io.IOUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springblade.core.tool.utils.StringUtil;
|
|
import javax.servlet.http.Cookie;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.net.URLEncoder;
|
import java.util.Map;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
public class ControllerUtil {
|
private static Logger logger = LoggerFactory.getLogger(ControllerUtil.class);
|
public static final Map<String, String> tempFileForDownloadMap = new ConcurrentHashMap();
|
|
public ControllerUtil() {
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, String fileName) throws FileNotFoundException, IOException {
|
writeFileToResponse(response, fileName, true);
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, String fileName, boolean deleteFile) throws FileNotFoundException, IOException {
|
writeFileToResponse(response, (String)fileName, (String)null, deleteFile);
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, String fileName, String showName, boolean deleteFile) throws FileNotFoundException, IOException {
|
File file = new File(fileName);
|
if (!file.exists()) {
|
throw new FileNotFoundException(fileName);
|
} else {
|
writeFileToResponse(response, file, showName, deleteFile);
|
}
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, File file) throws FileNotFoundException, IOException {
|
writeFileToResponse(response, file, true);
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, File file, boolean deleteFile) throws FileNotFoundException, IOException {
|
writeFileToResponse(response, (File)file, (String)null, deleteFile);
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, File file, String showName, boolean deleteFile) throws FileNotFoundException, IOException {
|
writeFileToResponse(response, file, showName, deleteFile, (String)null);
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, File file, String showName, boolean deleteFile, String contentType) throws FileNotFoundException, IOException {
|
VciBaseUtil.alertNotNull(new Object[]{file, "文件对象"});
|
if (!file.exists()) {
|
throw new FileNotFoundException(file.getName());
|
} else {
|
if (StringUtil.isBlank(showName)) {
|
showName = file.getName();
|
}
|
|
FileInputStream in = null;
|
boolean var11 = false;
|
|
File parentFile;
|
label96: {
|
try {
|
var11 = true;
|
in = new FileInputStream(file);
|
writeFileToResponse(response, in, showName, contentType);
|
var11 = false;
|
break label96;
|
} catch (IOException var12) {
|
if (logger.isErrorEnabled()) {
|
logger.error("写入文件到响应流出错", var12);
|
var11 = false;
|
} else {
|
var11 = false;
|
}
|
} finally {
|
if (var11) {
|
IOUtils.closeQuietly(in);
|
if (deleteFile) {
|
File parentFile1 = file.getParentFile();
|
file.delete();
|
parentFile1.delete();
|
}
|
|
}
|
}
|
|
IOUtils.closeQuietly(in);
|
if (deleteFile) {
|
parentFile = file.getParentFile();
|
file.delete();
|
parentFile.delete();
|
}
|
|
return;
|
}
|
|
IOUtils.closeQuietly(in);
|
if (deleteFile) {
|
parentFile = file.getParentFile();
|
file.delete();
|
parentFile.delete();
|
}
|
|
}
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, InputStream ins, String showName) throws IOException {
|
writeFileToResponse(response, ins, showName, (String)null);
|
}
|
|
public static void writeFileToResponse(HttpServletResponse response, InputStream ins, String showName, String contentType) throws IOException {
|
if (StringUtil.isBlank(contentType)) {
|
contentType = "application/force-download";
|
}
|
|
if (StringUtil.isBlank(showName)) {
|
showName = "下载文件";
|
}
|
|
response.setContentType(contentType);
|
|
try {
|
String fileName = URLEncoder.encode(showName, "UTF8");
|
response.addHeader("Content-Disposition", "attachment; filename=" + fileName + ";filename*=utf-8''" + fileName);
|
} catch (Exception var12) {
|
if (logger.isErrorEnabled()) {
|
logger.error("设置文件的名称到响应流的时候出错", var12);
|
}
|
}
|
|
response.setCharacterEncoding("UTF-8");
|
Cookie cookie = new Cookie("fileDownload", "true");
|
cookie.setPath("/");
|
response.addCookie(cookie);
|
|
try {
|
IOUtils.copy(ins, response.getOutputStream());
|
} catch (IOException var10) {
|
if (logger.isErrorEnabled()) {
|
logger.error("写入文件到响应流出错", var10);
|
}
|
|
throw var10;
|
} finally {
|
IOUtils.closeQuietly(ins);
|
}
|
|
}
|
|
public static void writeDataToResponse(HttpServletResponse response, byte[] data, String contentType) throws IOException {
|
if (StringUtil.isBlank(contentType)) {
|
contentType = "application/force-download";
|
}
|
|
response.setContentType(contentType);
|
response.setCharacterEncoding("UTF-8");
|
Cookie cookie = new Cookie("fileDownload", "true");
|
cookie.setPath("/");
|
response.addCookie(cookie);
|
|
try {
|
response.getOutputStream().write(data);
|
} catch (IOException var5) {
|
if (logger.isErrorEnabled()) {
|
logger.error("写入文件到响应流出错", var5);
|
}
|
|
throw var5;
|
}
|
}
|
|
public static void writeClasspathFileToResponse(HttpServletResponse response, String classPathFileName) throws FileNotFoundException, IOException {
|
writeClasspathFileToResponse(response, classPathFileName, (String)null);
|
}
|
|
public static void writeClasspathFileToResponse(HttpServletResponse response, String classPathFileName, String showName) throws FileNotFoundException, IOException {
|
InputStream in = Thread.currentThread().getContextClassLoader().getResourceAsStream(classPathFileName);
|
if (StringUtil.isBlank(showName)) {
|
showName = classPathFileName.contains("/") ? classPathFileName.substring(classPathFileName.lastIndexOf("/") + 1) : classPathFileName;
|
}
|
|
try {
|
writeFileToResponse(response, in, showName);
|
} catch (IOException var8) {
|
throw var8;
|
} finally {
|
IOUtils.closeQuietly(in);
|
}
|
|
}
|
|
public static String getClientInfo(HttpServletRequest request) {
|
String ip = request.getHeader("X-Forwarded-For");
|
if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
|
ip = request.getHeader("Proxy-Client-IP");
|
}
|
|
if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
|
ip = request.getHeader("WL-Proxy-Client-IP");
|
}
|
|
if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
|
ip = request.getHeader("HTTP_CLIENT_IP");
|
}
|
|
if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
|
ip = request.getHeader("HTTP_X_FORWARDED_FOR");
|
}
|
|
if (ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip)) {
|
ip = request.getRemoteAddr();
|
}
|
|
if (ip == null || ip.length() == 0 || ip.indexOf("0:0:0:0:0:0:0:1") > -1) {
|
ip = "127.0.0.1";
|
}
|
|
return ip;
|
}
|
|
public static void setQueryTotal(HttpServletRequest request, boolean isQueryTotal) {
|
WebThreadLocalUtil.getNeedQueryTotalInThread().set(isQueryTotal ? "true" : "false");
|
}
|
|
public static String putErrorFile(String errorFile) {
|
String uuid = VciBaseUtil.getPk();
|
tempFileForDownloadMap.put(uuid, errorFile);
|
return uuid;
|
}
|
|
public static void downloadErrorFile(HttpServletResponse response, String uuid) throws IOException {
|
String errorFile = (String)tempFileForDownloadMap.getOrDefault(uuid, "");
|
|
try {
|
if (StringUtil.isNotBlank(errorFile)) {
|
writeFileToResponse(response, new File(errorFile), (String)null, true, "application/msexcel");
|
}
|
} finally {
|
tempFileForDownloadMap.remove(uuid);
|
}
|
|
}
|
}
|
