Jcdm042-Mes.git

懒羊羊

2024-01-31 e57a8990ae56f657a59c435a0613c5f7a8728003

提交 \| 用户 \| 时间
e57a89	1	package com.jcdm.web.controller.common;
懒	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	}