From 1fb23cbeeaaeda0a66cf8fe8977c3f5915743cc7 Mon Sep 17 00:00:00 2001 From: fengjin <1435304038@qq.com> Date: 星期五, 21 十月 2022 15:13:10 +0800 Subject: [PATCH] Merge branch 'zigonggao_dev' into huacheng_test --- flower_city/src/main/java/com/dg/core/controller/CommonController.java | 114 +++++++++++++++++++++++++++++++++++++++++++++++++------- 1 files changed, 99 insertions(+), 15 deletions(-) diff --git a/flower_city/src/main/java/com/dg/core/controller/CommonController.java b/flower_city/src/main/java/com/dg/core/controller/CommonController.java index 274a7ac..9c89361 100644 --- a/flower_city/src/main/java/com/dg/core/controller/CommonController.java +++ b/flower_city/src/main/java/com/dg/core/controller/CommonController.java @@ -1,21 +1,35 @@ package com.dg.core.controller; +import cn.hutool.core.date.DateTime; +import cn.hutool.core.date.DateUtil; import com.dg.core.CorsConfig; import com.dg.core.ResultData; import com.dg.core.db.gen.AjaxResult; +import com.dg.core.file.FtpConstants; +import com.dg.core.file.MimeTypeUtils; +import com.dg.core.file.SFTPUtil; +import com.dg.core.util.TestDate; +import com.iceyyy.workday.WorkUtils; import io.swagger.annotations.Api; import io.swagger.annotations.ApiOperation; + import org.apache.commons.io.FilenameUtils; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; +import javax.annotation.Resource; import java.io.InputStream; +import java.text.SimpleDateFormat; import java.util.ArrayList; +import java.util.Date; import java.util.List; import java.util.UUID; + +import static com.dg.core.file.FileTypeUploadUtils.assertAllowed; @Api(tags = {"通用接口"}) @@ -27,6 +41,23 @@ @Autowired private CorsConfig corsConfig; + + + + // FTP 登录用户名 + @Value("${ftp.username}") + private String userName; + // FTP 登录密码 + @Value("${ftp.password}") + private String password; + // FTP 服务器地址IP地址 + @Value("${ftp.host}") + private String host; + // FTP 端口 + @Value("${ftp.port}") + private int port; + @Value("${ftp.url}") + private String url; @RequestMapping("/hello") @@ -45,35 +76,88 @@ { List<String> urls = new ArrayList<String>(); List<String> fileNames = new ArrayList<String>(); - List<String> newFileNames = new ArrayList<String>(); List<String> originalFilenames = new ArrayList<String>(); for (MultipartFile file : files) { String extension = FilenameUtils.getExtension(file.getOriginalFilename()); String name = UUID.randomUUID().toString().replaceAll("-", "") + "." + extension; try { -// assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); -// SFTPUtil sftp = new SFTPUtil(); -// sftp.login(); -// InputStream is = file.getInputStream(); -// sftp.uploadMore(FtpConstants.FTPFILEPATH_IDCARD, name, is); -// sftp.logout(); -// -// urls.add(url + "/idcard/" + name); -// fileNames.add(name); -// newFileNames.add(FileUtils.getName(name)); -// originalFilenames.add(file.getOriginalFilename()); + assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); + SFTPUtil sftp = new SFTPUtil(userName, password, host, port); + sftp.login(); + InputStream is = file.getInputStream(); + sftp.uploadMore(FtpConstants.FTPFILEPATH_IDCARD, name, is); + sftp.logout(); + + urls.add(url + "/idcard/" + name); + fileNames.add(name); + originalFilenames.add(file.getOriginalFilename()); } catch (Exception e) { -// log.error("上传照片失败【{}】", e.getMessage()); - return ResultData.error("上传文件失败"); + return ResultData.error("上传文件失败 "+e.getMessage()); } } AjaxResult ajax = new AjaxResult(); ajax.put("urls", StringUtils.join(urls, FILE_DELIMETER)); ajax.put("fileNames", StringUtils.join(fileNames, FILE_DELIMETER)); - ajax.put("newFileNames", StringUtils.join(newFileNames, FILE_DELIMETER)); ajax.put("originalFilenames", StringUtils.join(originalFilenames, FILE_DELIMETER)); return ResultData.success(ajax); } + + + + /** + * 通用上传请求(多个) + */ + @ApiOperation("文件上传接口(通用)") + @PostMapping("/upload") + public ResultData uploadFile(MultipartFile file) throws Exception + { + List<String> urls = new ArrayList<String>(); + List<String> fileNames = new ArrayList<String>(); + List<String> originalFilenames = new ArrayList<String>(); + + String extension = FilenameUtils.getExtension(file.getOriginalFilename()); + String name = UUID.randomUUID().toString().replaceAll("-", "") + "." + extension; + try { + assertAllowed(file, MimeTypeUtils.DEFAULT_ALLOWED_EXTENSION); + SFTPUtil sftp = new SFTPUtil(userName, password, host, port); + sftp.login(); + InputStream is = file.getInputStream(); + sftp.uploadMore(FtpConstants.FTPFILEPATH_IDCARD, name, is); + sftp.logout(); + +// urls.add(url + "/idcard/" + name); +// fileNames.add(name); +// originalFilenames.add(file.getOriginalFilename()); + return ResultData.success(url + "/idcard/" + name); + } catch (Exception e) { + return ResultData.error("上传文件失败 "+e.getMessage()); + } + +// AjaxResult ajax = new AjaxResult(); +// ajax.put("urls", StringUtils.join(urls, FILE_DELIMETER)); +// ajax.put("fileNames", StringUtils.join(fileNames, FILE_DELIMETER)); +// ajax.put("originalFilenames", StringUtils.join(originalFilenames, FILE_DELIMETER)); + } + + /** + * 判断当前时间是否是工作日,格式为“2022-10-16”,不传默认为今天 + */ + @ApiOperation("判断当前时间是否是工作日,格式为“20221016”,不传默认为今天") + @GetMapping("/isWeekday") + public ResultData isWeekday(@RequestParam(value = "dateTime",required = false) String dateTime) + { + if (dateTime==null){ + Date date = new Date(); + SimpleDateFormat fmt = new SimpleDateFormat("yyyyMMdd"); + dateTime = fmt.format(date); + } + if (WorkUtils.isWorkendDay(dateTime)) + return success("休息日或者周末"); + else + return success("工作日"); + } + + } -- Gitblit v1.7.1