From b57fd54e9e3b9a5f84981a23560ad3200e7b7e5a Mon Sep 17 00:00:00 2001 From: dingjie Date: Fri, 15 Dec 2023 21:07:49 +0800 Subject: [PATCH] =?UTF-8?q?=E6=96=87=E4=BB=B6=E4=B8=8A=E4=BC=A0=E6=8E=A5?= =?UTF-8?q?=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../system/controller/SysFileController.java | 92 +++++++++ .../bonus/sgzb/system/domain/FileInfo.java | 40 ++++ .../sgzb/system/mapper/FileInfoMapper.java | 17 ++ .../sgzb/system/service/SysFileService.java | 23 +++ .../service/impl/SysFileServiceImpl.java | 179 ++++++++++++++++++ .../mapper/system/FileInfoMapper.xml | 79 ++++++++ 6 files changed, 430 insertions(+) create mode 100644 sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/controller/SysFileController.java create mode 100644 sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/domain/FileInfo.java create mode 100644 sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/mapper/FileInfoMapper.java create mode 100644 sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/SysFileService.java create mode 100644 sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/impl/SysFileServiceImpl.java create mode 100644 sgzb-modules/sgzb-system/src/main/resources/mapper/system/FileInfoMapper.xml diff --git a/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/controller/SysFileController.java b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/controller/SysFileController.java new file mode 100644 index 00000000..b7aaaa1c --- /dev/null +++ b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/controller/SysFileController.java @@ -0,0 +1,92 @@ +package com.bonus.sgzb.system.controller; + +import com.bonus.sgzb.common.core.utils.StringHelper; +import com.bonus.sgzb.common.core.web.domain.AjaxResult; +import com.bonus.sgzb.system.domain.FileInfo; +import com.bonus.sgzb.system.service.SysFileService; +import io.swagger.annotations.ApiOperation; +import org.apache.poi.util.IOUtils; +import org.springframework.core.io.ResourceLoader; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestMapping; +import org.springframework.web.bind.annotation.RestController; + +import javax.annotation.Resource; +import javax.servlet.ServletOutputStream; +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.io.InputStream; +import java.net.URLEncoder; +import java.nio.charset.StandardCharsets; + +@RestController +@RequestMapping("/sys/file") +public class SysFileController { + + @Resource + private SysFileService service; + + @Resource + private ResourceLoader resourceLoader; + + + @PostMapping("/upload") + @ApiOperation(value = "文件上传") + public AjaxResult upload(HttpServletRequest request) { + String limitWords = request.getParameter("limitWords"); + FileInfo file = new FileInfo(); + try { + file = service.uploadFile(request); + if (limitWords != null && file.getWords() > Integer.parseInt(limitWords)){ + return AjaxResult.error("上传文件字数超出限制字数！"); + } + }catch (Exception e){ + e.printStackTrace(); + } + if (file != null && file.getId() != 0){ + return AjaxResult.success(file); + }else { + return AjaxResult.error("文件上传失败！"); + } + + } + + @ApiOperation(value = "模板", httpMethod = "GET") + @GetMapping("download") + public void download(HttpServletRequest request, HttpServletResponse response, String filename) { + InputStream inputStream = null; + ServletOutputStream servletOutputStream = null; + try { + String path = "download/" + filename; + org.springframework.core.io.Resource resource = resourceLoader.getResource("classpath:" + path); + response.setContentType("application/vnd.ms-excel"); + response.addHeader("Cache-Control", "no-cache, no-store, must-revalidate"); + response.addHeader("charset", "utf-8"); + response.addHeader("Pragma", "no-cache"); + String encodeName = URLEncoder.encode(filename, StandardCharsets.UTF_8.toString()); + response.setHeader("Content-Disposition", "attachment; filename=\"" + encodeName + "\"; filename*=utf-8''" + encodeName); + + inputStream = resource.getInputStream(); + servletOutputStream = response.getOutputStream(); + IOUtils.copy(inputStream, servletOutputStream); + response.flushBuffer(); + } catch (Exception e) { + e.printStackTrace(); + } finally { + try { + if (servletOutputStream != null) { + servletOutputStream.close(); + } + if (inputStream != null) { + inputStream.close(); + } + // 召唤jvm的垃圾回收器 + System.gc(); + } catch (Exception e) { + e.printStackTrace(); + } + } + } + +} diff --git a/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/domain/FileInfo.java b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/domain/FileInfo.java new file mode 100644 index 00000000..aaad32f6 --- /dev/null +++ b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/domain/FileInfo.java @@ -0,0 +1,40 @@ +package com.bonus.sgzb.system.domain; + +import com.fasterxml.jackson.annotation.JsonFormat; +import io.swagger.annotations.ApiModelProperty; +import lombok.Data; + +import java.util.Date; + +@Data +public class FileInfo { + + /** $column.columnComment */ + private Long id; + + /** 模块id */ + private Long modelId; + /** 文件名称 */ + @ApiModelProperty(value = "文件名称") + private String fileName; + + /** 文件路径 */ + @ApiModelProperty(value = "文件路径") + private String fileUrl; + + /** 数据字典 */ + @ApiModelProperty(value = "数据字典") + private Long dicId; + + /** 创建者 */ + private String createBy; + + /** 创建时间 */ + @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") + private Date createTime; + private String size; + private String type; + private int words; + private String creator; + +} diff --git a/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/mapper/FileInfoMapper.java b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/mapper/FileInfoMapper.java new file mode 100644 index 00000000..fc1a4522 --- /dev/null +++ b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/mapper/FileInfoMapper.java @@ -0,0 +1,17 @@ +package com.bonus.sgzb.system.mapper; + +import com.bonus.sgzb.system.domain.FileInfo; + +/** + *

+ * Mapper 接口 + *

+ * + * @author laker + * @since 2021-08-05 + */ +public interface FileInfoMapper { + + int insertFileInfo(FileInfo o); + +} diff --git a/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/SysFileService.java b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/SysFileService.java new file mode 100644 index 00000000..0f70509d --- /dev/null +++ b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/SysFileService.java @@ -0,0 +1,23 @@ +package com.bonus.sgzb.system.service; + + +import com.bonus.sgzb.system.domain.FileInfo; + +import javax.servlet.http.HttpServletRequest; + +/** + * 文件上传接口 + * + * @author zys + */ +public interface SysFileService +{ + /** + * 文件上传接口 + * + * @return + * @throws Exception + */ + FileInfo uploadFile(HttpServletRequest request) throws Exception; + +} diff --git a/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/impl/SysFileServiceImpl.java b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/impl/SysFileServiceImpl.java new file mode 100644 index 00000000..62ec5e23 --- /dev/null +++ b/sgzb-modules/sgzb-system/src/main/java/com/bonus/sgzb/system/service/impl/SysFileServiceImpl.java @@ -0,0 +1,179 @@ +package com.bonus.sgzb.system.service.impl; + +import cn.hutool.core.util.IdUtil; +import com.bonus.sgzb.common.core.utils.DateTimeHelper; +import com.bonus.sgzb.common.security.utils.SecurityUtils; +import com.bonus.sgzb.system.api.model.LoginUser; +import com.bonus.sgzb.system.domain.FileInfo; +import com.bonus.sgzb.system.mapper.FileInfoMapper; +import com.bonus.sgzb.system.service.SysFileService; +import org.apache.poi.hwpf.extractor.WordExtractor; +import org.apache.poi.openxml4j.util.ZipSecureFile; +import org.apache.poi.xwpf.extractor.XWPFWordExtractor; +import org.apache.poi.xwpf.usermodel.XWPFDocument; +import org.springframework.context.annotation.Primary; +import org.springframework.stereotype.Service; +import org.springframework.web.multipart.MultipartFile; +import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest; + +import javax.annotation.Resource; +import javax.servlet.http.HttpServletRequest; +import java.io.File; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.InputStream; +import java.security.Security; +import java.util.*; + +/** + * 本地文件存储 + * + * @author zys + */ +@Primary +@Service("SysFileService") +public class SysFileServiceImpl implements SysFileService { + + @Resource + private FileInfoMapper dao; + + /** + * 本地文件上传接口 + * + * @return 访问地址 + * @throws Exception + */ + @Override + public FileInfo uploadFile(HttpServletRequest request) throws Exception { + FileInfo file = new FileInfo(); + StandardMultipartHttpServletRequest req = (StandardMultipartHttpServletRequest) request; + String photoType = req.getParameter("fileType"); + String userId = req.getParameter("userId"); + HashMap map = getFile(req); + List items = (List) map.get("filePath"); + MultipartFile item = items.get(0); + try { + String url = saveFile(request, item, photoType); + if(url != null){ + int words = getFileText(item); + String fileName = item.getOriginalFilename(); + String type = fileName.substring(fileName.lastIndexOf(".") + 1); + long size = item.getSize()/1024/1024; + file.setFileName(fileName); + file.setFileUrl(url); + file.setCreator(userId); + file.setType(type); + file.setSize(size + "M"); + file.setWords(words); + file.setCreateBy(SecurityUtils.getUserId().toString()); + file.setCreateTime(new Date()); + dao.insertFileInfo(file); + } + } catch (Exception e) { + e.printStackTrace(); + } + return file; + } + + public HashMap getFile(StandardMultipartHttpServletRequest request) { + MultipartFile multipartFile; + HashMap map = new HashMap(); + try { + List tmps = new ArrayList(); + Iterator itr = request.getFileNames(); + while (itr.hasNext()) { + multipartFile = request.getFile(itr.next()); + tmps.add(multipartFile); + } + map.put("filePath",tmps); + } catch (Exception e) { + e.printStackTrace(); + } + return map; + } + + public String saveFile(HttpServletRequest request, MultipartFile multipartFile, String fileType) throws Exception { + String url = ""; + String tmpName = multipartFile.getOriginalFilename();// 完整路径 IE + tmpName = tmpName.substring(tmpName.lastIndexOf("\\") + 1); + tmpName = IdUtil.fastSimpleUUID() + tmpName; + String imageFiles="/data/sz/zstp/" + fileType + "/"; + String os = System.getProperty("os.name"); + if(os.toLowerCase().startsWith("win")){ + imageFiles="D://files/" + fileType + "/"; + } + String year = DateTimeHelper.getNowYear(); + String month = DateTimeHelper.getNowMonths(); + String day = DateTimeHelper.getNowDay(); + String specfile = imageFiles + year +"/" + month +"/"+ day; + File file = new File(specfile + "/" + tmpName); + + if (!file.getParentFile().exists()) { + file.getParentFile().mkdirs(); + } + url = "/" + fileType + "/" + year +"/" + month +"/"+ day + "/" + tmpName; + if (!multipartFile.isEmpty()) { + try { + FileOutputStream fos = new FileOutputStream(file); + InputStream in = multipartFile.getInputStream(); + byte[] bytes = new byte[1024]; + int len = 0; + while ((len = in.read(bytes)) != -1) { + fos.write(bytes, 0, len); + } + fos.close(); + in.close(); + } catch (Exception e) { + e.printStackTrace(); + } + } + return url; + } + + private int getFileText(MultipartFile file) throws IOException { + int length = 0; + String fileName = file.getOriginalFilename(); + InputStream fileInputStream = file.getInputStream(); + try { + ZipSecureFile.setMinInflateRatio(-1.0d); + //获取文件后缀名 + String suffix = fileName.substring(fileName.lastIndexOf(".") + 1); + //定义word内容 + String content = ""; + switch (suffix) { + case "doc": + WordExtractor wordExtractor = new WordExtractor(fileInputStream); + content = wordExtractor.getText(); + break; + case "docx": + XWPFDocument document = new XWPFDocument(fileInputStream); + XWPFWordExtractor extractor = new XWPFWordExtractor(document); + content = extractor.getText(); + break; + default: + break; + } + //中文单词 + String cnWords = content.replaceAll("[^(\\u4e00-\\u9fa5，。《》？；’‘：“”【】、）（……￥！·)]", ""); + int cnWordsCount = cnWords.length(); + //非中文单词 + String noCnWords = content.replaceAll("[^(a-zA-Z0-9`\\-=\';.,/~!@#$%^&*()_+|}{\":> + + + + + + + + + + + + + + + select id, model_id, file_name, file_url, dic_id, create_by, create_time from sys_file_info + + + + + + + + insert into sys_file_info + + model_id, + file_name, + file_url, + dic_id, + create_by, + create_time, + + + #{modelId}, + #{fileName}, + #{fileUrl}, + #{dicId}, + #{createBy}, + #{createTime}, + + + + + update sys_file_info + + model_id = #{modelId}, + file_name = #{fileName}, + file_url = #{fileUrl}, + dic_id = #{dicId}, + create_by = #{createBy}, + create_time = #{createTime}, + + where id = #{id} + + + + delete from sys_file_info where id = #{id} + + + + delete from sys_file_info where id in + + #{id} + + + \ No newline at end of file