cn.jiangzeyin.controller.base.AbstractBaseControl Maven / Gradle / Ivy
package cn.jiangzeyin.controller.base;
import cn.hutool.core.convert.Convert;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.text.UnicodeUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.extra.servlet.ServletUtil;
import cn.hutool.http.HttpUtil;
import cn.jiangzeyin.common.interceptor.BaseCallbackController;
import cn.jiangzeyin.controller.multipart.MultipartFileConfig;
import org.apache.tomcat.util.http.fileupload.servlet.ServletFileUpload;
import org.springframework.http.HttpHeaders;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Objects;
/**
* base
* 公共的获取参数
*
* @author jiangzeyin
* Created by jiangzeyin on 2017/1/12.
*/
public abstract class AbstractBaseControl extends BaseCallbackController {
/**
* 拦截器注入
*/
@Override
public void resetInfo() {
super.resetInfo();
}
/**
* 获取请求的ip 地址
*
* @return ip
*/
protected String getIp() {
return ServletUtil.getClientIP(getRequest());
}
protected String getHeader(String name) {
return getRequest().getHeader(name);
}
/**
* 获取cookie 值
*
* @param name name
* @return value
*/
protected String getCookieValue(String name) {
Cookie cookie = ServletUtil.getCookie(getRequest(), name);
if (cookie == null) {
return "";
}
return cookie.getValue();
}
protected String getParameter(String name) {
return getParameter(name, null);
}
protected String[] getParameters(String name) {
return getRequest().getParameterValues(name);
}
/**
* 获取指定参数名的值
*
* @param name 参数名
* @param def 默认值
* @return str
*/
protected String getParameter(String name, String def) {
String value = getRequest().getParameter(name);
return value == null ? def : value;
}
protected int getParameterInt(String name, int def) {
return Convert.toInt(getParameter(name), def);
}
protected int getParameterInt(String name) {
return getParameterInt(name, 0);
}
protected long getParameterLong(String name, long def) {
String value = getParameter(name);
return Convert.toLong(value, def);
}
protected long getParameterLong(String name) {
return getParameterLong(name, 0L);
}
/**
* 获取来源的url 参数
*
* @return map
*/
protected Map getRefererParameter() {
String referer = getHeader(HttpHeaders.REFERER);
return HttpUtil.decodeParamMap(referer, "UTF-8");
}
/**
* 获取表单数据到实体中
*
* @param tClass class
* @param t
* @return t
*/
protected T getObject(Class tClass) {
return ServletUtil.toBean(getRequest(), tClass, true);
}
/**
* 获取所有请求头
*
* @return map
*/
protected Map getHeaders() {
return RequestUtil.getHeaderMapValues(getRequest());
}
/**
* 所有参数
*
* @return map 值为数组类型
*/
protected Map getParametersMap() {
return getRequest().getParameterMap();
}
// ----------------文件上传
private static final ThreadLocal THREAD_LOCAL_MULTIPART_HTTP_SERVLET_REQUEST = new ThreadLocal<>();
/**
* 释放资源
*/
public static void clearResources() {
THREAD_LOCAL_MULTIPART_HTTP_SERVLET_REQUEST.remove();
}
/**
* 获取文件上传请求对象
*
* @return multipart
*/
protected MultipartHttpServletRequest getMultiRequest() {
HttpServletRequest request = getRequest();
if (request instanceof MultipartHttpServletRequest) {
return (MultipartHttpServletRequest) request;
}
if (ServletFileUpload.isMultipartContent(request)) {
MultipartHttpServletRequest multipartHttpServletRequest = THREAD_LOCAL_MULTIPART_HTTP_SERVLET_REQUEST.get();
if (multipartHttpServletRequest != null) {
return multipartHttpServletRequest;
}
multipartHttpServletRequest = new StandardMultipartHttpServletRequest(request);
THREAD_LOCAL_MULTIPART_HTTP_SERVLET_REQUEST.set(multipartHttpServletRequest);
return multipartHttpServletRequest;
}
throw new IllegalArgumentException("not MultipartHttpServletRequest");
}
protected MultipartFile getFile(String name) {
return getMultiRequest().getFile(name);
}
protected List getFiles(String name) {
return getMultiRequest().getFiles(name);
}
/**
* 接收文件
*
* @param name 字段名称
* @return 保存位置
* @throws IOException IO
*/
protected String upload(String name) throws IOException {
return upload(name, 0);
}
/**
* 接收文件
*
* @param name 字段名称
* @param size 限制文件大小
* @return 保存位置
* @throws IOException IO
*/
protected String upload(String name, long size) throws IOException {
return upload(size, name)[0];
}
/**
* 接收单字段多文件上传,默认不限制大小
*
* @param name 字段名称
* @return 本地路径
* @throws IOException IO
*/
protected String[] uploads(String name) throws IOException {
return uploads(name, 0);
}
/**
* 接收单字段多文件上传
*
* @param name 字段名称
* @param size 文件限制大小
* @return 本地路径
* @throws IOException IO
*/
protected String[] uploads(String name, long size) throws IOException {
List multipartFiles = getFiles(name);
String[] paths = new String[multipartFiles.size()];
for (int i = 0, len = multipartFiles.size(); i < len; i++) {
paths[i] = save(multipartFiles.get(i), size);
}
return paths;
}
/**
* 多文件字段接收,默认不限制文件大小
*
* @param name 文件字段名称
* @return 本地路径数组
* @throws IOException IO
*/
protected String[] upload(String... name) throws IOException {
return upload(0, name);
}
/**
* 多文件字段接收
*
* @param size 限制文件大小
* @param name 文件字段名称
* @return 本地路径数组
* @throws IOException IO
*/
protected String[] upload(long size, String... name) throws IOException {
Objects.requireNonNull(name);
String[] path = new String[name.length];
for (int i = 0, len = path.length; i < len; i++) {
String item = name[i];
MultipartFile multiFile = getFile(item);
if (multiFile == null) {
continue;
}
path[i] = save(multiFile, size);
}
return path;
}
/**
* 保存文件
*
* @param multiFile 文件对象
* @param size 限制大小
* @return 本地路径
* @throws IOException Io
*/
private String save(MultipartFile multiFile, long size) throws IOException {
String fileName = multiFile.getOriginalFilename();
if (StrUtil.isEmpty(fileName)) {
throw new IllegalArgumentException("不能获取到文件名");
}
long fileSize = multiFile.getSize();
if (fileSize <= 0) {
throw new IllegalArgumentException("文件内容为空");
}
// 文件大小
if (size > 0 && fileSize > size) {
throw new IOException("too big:" + fileSize + ">" + size);
}
String localPath = MultipartFileConfig.getFileTempPath();
// 防止中文乱码
fileName = UnicodeUtil.toUnicode(fileName);
String filePath = FileUtil.normalize(String.format("%s/%s_%s", localPath, IdUtil.objectId(), fileName));
FileUtil.writeFromStream(multiFile.getInputStream(), filePath);
return filePath;
}
// ------------------------文件上传结束
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy