package com.doumee.api.common; import com.alibaba.fastjson.JSONObject; import com.doumee.api.BaseController; import com.doumee.biz.system.SystemDictDataBiz; import com.doumee.core.annotation.trace.Trace; import com.doumee.core.constants.Constants; import com.doumee.core.utils.DateUtil; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.PostMapping; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import org.springframework.web.multipart.commons.CommonsMultipartResolver; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.io.File; import java.io.IOException; import java.io.PrintWriter; import java.util.HashMap; import java.util.Iterator; import java.util.Map; import java.util.UUID; /** * @author Eva.Caesar Liu * @date 2023/02/14 11:14 */ @Api(tags = "公共上传接口") @Trace(exclude = true) @RestController @RequestMapping("/web/public") @Slf4j public class PublicCloudController extends BaseController { @Autowired private SystemDictDataBiz systemDictDataBiz; @ApiOperation(value = "上传", notes = "上传", httpMethod = "POST", position = 6) @ApiImplicitParams({ @ApiImplicitParam(name = "folder", value = "文件夹", required = true, paramType = "query", dataType = "String", dataTypeClass = String.class), @ApiImplicitParam(name = "file", value = "文件夹", required = true, paramType = "query", dataType = "String", dataTypeClass = String.class), }) @PostMapping(value = "/uploadLocal", headers = "content-type=multipart/form-data") public void uploadLocal(HttpServletRequest request, HttpServletResponse response) throws Exception { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; uploadFileLocal(multipartRequest, response, systemDictDataBiz.queryByCode(Constants.SYSTEM,Constants.IMG_DIR).getCode() , systemDictDataBiz.queryByCode(Constants.SYSTEM,Constants.FILE_DIR).getCode()); } public void uploadFileLocal(HttpServletRequest request, HttpServletResponse response, String rootPath,String dir) throws Exception { response.setCharacterEncoding("UTF-8"); response.setContentType("text/html;charset=UTF-8"); CommonsMultipartResolver multipartResovler = new CommonsMultipartResolver(); Map context = new HashMap<>(); if (multipartResovler.isMultipart(request)) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; Iterator it = multipartRequest.getFileNames(); while (it.hasNext()) { MultipartFile file = multipartRequest.getFile((String) it.next()); if (file != null) { if (file.getSize() > 200 * 1024 * 1024L) { context.put("code", 4000); context.put("message", "上传文件过大"); return; } System.out.println(file.getOriginalFilename()); if (file.getOriginalFilename() == null) { context.put("code", 4000); context.put("message", "文件名不可为空"); return; } String nowDate = DateUtil.getNowShortDate(); String folder = nowDate + "/"; String strDirPath = rootPath + folder; File dirPath = new File(strDirPath); if (!dirPath.exists()) { dirPath.mkdirs(); } String uploadFileName = file.getOriginalFilename(); String x = UUID.randomUUID().toString().replace("-", "") + uploadFileName.substring(uploadFileName.lastIndexOf(".")); String fileName = folder + x; String fileNames = nowDate + "/" + x; uploadFileName = uploadFileName.replace(" ", ""); uploadFileName = uploadFileName.replace("，", ","); uploadFileName = uploadFileName.replaceAll(",", "-"); String fileAndPath = dir + fileName; // 判断如果临时目录中存在相同名称的文件先删除，在上传 File tempFile = new File(rootPath + fileName); if (tempFile.isFile() && tempFile.exists()) { tempFile.getAbsoluteFile().delete(); } // 上传到服务器临时文件夹 file.transferTo(tempFile); // 转移到FTP服务器 String nfix = ""; if (null != uploadFileName) { nfix = uploadFileName.substring(uploadFileName.lastIndexOf(".")); } if (StringUtils.equalsIgnoreCase(nfix, ".exe")) { context.put("code", 4000); context.put("message", "对不起，文件格式\".exe\"上传有误！"); return; } if (StringUtils.equalsIgnoreCase(nfix, ".dll")) { context.put("code", 4000); context.put("message", "对不起，文件格式\".dll\"上传有误！"); return; } String remoteFileName = fileAndPath; if (true) { context.put("success", true); context.put("code", 200); context.put("errno",0); JSONObject fileJSON = new JSONObject(); fileJSON.put("url", remoteFileName); fileJSON.put("imgaddr", fileNames); fileJSON.put("imgname", uploadFileName); fileJSON.put("originname", uploadFileName); context.put("data",fileJSON); context.put("message","请求成功"); writerJson(response, context); return; } else { // 移动失败 context.put("code", 0); context.put("message", "上传失败"); writerJson(response, context); } } context.put("code", 0); context.put("message", "上传失败"); writerJson(response, context); return; } } context.put("code", 0); context.put("message", "上传失败"); writerJson(response, context); return; } public static void writerJson(HttpServletResponse response, Object object) { response.setContentType("application/json"); writer(response, JSONObject.toJSONString(object)); } private static void writer(HttpServletResponse response, String str) { try { StringBuffer result = new StringBuffer(); //设置页面不缓存 response.setHeader("Pragma", "No-cache"); response.setHeader("Cache-Control", "no-cache"); response.setCharacterEncoding("UTF-8"); PrintWriter out = null; out = response.getWriter(); out.print(str); out.flush(); out.close(); } catch (IOException e) { e.printStackTrace(); } } }