admin
2024-04-24 363457b34e0e4f26ffe51aa80ecb227bf7873308
提交 | 用户 | 时间
363457 1 package com.jcdm.web.controller.common;
A 2
3 import java.util.ArrayList;
4 import java.util.List;
5 import javax.servlet.http.HttpServletRequest;
6 import javax.servlet.http.HttpServletResponse;
7 import org.slf4j.Logger;
8 import org.slf4j.LoggerFactory;
9 import org.springframework.beans.factory.annotation.Autowired;
10 import org.springframework.http.MediaType;
11 import org.springframework.web.bind.annotation.GetMapping;
12 import org.springframework.web.bind.annotation.PostMapping;
13 import org.springframework.web.bind.annotation.RequestMapping;
14 import org.springframework.web.bind.annotation.RestController;
15 import org.springframework.web.multipart.MultipartFile;
16 import com.jcdm.common.config.MesConfig;
17 import com.jcdm.common.constant.Constants;
18 import com.jcdm.common.core.domain.AjaxResult;
19 import com.jcdm.common.utils.StringUtils;
20 import com.jcdm.common.utils.file.FileUploadUtils;
21 import com.jcdm.common.utils.file.FileUtils;
22 import com.jcdm.framework.config.ServerConfig;
23
24 /**
25  * 通用请求处理
26  * 
27  * @author jc
28  */
29 @RestController
30 @RequestMapping("/common")
31 public class CommonController
32 {
33     private static final Logger log = LoggerFactory.getLogger(CommonController.class);
34
35     @Autowired
36     private ServerConfig serverConfig;
37
38     private static final String FILE_DELIMETER = ",";
39
40     /**
41      * 通用下载请求
42      * 
43      * @param fileName 文件名称
44      * @param delete 是否删除
45      */
46     @GetMapping("/download")
47     public void fileDownload(String fileName, Boolean delete, HttpServletResponse response, HttpServletRequest request)
48     {
49         try
50         {
51             if (!FileUtils.checkAllowDownload(fileName))
52             {
53                 throw new Exception(StringUtils.format("文件名称({})非法,不允许下载。 ", fileName));
54             }
55             String realFileName = System.currentTimeMillis() + fileName.substring(fileName.indexOf("_") + 1);
56             String filePath = MesConfig.getDownloadPath() + fileName;
57
58             response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
59             FileUtils.setAttachmentResponseHeader(response, realFileName);
60             FileUtils.writeBytes(filePath, response.getOutputStream());
61             if (delete)
62             {
63                 FileUtils.deleteFile(filePath);
64             }
65         }
66         catch (Exception e)
67         {
68             log.error("下载文件失败", e);
69         }
70     }
71
72     /**
73      * 通用上传请求(单个)
74      */
75     @PostMapping("/upload")
76     public AjaxResult uploadFile(MultipartFile file) throws Exception
77     {
78         try
79         {
80             // 上传文件路径
81             String filePath = MesConfig.getUploadPath();
82             // 上传并返回新文件名称
83             String fileName = FileUploadUtils.upload(filePath, file);
84             String url = serverConfig.getUrl() + fileName;
85             AjaxResult ajax = AjaxResult.success();
86             ajax.put("url", url);
87             ajax.put("fileName", fileName);
88             ajax.put("newFileName", FileUtils.getName(fileName));
89             ajax.put("originalFilename", file.getOriginalFilename());
90             return ajax;
91         }
92         catch (Exception e)
93         {
94             return AjaxResult.error(e.getMessage());
95         }
96     }
97
98     /**
99      * 通用上传请求(多个)
100      */
101     @PostMapping("/uploads")
102     public AjaxResult uploadFiles(List<MultipartFile> files) throws Exception
103     {
104         try
105         {
106             // 上传文件路径
107             String filePath = MesConfig.getUploadPath();
108             List<String> urls = new ArrayList<String>();
109             List<String> fileNames = new ArrayList<String>();
110             List<String> newFileNames = new ArrayList<String>();
111             List<String> originalFilenames = new ArrayList<String>();
112             for (MultipartFile file : files)
113             {
114                 // 上传并返回新文件名称
115                 String fileName = FileUploadUtils.upload(filePath, file);
116                 String url = serverConfig.getUrl() + fileName;
117                 urls.add(url);
118                 fileNames.add(fileName);
119                 newFileNames.add(FileUtils.getName(fileName));
120                 originalFilenames.add(file.getOriginalFilename());
121             }
122             AjaxResult ajax = AjaxResult.success();
123             ajax.put("urls", StringUtils.join(urls, FILE_DELIMETER));
124             ajax.put("fileNames", StringUtils.join(fileNames, FILE_DELIMETER));
125             ajax.put("newFileNames", StringUtils.join(newFileNames, FILE_DELIMETER));
126             ajax.put("originalFilenames", StringUtils.join(originalFilenames, FILE_DELIMETER));
127             return ajax;
128         }
129         catch (Exception e)
130         {
131             return AjaxResult.error(e.getMessage());
132         }
133     }
134
135     /**
136      * 本地资源通用下载
137      */
138     @GetMapping("/download/resource")
139     public void resourceDownload(String resource, HttpServletRequest request, HttpServletResponse response)
140             throws Exception
141     {
142         try
143         {
144             if (!FileUtils.checkAllowDownload(resource))
145             {
146                 throw new Exception(StringUtils.format("资源文件({})非法,不允许下载。 ", resource));
147             }
148             // 本地资源路径
149             String localPath = MesConfig.getProfile();
150             // 数据库资源地址
151             String downloadPath = localPath + StringUtils.substringAfter(resource, Constants.RESOURCE_PREFIX);
152             // 下载名称
153             String downloadName = StringUtils.substringAfterLast(downloadPath, "/");
154             response.setContentType(MediaType.APPLICATION_OCTET_STREAM_VALUE);
155             FileUtils.setAttachmentResponseHeader(response, downloadName);
156             FileUtils.writeBytes(downloadPath, response.getOutputStream());
157         }
158         catch (Exception e)
159         {
160             log.error("下载文件失败", e);
161         }
162     }
163 }