springboot业务功能实战（十六）文件服务器的上传下载接口

上传

这里有一个实体 用于存放文件信息 存放路径commonfile在配置文件中配置

      @RestController
      public class FileUpload {
         @Value("${commonfile}")
         private String commonfile;
         /**
       * 使用Spring MVC的multipartFile 类作为参数
       *
       * @param multipartFile
       * @return
       */
         @PostMapping("/upload")
         @ResponseBody
         @ArchivesLog(operationName = "上传")
         public Result<FileMessage> uploadMultipartFile(@RequestParam("file") MultipartFile multipartFile) {
              Result<FileMessage> result = new Result<FileMessage>();
              FileMessage fileMessageVO = new FileMessage();
             String fileName = multipartFile.getOriginalFilename();
             try {
                 // 获取文件字节数组
                  byte[] bytes = multipartFile.getBytes();
                 // 文件存储路径(/fileupload1/ 这样会在根目录下创建问价夹)
                  File pfile = new File(commonfile);
                 // 判断文件夹是否存在
                 if (!pfile.exists()) {
                     // 不存在时,创建文件夹
                      pfile.mkdirs();
                  }
                 // 创建文件
                  File file = new File(pfile, fileName);
                 // 写入指定文件夹
                  OutputStream out = new FileOutputStream(file);
                  out.write(bytes);
                 // 组装对象
                  fileMessageVO.setFileName(fileName);
                  fileMessageVO.setFileRoute(commonfile);
                  fileMessageVO.setFileDownloadUri(commonfile + fileName);
                  fileMessageVO.setSize(getPrintSize(file.length()));
                  fileMessageVO.setFileType(fileName.substring(fileName.lastIndexOf(".") + 1));
                  result.setData(fileMessageVO);
              } catch (Exception e) {
                  e.printStackTrace();
              }
             return result;
          }
         public static String getPrintSize(long size) {
             // 如果字节数少于1024，则直接以B为单位，否则先除于1024，后3位因太少无意义
             if (size < 1024) {
                 return String.valueOf(size) + "B";
              } else {
                  size = size / 1024;
              }
             // 如果原字节数除于1024之后，少于1024，则可以直接以KB作为单位
             // 因为还没有到达要使用另一个单位的时候
             // 接下去以此类推
             if (size < 1024) {
                 return String.valueOf(size) + "KB";
              } else {
                  size = size / 1024;
              }
             if (size < 1024) {
                 // 因为如果以MB为单位的话，要保留最后1位小数，
                 // 因此，把此数乘以100之后再取余
                  size = size * 100;
                 return String.valueOf((size / 100)) + "." + String.valueOf((size % 100)) + "MB";
              } else {
                 // 否则如果要以GB为单位的，先除于1024再作同样的处理
                  size = size * 100 / 1024;
                 return String.valueOf((size / 100)) + "." + String.valueOf((size % 100)) + "GB";
              }
          }
      }
  
 

实体类

      public class FileMessage {
         /**
       * 文件名称
       */
         private String fileName;
         /**
       * 文件路径
       */
         private String fileRoute;
         /**
       * 下载地址
       */
         private String fileDownloadUri;
         /**
       * 文件类型
       */
         private String fileType;
         /**
       * 文件大小
       */
         private String size;
      }
  
 

下载

      @RestController
      public class FileDownload {
          @Value("${commonfile}")
         private String commonfile;
          @GetMapping("/download")
         public String downloadFile(FileMessage fileMessage) throws UnsupportedEncodingException {
             // 获取HttpServletResponse
              HttpServletResponse response =
                  ((ServletRequestAttributes)RequestContextHolder.getRequestAttributes()).getResponse();
             String fileName = fileMessage.getFileName();// 文件名
             if (fileName != null) {
                 // 设置文件路径
                  File file = new File(commonfile + fileName);
                 if (file.exists()) {
                      RequestAttributes requsetAttributes = RequestContextHolder.currentRequestAttributes();
                      HttpServletRequest request = ((ServletRequestAttributes)requsetAttributes).getRequest();
                     // 获得浏览器代理信息
                     final String userAgent = request.getHeader("USER-AGENT");
                     // 判断浏览器代理并分别设置响应给浏览器的编码格式
                     if (StringUtils.contains(userAgent, "MSIE") || StringUtils.contains(userAgent, "Trident")) {
                         // IE浏览器
                          fileName = URLEncoder.encode(fileName, "UTF-8");
                      } else if (StringUtils.contains(userAgent, "Mozilla")) {
                         // google,火狐浏览器
                          fileName = new String(fileName.getBytes("UTF-8"), "ISO8859-1");
                      } else {
                         // 其他浏览器
                          fileName = URLEncoder.encode(fileName, "UTF-8");// 其他浏览器
                      }
                     // 设置HTTP响应头
                      response.reset();
                     // 重置 如果不在页面上显示而是下载下来 则放开注释
                     // response.setContentType("application/octet-stream");// 告知浏览器下载文件，而不是直接打开，浏览器默认为打开
                     // response.addHeader("Content-Disposition", "attachment;filename=\"" + fileName + "\"");// 下载文件的名称
                     try {
                          OutputStream os = response.getOutputStream();
                         // 读取文件
                          InputStream in = new FileInputStream(file);
                         // copy文件
                          IOUtils.copy(in, os);
                          in.close();
                          os.close();
                         return "下载成功";
                      } catch (Exception e) {
                          e.printStackTrace();
                      }
                  }
              }
             return "下载失败";
          }
      }
  
 

文章来源: baocl.blog.csdn.net，作者：小黄鸡1992，版权归原作者所有，如需转载，请联系作者。

原文链接：baocl.blog.csdn.net/article/details/106129270