package com.doumee.api.common; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.doumee.api.BaseController; import com.doumee.biz.system.SystemDictDataBiz; import com.doumee.config.annotation.EncryptionReq; import com.doumee.config.annotation.EncryptionResp; import com.doumee.core.annotation.trace.Trace; import com.doumee.core.constants.Constants; import com.doumee.core.constants.ResponseStatus; import com.doumee.core.exception.BusinessException; import com.doumee.core.model.ApiResponse; import com.doumee.core.utils.DateUtil; import com.doumee.core.utils.FtpUtil; import com.doumee.core.utils.azure.AzureBlobUtil; import com.doumee.core.utils.qiyeweixin.QywxUtil; import com.doumee.core.utils.qiyeweixin.model.response.QywxUploadMediaResponse; import com.doumee.dao.business.dto.WebQwUploadDto; import com.doumee.dao.system.model.SystemDictData; import com.doumee.service.common.EmailService; import io.swagger.annotations.Api; 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.beans.factory.annotation.Value; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import org.springframework.web.multipart.MultipartHttpServletRequest; import org.springframework.web.multipart.MultipartRequest; 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.InputStream; import java.io.PrintWriter; import java.util.*; /** * @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; @Autowired private EmailService emailService; @Value("${upload.type}") private String uploadType; public static FtpUtil ftp = null; public static AzureBlobUtil azureBlobUtil = null; @GetMapping("/image/{folder}/{date}/{name}") public void getImage(@PathVariable String folder,@PathVariable String date,@PathVariable String name, HttpServletResponse response) throws IOException { try { initUploadTool(); response.setContentType("image/jpeg"); // 根据你的图片类型设置正确的MIME类型 log.info("===================图片blob名称"+folder+"/"+date+"/"+name); azureBlobUtil.downloadBlobToStream(folder+"/"+date+"/"+name,response.getOutputStream()); }catch (Exception e) { e.printStackTrace(); } } @ApiOperation(value = "测试企业素材上传下载") @RequestMapping(method= RequestMethod.POST,value="/testQwImg") @ResponseBody public ApiResponse> testQwImg( @RequestParam(required = false) String imgurl ) throws Exception { QywxUploadMediaResponse r = QywxUtil.uploadMedia(systemDictDataBiz.queryByCode(Constants.QYWX,Constants.QYWX_TOKEN).getCode(),"image",imgurl); if(r!=null && r.getMedia_id()!=null){ String folder = systemDictDataBiz.queryByCode(Constants.FTP,Constants.WORKORDER_FILE_PATH).getCode(); String prefixPath = systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_RESOURCE_PATH).getCode(); Map context = new HashMap<>(); try { initUploadTool(); context.put("success", true); context.put("code", 200); context.put("errno",0); JSONArray jsonArray = new JSONArray(); Integer index = Constants.ONE; //获得每一个文件 String endType = ".jpg"; String originname = UUID.randomUUID().toString()+endType; InputStream is = QywxUtil.getMediaInputstream(systemDictDataBiz.queryByCode(Constants.QYWX,Constants.QYWX_TOKEN).getCode(),r.getMedia_id()); if(is == null){ context.put("code", 0); context.put("message", "上传失败"); context.put("errno",0); }else{ String date = DateUtil.getNowShortDate(); String fName = date + "/" + originname; String fileName = folder + fName; boolean r1 =false; if(StringUtils.equals(Constants.uploadType.blob,uploadType)){ r1 = azureBlobUtil.uploadFileWithResponseAndSize( fileName,is,null);//上传 }else{ r1 = ftp.uploadInputstreamBatch(is, fileName, true,index); } if (r1) { JSONObject fileJSON = new JSONObject(); fileJSON.put("url", prefixPath + fileName); fileJSON.put("imgaddr", fName); fileJSON.put("imgname", fileName); fileJSON.put("originname", originname); jsonArray.add(fileJSON); } } context.put("data",jsonArray); } catch (Exception e) { log.error("【上传FTP失败】======================"+e.getMessage()); context.put("code", 0); context.put("message", "上传失败"); context.put("errno",0); } return ApiResponse.success(context); } return ApiResponse.failed("上传失败"+r!=null?JSONObject.toJSONString(r):""); } @ApiOperation(value = "测试邮件发送") @RequestMapping(method= RequestMethod.POST,value="/testEmail") @ResponseBody public void test( @RequestParam(required = false) String id, @RequestParam(required = false) String email) throws Exception { // sendEmailWithImages(String toEmail, String title, Map contentForm, List imgList) Map content = new HashMap<>(); content.put("标题","这是标题"); content.put("简介","这是简介"); List list = new ArrayList<>(); list.add("https://dmtest.ahapp.net/file/workorder/20250410/1.jpg"); list.add("https://dmtest.ahapp.net/file/workorder/20250410/2.png"); list.add("https://dmtest.ahapp.net/file/workorder/20250410/4.png"); emailService.sendEmailWithImages(StringUtils.defaultString(email,"jp@doumee.com"),"来自蹄蹄的邮件"+DateUtil.getPlusTime2(new Date()),content,list); } @ApiOperation(value = "批量上传文件到FTP") @RequestMapping(method= RequestMethod.POST,value="/uploadBatch") @ResponseBody public void uploadBatch(MultipartRequest request, HttpServletResponse response, String folder) throws Exception { if(Objects.isNull(folder)){ throw new BusinessException(ResponseStatus.BAD_REQUEST); } Set setList = request.getMultiFileMap().keySet(); if(Objects.isNull(setList)){ throw new BusinessException(ResponseStatus.BAD_REQUEST); } List fileList = new ArrayList<>(); for(String str:setList){ List fl = request.getMultiFileMap().get(str); fileList.addAll(fl); } if(Objects.isNull(fileList)){ throw new BusinessException(ResponseStatus.BAD_REQUEST); } SystemDictData folderData = systemDictDataBiz.queryByCode(Constants.FTP,folder); if(Objects.isNull(folderData)||StringUtils.isBlank(folderData.getCode())){ throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"目标文件夹错误"); } folder = systemDictDataBiz.queryByCode(Constants.FTP,folder).getCode(); String prefixPath = systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_RESOURCE_PATH).getCode(); response.setCharacterEncoding("UTF-8"); response.setContentType("text/html;charset=UTF-8"); Map context = new HashMap<>(); try { initUploadTool(); context.put("success", true); context.put("code", 200); context.put("errno",0); JSONArray jsonArray = new JSONArray(); if (fileList.size() > 0) { //遍历文件列表 Integer index = Constants.ONE; for (MultipartFile file :fileList) { //获得每一个文件 String originname = file.getOriginalFilename(); log.error("【上传FTP成功】=============开始========="+index+"===="+originname); InputStream is = file.getInputStream(); String endType = originname.substring(originname.lastIndexOf("."), originname.length()); String date = DateUtil.getNowShortDate(); String fName = date + "/" + UUID.randomUUID() + endType; String fileName = folder + fName; boolean r =false; if(StringUtils.equals(Constants.uploadType.blob,uploadType)){ r = azureBlobUtil.uploadFileWithResponseAndSize( fileName,is,null);//上传 }else{ r = ftp.uploadInputstreamBatch(is, fileName, Constants.equalsInteger(index ,fileList.size()),index); } if (r) { JSONObject fileJSON = new JSONObject(); fileJSON.put("url", prefixPath + fileName); fileJSON.put("imgaddr", fName); fileJSON.put("imgname", fileName); fileJSON.put("originname", originname); jsonArray.add(fileJSON); } log.error("【上传FTP成功】=============结束========="+index+"====="+r+"===="+fName); index = index + 1; } context.put("data",jsonArray); } } catch (Exception e) { log.error("【上传FTP失败】======================"+e.getMessage()); context.put("code", 0); context.put("message", "上传失败"); context.put("errno",0); writerJson(response, context); return; } writerJson(response, context); return; } @ApiOperation(value = "批量从企业临时素材库上传文件到文件服务器") @PostMapping(value="/uploadBatchFromWx") @ResponseBody public void uploadBatchFromWx(@RequestBody WebQwUploadDto param, HttpServletResponse response) throws Exception { if(Objects.isNull(param.getFolder()) || param.getMediaIds()==null || param.getMediaIds().size()==0){ throw new BusinessException(ResponseStatus.BAD_REQUEST); } SystemDictData folderData = systemDictDataBiz.queryByCode(Constants.FTP,param.getFolder()); if(Objects.isNull(folderData)||StringUtils.isBlank(folderData.getCode())){ throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"目标文件夹错误"); } String folder = systemDictDataBiz.queryByCode(Constants.FTP,param.getFolder()).getCode(); String prefixPath = systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_RESOURCE_PATH).getCode(); response.setCharacterEncoding("UTF-8"); response.setContentType("text/html;charset=UTF-8"); Map context = new HashMap<>(); try { initUploadTool(); context.put("success", true); context.put("code", 200); context.put("errno",0); JSONArray jsonArray = new JSONArray(); Integer index = Constants.ONE; for (String media : param.getMediaIds()) { //获得每一个文件 String endType = ".jpg"; String originname = UUID.randomUUID().toString()+endType; Map map = QywxUtil.getMediaInputstreamMap(systemDictDataBiz.queryByCode(Constants.QYWX,Constants.QYWX_TOKEN).getCode(),media); if(map == null || map.get("data") == null){ log.error("企业微信素材下载失败:========="+media); continue; } log.error("企业微信素材下载成功:========="+media); String date = DateUtil.getNowShortDate(); String fName = date + "/" + originname; String fileName = folder + fName; boolean r =false; if(StringUtils.equals(Constants.uploadType.blob,uploadType)){ r = azureBlobUtil.uploadFileWithResponseAndSize( fileName, (InputStream) map.get("data"),(Integer)map.get("size"));//上传 }else{ r = ftp.uploadInputstreamBatch( (InputStream) map.get("data"), fileName, Constants.equalsInteger(index ,param.getMediaIds().size()),index); } if (r) { JSONObject fileJSON = new JSONObject(); fileJSON.put("url", prefixPath + fileName); fileJSON.put("imgaddr", fName); fileJSON.put("imgname", fileName); fileJSON.put("originname", originname); jsonArray.add(fileJSON); } index = index + 1; } context.put("data",jsonArray); } catch (Exception e) { log.error("【上传FTP失败】======================"+e.getMessage()); context.put("code", 0); context.put("message", "上传失败"); context.put("errno",0); writerJson(response, context); return; } writerJson(response, context); return; } @ApiOperation(value = "批量上传文件到FTP") @RequestMapping(method= RequestMethod.POST,value="/uploadBatchOld") @ResponseBody public void uploadBatchOld( HttpServletRequest request, HttpServletResponse response, String folder) throws Exception { if(Objects.isNull(folder)){ throw new BusinessException(ResponseStatus.BAD_REQUEST); } SystemDictData folderData = systemDictDataBiz.queryByCode(Constants.FTP,folder); if(Objects.isNull(folderData)||StringUtils.isBlank(folderData.getCode())){ throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"目标文件夹错误"); } folder = systemDictDataBiz.queryByCode(Constants.FTP,folder).getCode(); String prefixPath = systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_RESOURCE_PATH).getCode(); response.setCharacterEncoding("UTF-8"); response.setContentType("text/html;charset=UTF-8"); Map context = new HashMap<>(); try { initUploadTool(); context.put("success", true); context.put("code", 200); context.put("errno",0); CommonsMultipartResolver multipartResovler = new CommonsMultipartResolver(); if (multipartResovler.isMultipart(request)) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; List fileList = multipartRequest.getFiles("file"); Iterator it = multipartRequest.getFileNames(); JSONArray jsonArray = new JSONArray(); if (fileList.size() > 0) { //遍历文件列表 Iterator fileIte = fileList.iterator(); Integer index = Constants.ONE; while (fileIte.hasNext()) { //获得每一个文件 MultipartFile file = fileIte.next(); String originname = file.getOriginalFilename(); InputStream is = file.getInputStream(); String endType = originname.substring(originname.lastIndexOf("."), originname.length()); String date = DateUtil.getNowShortDate(); String fName = date + "/" + UUID.randomUUID() + endType; String fileName = folder + fName; boolean r =false; if(StringUtils.equals(Constants.uploadType.blob,uploadType)){ r = azureBlobUtil.uploadFileWithResponseAndSize( fileName,is,null);//上传 }else{ r = ftp.uploadInputstreamBatch(is, fileName, Constants.equalsInteger(index ,fileList.size()),index); } if (r) { JSONObject fileJSON = new JSONObject(); fileJSON.put("url", prefixPath + fileName); fileJSON.put("imgaddr", fName); fileJSON.put("imgname", fileName); fileJSON.put("originname", originname); jsonArray.add(fileJSON); } index = index + 1; } } context.put("data",jsonArray); } } catch (Exception e) { log.error("【上传FTP失败】======================"+e.getMessage()); context.put("code", 0); context.put("message", "上传失败"); context.put("errno",0); writerJson(response, context); return; } writerJson(response, context); return; } private void initUploadTool() throws Exception{ if(StringUtils.equals(Constants.uploadType.blob,uploadType)){ if(azureBlobUtil == null){ azureBlobUtil = new AzureBlobUtil( systemDictDataBiz.queryByCode(Constants.AZURE_BLOB,Constants.AZURE_BLOB_ACCESSNAME).getCode(), systemDictDataBiz.queryByCode(Constants.AZURE_BLOB,Constants.AZURE_BLOB_ACCESSKEY).getCode(), systemDictDataBiz.queryByCode(Constants.AZURE_BLOB,Constants.AZURE_BLOB_ENDPOINT).getCode(), systemDictDataBiz.queryByCode(Constants.AZURE_BLOB,Constants.AZURE_BLOB_CONTAINER).getCode()); }else{ azureBlobUtil.initClient(); } }else{ if(ftp == null){ ftp = new FtpUtil(systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_HOST).getCode(), Integer.parseInt(systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_PORT).getCode()), systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_USERNAME).getCode(), systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_PWD).getCode()); }else{ ftp.connect(); } } } @ApiOperation(value = "上传文件到FTP") @RequestMapping(method= RequestMethod.POST,value="/upload") @ResponseBody public void upload(HttpServletRequest request, HttpServletResponse response, String folder) throws Exception { // folder = systemDictDataBiz.queryByCode(Constants.FTP,folder).getCode(); Date d1 = new Date(); log.error("总得上传文件成功=============开始========="+DateUtil.getPlusTime2(d1)); SystemDictData folderData = systemDictDataBiz.queryByCode(Constants.FTP,folder); if(Objects.isNull(folderData)||StringUtils.isBlank(folderData.getCode())){ throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"目标文件夹错误"); } folder =folderData.getCode().replace("/", ""); String prefixPath = systemDictDataBiz.queryByCode(Constants.FTP,Constants.FTP_RESOURCE_PATH).getCode(); InputStream is = null; response.setCharacterEncoding("UTF-8"); response.setContentType("text/html;charset=UTF-8"); Map context = new HashMap<>(); try { initUploadTool(); CommonsMultipartResolver multipartResovler = new CommonsMultipartResolver(); if (multipartResovler.isMultipart(request)) { MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request; Iterator it = multipartRequest.getFileNames(); while (it.hasNext()) { MultipartFile file = multipartRequest.getFile(it.next()); System.out.println(file.getSize()); String originname = file.getOriginalFilename(); is = file.getInputStream(); String endType = ".jpg"; if(originname.indexOf(".")>0){ endType=originname.substring(originname.lastIndexOf("."),originname.length()); } String date = DateUtil.getNowShortDate(); String fName = date+"/"+ UUID.randomUUID()+endType; String fileName = folder+"/"+fName; boolean r =false; if(StringUtils.equals(Constants.uploadType.blob,uploadType)){ r = azureBlobUtil.uploadFileWithResponseAndSize( fileName,is,null);//上传 }else{ r = ftp.uploadInputstream(is,fileName); } if(r){ context.put("success", true); context.put("code", 200); context.put("errno",0); JSONObject fileJSON = new JSONObject(); // fileJSON.put("prefixPath", prefixPath); // fileJSON.put("folder", folder); fileJSON.put("url", prefixPath+fileName); fileJSON.put("imgaddr", fName); fileJSON.put("imgname", fileName); fileJSON.put("originname", originname); context.put("data",fileJSON); context.put("message","请求成功"); writerJson(response, context); Date d2= new Date(); log.error("总的上传文件成功=============结束========="+DateUtil.getPlusTime2(d2) +"耗时秒:"+( (d2.getTime()-d1.getTime()) /1000)); return; } } } } catch (Exception e) { log.error("【上传FTP失败】======================"+e.getMessage()); } context.put("code", 0); context.put("message", "上传失败"); context.put("errno",0); 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(); } } public void uploadFileLocal(HttpServletRequest request, String folder, 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());// file // = // (CommonsMultipartFile) // 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; } /* * if(file.getOriginalFilename().contains(",")||file.getOriginalFilename(). * contains(" ")){ sendFailureMessage(response,"文件名称有误,不可含有逗号等特殊字符"); } */ String nowDate = DateUtil.getNowShortDate(); 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(",", "-"); System.err.println("R:" + fileName); String fileAndPath = dir + fileName; System.err.println("A:" + fileAndPath); // 判断如果临时目录中存在相同名称的文件先删除,在上传 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 remoteName = fileName; String remoteFileName = fileAndPath; /* Map map = new HashMap(); map.put("url", remoteFileName); map.put("alt", uploadFileName); map.put("href", remoteFileName); List maps = Arrays.asList(remoteFileName);*/ if (true) { // 关闭FTP流 // 移动成功,返回文件名 /*context.put("error", 0); context.put("url", remoteFileName); context.put("fullurl", remoteFileName); context.put("fname", uploadFileName); context.put("data", maps); context.put("halfurl", fileNames);*/ 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; } }