From 6cdbd8f9cb636cfa43c26651f76b14ce81f16ef6 Mon Sep 17 00:00:00 2001
From: xusd <hugeinfo123>
Date: Tue, 10 Sep 2024 10:17:55 +0800
Subject: [PATCH] ftp服务器修正
---
dyh-service/dyh-sys/src/main/java/cn/huge/module/file/controller/web/FileInfoWebController.java | 318 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 313 insertions(+), 5 deletions(-)
diff --git a/dyh-service/dyh-sys/src/main/java/cn/huge/module/file/controller/web/FileInfoWebController.java b/dyh-service/dyh-sys/src/main/java/cn/huge/module/file/controller/web/FileInfoWebController.java
index 088af92..d8987ec 100644
--- a/dyh-service/dyh-sys/src/main/java/cn/huge/module/file/controller/web/FileInfoWebController.java
+++ b/dyh-service/dyh-sys/src/main/java/cn/huge/module/file/controller/web/FileInfoWebController.java
@@ -1,22 +1,43 @@
package cn.huge.module.file.controller.web;
+import cn.huge.base.common.exception.ClientException;
+import cn.huge.module.file.domain.dto.IdFileDTO;
+import cn.huge.module.utils.BaiduOcrUtils;
+import cn.huge.base.common.utils.ContentTypeUtils;
import cn.huge.base.common.utils.ReturnFailUtils;
import cn.huge.base.common.utils.ReturnSucUtils;
+import cn.huge.base.config.CurrentUser;
+import cn.huge.module.client.api.impl.CustClientImpl;
+import cn.huge.module.constant.BaseConsts;
+import cn.huge.module.cust.dto.CtUserDTO;
+import cn.huge.module.file.domain.dto.UploaderDTO;
import cn.huge.module.file.domain.po.FileInfo;
import cn.huge.module.file.service.FileInfoService;
+import cn.huge.module.file.service.FileRelateService;
+import cn.huge.module.file.utils.FileUtils;
+import cn.huge.module.file.utils.FtpUtils;
+import cn.huge.module.sys.constant.FileBaseConsts;
+import cn.huge.module.sys.constant.FileOwnerTypeBaseEnum;
+import cn.huge.module.sys.dto.FileInfoBaseDTO;
import com.google.common.collect.Maps;
import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.data.domain.Page;
-import org.springframework.data.domain.PageRequest;
-import org.springframework.data.domain.Sort;
import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+import org.springframework.web.multipart.MultipartHttpServletRequest;
import javax.servlet.http.HttpServletRequest;
-import java.util.Map;
+import javax.servlet.http.HttpServletResponse;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.net.URLEncoder;
+import java.util.*;
import java.util.stream.Collectors;
-import java.util.stream.Stream;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
/**
* @title: 附件信息表接口api-web端
@@ -36,5 +57,292 @@
@Autowired
private FileInfoService service;
+ @Autowired
+ private FileRelateService fileRelateService;
+ @Autowired
+ private CustClientImpl custClient;
+ /**
+ * 页面附件上传-文件流方式
+ *
+ * @param mainId 所属业务主体编号
+ * @param ownerId 所属业务编号
+ * @param ownerType 所属业务类型
+ * @param request 请求头
+ * @return Object
+ * @url {ctx}/api/web/fileInfo/upload?mainId=&ownerId=&ownerType=
+ */
+ @PostMapping("/upload")
+ public Object upload(@RequestParam(value = "mainId") String mainId,
+ @RequestParam(value = "ownerId") String ownerId,
+ @RequestParam(value = "ownerType") String ownerType,
+ @CurrentUser String userId, MultipartHttpServletRequest request) {
+ try {
+ CtUserDTO loginUser = custClient.clientGetUser(userId);
+ UploaderDTO uploaderDTO = new UploaderDTO();
+ uploaderDTO.setUploaderId(loginUser.getId());
+ uploaderDTO.setUploaderName(loginUser.getTrueName());
+ uploaderDTO.setUploaderType(FileBaseConsts.UPLOADER_TYPE_1);
+ uploaderDTO.setCustId(loginUser.getCustId());
+ List<FileInfoBaseDTO> files = new ArrayList<>();
+ Iterator<String> itr = request.getFileNames();
+ int fileCount = fileRelateService.countByOwnerIdAndType(ownerId, ownerType) + 1;
+ while (itr.hasNext()) {
+ MultipartFile file = request.getFile(itr.next());
+ if (file.getSize() > 30000000) {
+ return ReturnFailUtils.getRepInfo("您上传的文件已超过30M大小限制！");
+ } else {
+ String fileSuffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf("."));
+ if (FileUtils.checkFileSuffix(fileSuffix)) {
+ if (StringUtils.isEmpty(ownerType)) {
+ ownerType = FileOwnerTypeBaseEnum.OWNER_TYPE_000.getIndex();
+ }
+ FileInfo fileInfo = service.webUploadFile(file, mainId, ownerId, ownerType, fileCount, uploaderDTO);
+ FileInfoBaseDTO fileInfoBaseDTO = new FileInfoBaseDTO();
+ BeanUtils.copyProperties(fileInfo, fileInfoBaseDTO);
+
+ // 如果是身份证，就调用ocr识别
+ if (FileOwnerTypeBaseEnum.OWNER_TYPE_202.getIndex().equals(ownerType)) {
+ fileInfoBaseDTO.setIdcardOcrResult(BaiduOcrUtils.ocrIdcard(file.getBytes()));
+ }
+
+ files.add(fileInfoBaseDTO);
+ fileCount++;
+ } else {
+ return ReturnFailUtils.getRepInfo("您上传的文件格式不正确！");
+ }
+ }
+ }
+ return ReturnSucUtils.getRepInfo("上传成功！", files);
+ } catch (Exception e) {
+ return ReturnFailUtils.getRepInfo();
+ }
+ }
+
+ /**
+ * 在线查附件-文件流方式
+ *
+ * @param id 附件编号
+ * @param response 响应头
+ * @throws Exception
+ * @url {ctx}/api/web/fileInfo/show/{id}
+ */
+ @GetMapping("/show/{id}")
+ public void show(@PathVariable(value = "id") String id, HttpServletResponse response) {
+ try {
+ FileInfo fileInfo = service.getById(id);
+ response.reset();
+ String contentType = ContentTypeUtils.getContentType(fileInfo.getSuffix());
+ response.setContentType(contentType);
+ OutputStream outputStream = response.getOutputStream();
+ FtpUtils ftpUtil = new FtpUtils();
+ InputStream inputStream = ftpUtil.retrieveFileStream(fileInfo.getFullPath());
+
+// double fileSize = fileInfo.getSize(); // 文件大小
+ int chunkSize = 512 * 1024; // 分块大小（ KB）
+ byte[] buffer = new byte[chunkSize];
+ int bytesRead;
+
+ // 循环读取和发送文件的各个部分
+ while ((bytesRead = inputStream.read(buffer)) != -1) {
+ outputStream.write(buffer, 0, bytesRead);
+ outputStream.flush();
+ }
+
+// IOUtils.copy(inputStream, outputStream, chunkSize);
+ //关闭流
+ IOUtils.closeQuietly(inputStream);
+ IOUtils.closeQuietly(outputStream);
+ } catch (Exception e) {
+ log.error("controller接口[FileInfoWebController.show]请求失败，异常信息：" + e, e);
+ throw new ClientException("FileInfoWebController.show", e);
+ }
+ }
+
+ /**
+ * 在线下载附件-文件流方式
+ *
+ * @param id 附件编号
+ * @param response 响应头
+ * @url {ctx}/api/web/fileInfo/down/{id}
+ */
+ @GetMapping("/down/{id}")
+ public void down(@PathVariable(value = "id") String id, HttpServletResponse response) {
+ try {
+ FileInfo fileInfo = service.getById(id);
+ String contentType = ContentTypeUtils.getContentType(fileInfo.getSuffix());
+ response.setContentType(contentType);
+ String fileName = URLEncoder.encode(fileInfo.getName() + BaseConsts.DOT + fileInfo.getSuffix(), "UTF-8");
+ response.addHeader("Content-Disposition", "attachment;filename=" + fileName);
+ OutputStream outputStream = response.getOutputStream();
+ FtpUtils ftpUtil = new FtpUtils();
+ InputStream inputStream = ftpUtil.retrieveFileStream(fileInfo.getFullPath());
+ IOUtils.copy(inputStream, outputStream);
+ //关闭流
+ IOUtils.closeQuietly(inputStream);
+ IOUtils.closeQuietly(outputStream);
+ } catch (Exception e) {
+ log.error("controller接口[FileInfoWebController.down]请求失败，异常信息：" + e, e);
+ throw new ClientException("FileInfoWebController.down", e);
+ }
+ }
+
+ @GetMapping("/down/batch")
+ public void downBatch(@RequestParam(value = "ids") String ids, HttpServletResponse response) {
+ try {
+ String[] id = ids.split(",");
+ // 设置ZIP文件响应类型
+ response.setContentType("application/zip");
+ response.setHeader("Content-Disposition", "attachment;filename=files.zip");
+
+ ZipOutputStream zipOutputStream = new ZipOutputStream(response.getOutputStream());
+ for (String s : id) {
+ FileInfo fileInfo = service.getById(s);
+ FtpUtils ftpUtil = new FtpUtils();
+ InputStream inputStream = ftpUtil.retrieveFileStream(fileInfo.getFullPath());
+
+ // 每个文件写入到ZIP包中
+ zipOutputStream.putNextEntry(new ZipEntry(fileInfo.getName() + BaseConsts.DOT + fileInfo.getSuffix()));
+ IOUtils.copy(inputStream, zipOutputStream);
+ zipOutputStream.closeEntry();
+ IOUtils.closeQuietly(inputStream);
+ }
+ IOUtils.closeQuietly(zipOutputStream);
+ } catch (Exception e) {
+ log.error("controller接口[FileInfoWebController.down]请求失败，异常信息：" + e, e);
+ throw new ClientException("FileInfoWebController.down", e);
+ }
+ }
+
+ /**
+ * ocr识别文字
+ *
+ * @param request
+ * @return Object
+ * @url {ctx}/api/web/fileInfo/recognitionText?ownerId=&ownerType=
+ */
+ @PostMapping(value = "/recognitionText")
+ public Object recognitionText(MultipartHttpServletRequest request) {
+ try {
+ Map<String, Object> result = Maps.newHashMap();
+ Iterator<String> itr = request.getFileNames();
+ while (itr.hasNext()) {
+ MultipartFile file = request.getFile(itr.next());
+ try {
+ Map<String, Object> map = BaiduOcrUtils.ocrText(file.getBytes());
+ result.put("ocrResult", map);
+ } catch (Exception e) {
+ log.error("Controller接口[FileInfoController.recognitionText]请求异常：" + e, e);
+ return ReturnFailUtils.getRepInfo("OCR失败！");
+ }
+ }
+ return ReturnSucUtils.getRepInfo("识别成功", result);
+ } catch (Exception e) {
+ return ReturnFailUtils.getRepInfo(e.getMessage());
+ }
+ }
+
+ /**
+ * 查看附件组件-分类查询附件
+ *
+ * @param mainId 所属业务主体编号
+ * @return Object
+ * @url {ctx}/api/web/fileInfo/listFileByCat
+ */
+ @GetMapping("/listFileByCat")
+ public Object listFileByCat(@RequestParam(value = "mainId") String mainId, @CurrentUser String userId) {
+ try {
+ Map<String, Object> terms = Maps.newHashMap();
+ // 材料大类
+ String fileOwnerCat = request.getParameter("fileOwnerCat");
+ if (StringUtils.isNotBlank(fileOwnerCat)) {
+ terms.put("ownerCat", fileOwnerCat);
+ }
+ // 上传时间区间
+ String createStart = request.getParameter("createStart");
+ String createEnd = request.getParameter("createEnd");
+ if (StringUtils.isNotBlank(createStart) && StringUtils.isNotBlank(createEnd)) {
+ terms.put("createStart", createStart);
+ terms.put("createEnd", createEnd);
+ }
+ // 上传人类型
+ String uploaderType = request.getParameter("uploaderType");
+ if (StringUtils.isNotBlank(uploaderType)) {
+ terms.put("uploaderType", uploaderType);
+ }
+ // 所属主体编号
+ terms.put("mainId", mainId);
+ return ReturnSucUtils.getRepInfo(service.webListFileByCat(terms));
+ } catch (Exception e) {
+ log.error("service方法[FileInfoWebController.listFileByCat]请求失败，异常信息：" + e, e);
+ return ReturnFailUtils.getRepInfo();
+ }
+ }
+
+ /**
+ * 删除附件
+ *
+ * @param ids 附件编号集合
+ * @return Object
+ * @url {ctx}/api/web/fileInfo/deleteFileById?id=
+ */
+ @DeleteMapping("/deleteFileByIds")
+ public Object deleteFileByIds(@RequestParam(value = "ids") String ids) {
+ try {
+ String[] id = ids.split(",");
+ for (String s : id) {
+ service.deleteFileById(s);
+ }
+ return ReturnSucUtils.getRepInfo();
+ } catch (Exception e) {
+ return ReturnFailUtils.getRepInfo();
+ }
+ }
+
+ @DeleteMapping("/deleteFileById")
+ public Object deleteFileById(@RequestParam(value = "id") String id) {
+ try {
+ service.deleteFileById(id);
+ return ReturnSucUtils.getRepInfo();
+ } catch (Exception e) {
+ return ReturnFailUtils.getRepInfo();
+ }
+ }
+
+ @DeleteMapping("/deleteFileById2")
+ public Object deleteFileById2(@RequestBody String id) {
+ try {
+ service.deleteFileById(id);
+ return ReturnSucUtils.getRepInfo();
+ } catch (Exception e) {
+ return ReturnFailUtils.getRepInfo();
+ }
+ }
+
+ /**
+ * 根据多个所属编号查询附件并根据先根据ownerId再根据附件类型分组
+ *
+ * @return Object
+ * @url {ctx}/api/web/fileInfo/listIdTypeInfo
+ */
+ @PostMapping("/listIdTypeInfo")
+ public Object listIdTypeInfo(@RequestBody IdFileDTO idFileDTO) {
+ try {
+ Map<String, Object> term = new HashMap<>();
+ String ownerIds = idFileDTO.getOwnerIdList().stream().map(String::valueOf).collect(Collectors.joining("','"));
+ String types = idFileDTO.getTypeList().stream().map(String::valueOf).collect(Collectors.joining("','"));
+ if (StringUtils.isNotBlank(ownerIds)) {
+ ownerIds = "'" + ownerIds + "'";
+ }
+ if (StringUtils.isNotBlank(types)) {
+ types = "'" + types + "'";
+ }
+ term.put("ownerIds", ownerIds);
+ term.put("types", types);
+ term.put("mainId", idFileDTO.getMainId());
+ return ReturnSucUtils.getRepInfo(service.listIdTypeInfoByOwnerIdList(term));
+ } catch (Exception e) {
+ return ReturnFailUtils.getRepInfo();
+ }
+ }
}
--
Gitblit v1.8.0