從零搭建本身的SpringBoot後臺框架(十六)

時間 2019-11-17

原文原文鏈接

Hello你們好，本章咱們添加多文件上傳功能。有問題能夠聯繫我mr_beany@163.com。另求各路大神指點，感謝

在一個系統中，文件上傳模塊確定是少不了的，例如頭像上傳，展現輪播圖等等，因此本章咱們來添加上傳文件功能java

一：添加commons-fileupload依賴

打開pom文件添加git

<dependency>
   <groupId>commons-fileupload</groupId>
   <artifactId>commons-fileupload</artifactId>
   <version>1.3.1</version>
</dependency>複製代碼

二：添加系統變量

打開core→constant文件，添加文件保存路徑github

//文件上傳儲存的地址
public static final String SAVEFILEPATH = "F://img";複製代碼

三：添加文件上傳限制

建立core→configurer→MultipartConfigurer.javaweb

package com.example.demo.core.configurer;

import org.springframework.boot.web.servlet.MultipartConfigFactory;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.servlet.MultipartConfigElement;

@Configuration
public class MultipartConfigurer {

    @Bean
    public MultipartConfigElement multipartConfigElement(){
        MultipartConfigFactory factory=new MultipartConfigFactory();
        factory.setMaxFileSize("10MB");
        factory.setMaxRequestSize("10MB");
        return factory.createMultipartConfig();
    }
}複製代碼

四：建立文件上傳工具類

建立core→utils→UploadActionUtil.javaspring

package com.example.demo.core.utils;

import com.example.demo.core.constant.ProjectConstant;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.text.SimpleDateFormat;
import java.util.*;

/**
 * @author 張瑤
 * 文件上傳控制器
 */
public class UploadActionUtil {

   public static List<String> uploadFile(HttpServletRequest request) throws Exception {
      List<String> list = new ArrayList<>();
      CommonsMultipartResolver multipartResolver = new CommonsMultipartResolver(
            request.getSession().getServletContext());
      if (multipartResolver.isMultipart(request)) {
         MultipartHttpServletRequest multiRequest = (MultipartHttpServletRequest) request;
         Iterator<String> iterator = multiRequest.getFileNames();
         while (iterator.hasNext()) {
            // 取得上傳文件
            MultipartFile file = multiRequest.getFile(iterator.next());
            if (file != null) {
               // 取得當前上傳文件的文件名稱
               String myFileName = file.getOriginalFilename();
               // 若是名稱不爲「」,說明該文件存在，不然說明該文件不存在
               if (myFileName.trim() != "") {
                  String fileTyps = myFileName.substring(myFileName.lastIndexOf("."));
                  // String tempName="demo"+fileTyps;
                  String tempName = UUID.randomUUID().toString() + fileTyps;
                  // 建立文件夾
                  String folderPath = ProjectConstant.SAVEFILEPATH + File.separator + folderName();
                  File fileFolder = new File(folderPath);
                  if (!fileFolder.exists() && !fileFolder.isDirectory()) {
                     fileFolder.mkdir();
                  }
                  File uploadFile = new File(folderPath + File.separator + tempName);
                  file.transferTo(uploadFile);
                  myFileName = folderName() + File.separator + tempName;
                  list.add(ProjectConstant.SAVEFILEPATH + "//" + myFileName);
               }
            }
         }
      }
      return list;
   }

   /**
    * 得年月日的文件夾名稱
    * 
    * @return
    */
   public static String getCurrentFilderName()  throws Exception{
      Calendar now = Calendar.getInstance();
      return now.get(Calendar.YEAR) + "" + (now.get(Calendar.MONTH) + 1) + "" + now.get(Calendar.DAY_OF_MONTH);
   }

   /**
    * 建立文件夾
    * 
    * @param filderName
    */
   public static void createFilder(String filderName) throws Exception {
      File file = new File(filderName);
      // 若是文件夾不存在則建立
      if (!file.exists() && !file.isDirectory()) {
         file.mkdirs();
      }
   }

   /**
    * 文件擴展名
    * 
    * @param fileName
    * @return
    */
   public static String extFile(String fileName)  throws Exception{
      return fileName.substring(fileName.lastIndexOf("."));
   }

   /**
    * 當前日期當文件夾名
    * 
    * @return
    */
   public static String folderName() throws Exception {
      SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
      String str = sdf.format(new Date());
      return str;
   }
}複製代碼

五：建立UploadFileController

package com.example.demo.controller;

import com.example.demo.core.ret.RetResponse;
import com.example.demo.core.ret.RetResult;
import com.example.demo.core.utils.UploadActionUtil;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.List;

@RestController
@RequestMapping("/uploadFile")
public class UploadFileController {

    @PostMapping("/upload")
    public RetResult<List<String>> upload(HttpServletRequest httpServletRequest) throws Exception {
        List<String> list = UploadActionUtil.uploadFile(httpServletRequest);
        return RetResponse.makeOKRsp(list);
    }
}複製代碼