溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊(cè)×
其他方式登錄
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》

Spring框架實(shí)現(xiàn)文件上傳功能

發(fā)布時(shí)間:2020-10-03 09:44:17 來(lái)源:腳本之家 閱讀:154 作者:chris_mao 欄目:編程語(yǔ)言

在Java中實(shí)現(xiàn)文件的上傳有多種方式,如smartUpload或是使用Strus2,本文與大家分享使用Spring框架中的MultipartFile類(lèi)來(lái)實(shí)例文件的上傳。

不啰嗦了,直接上干貨。先是編寫(xiě)了一個(gè)實(shí)現(xiàn)文件上傳的類(lèi)FileUploadingUtil,此類(lèi)中定義了兩個(gè)對(duì)外公開(kāi)的方法,upload和getFileMap。

前者需要傳入一個(gè)Map參數(shù),是用戶提交的表單中的文件列表,最終返回值的也是一個(gè)Map類(lèi)型對(duì)象,其鍵名為上傳的文件名稱(chēng),鍵值為文件在服務(wù)器上的存儲(chǔ)路徑;后者主要是用于測(cè)試用途,非主要功能,看官可以忽略此方法。

package com.emerson.cwms.utils; 
 
import java.io.File; 
import java.io.FileInputStream; 
import java.io.FileNotFoundException; 
import java.io.FileOutputStream; 
import java.io.IOException; 
import java.io.InputStream; 
import java.io.OutputStream; 
import java.util.Date; 
import java.util.HashMap; 
import java.util.Iterator; 
import java.util.Map; 
import java.util.Map.Entry; 
 
import org.springframework.web.multipart.MultipartFile; 
 
/** 
 * 文件上傳工具類(lèi) 
 * 
 * @author Chris Mao(Zibing) 
 * 
 */ 
public class FileUploadingUtil { 
 
  /** 
   * 服務(wù)器上的保存路徑,在使用到上傳功能的Controller中對(duì)其進(jìn)行賦值 
   */ 
  public static String FILEDIR = null; 
 
  /** 
   * 上傳多個(gè)文件,返回文件名稱(chēng)和服務(wù)器存儲(chǔ)路徑列表 
   * 
   * @param files 
   * @return 
   * @throws IOException 
   */ 
  public static Map<String, String> upload(Map<String, MultipartFile> files) throws IOException { 
    File file = new File(FILEDIR); 
    if (!file.exists()) { 
      file.mkdir(); 
    } 
 
    Map<String, String> result = new HashMap<String, String>(); 
    Iterator<Entry<String, MultipartFile>> iter = files.entrySet().iterator(); 
    while (iter.hasNext()) { 
      MultipartFile aFile = iter.next().getValue(); 
      if (aFile.getSize() != 0 && !"".equals(aFile.getName())) { 
        result.put(aFile.getOriginalFilename(), uploadFile(aFile)); 
      } 
    } 
    return result; 
  } 
 
  /** 
   * 上傳單個(gè)文件,并返回其在服務(wù)器中的存儲(chǔ)路徑 
   * 
   * @param aFile 
   * @return 
   * @throws FileNotFoundException 
   * @throws IOException 
   */ 
  private static String uploadFile(MultipartFile aFile) throws IOException { 
    String filePath = initFilePath(aFile.getOriginalFilename()); 
    try { 
      write(aFile.getInputStream(), new FileOutputStream(filePath)); 
    } catch (FileNotFoundException e) { 
      logger.error("上傳的文件: " + aFile.getName() + " 不存在??!"); 
      e.printStackTrace(); 
    } 
    return filePath; 
  } 
 
  /** 
   * 寫(xiě)入數(shù)據(jù) 
   * 
   * @param in 
   * @param out 
   * @throws IOException 
   */ 
  private static void write(InputStream in, OutputStream out) throws IOException { 
    try { 
      byte[] buffer = new byte[1024]; 
      int bytesRead = -1; 
      while ((bytesRead = in.read(buffer)) != -1) { 
        out.write(buffer, 0, bytesRead); 
      } 
      out.flush(); 
    } finally { 
      try { 
        in.close(); 
        out.close(); 
      } catch (IOException ex) { 
      } 
    } 
  } 
 
  /** 
   * 遍歷服務(wù)器目錄,列舉出目錄中的所有文件(含子目錄) 
   * @return 
   */ 
  public static Map<String, String> getFileMap() { 
    logger.info(FileUploadingUtil.FILEDIR); 
    Map<String, String> map = new HashMap<String, String>(); 
    File[] files = new File(FileUploadingUtil.FILEDIR).listFiles(); 
    if (files != null) { 
      for (File file : files) { 
        if (file.isDirectory()) { 
          File[] files2 = file.listFiles(); 
          if (files2 != null) { 
            for (File file2 : files2) { 
              String name = file2.getName(); 
              logger.info(file2.getParentFile().getAbsolutePath()); 
              logger.info(file2.getAbsolutePath()); 
              map.put(file2.getParentFile().getName() + "/" + name, 
                  name.substring(name.lastIndexOf("_") + 1)); 
            } 
          } 
        } 
      } 
    } 
    return map; 
  } 
 
  /** 
   * 返回文件存儲(chǔ)路徑,為防止重名文件被覆蓋,在文件名稱(chēng)中增加了隨機(jī)數(shù) 
   * @param name 
   * @return 
   */ 
  private static String initFilePath(String name) { 
    String dir = getFileDir(name) + ""; 
    File file = new File(FILEDIR + dir); 
    if (!file.exists()) { 
      file.mkdir(); 
    } 
    Long num = new Date().getTime(); 
    Double d = Math.random() * num; 
    return (file.getPath() + "/" + num + d.longValue() + "_" + name).replaceAll(" ", "-"); 
  } 
 
  /** 
   * 
   * @param name 
   * @return 
   */ 
  private static int getFileDir(String name) { 
    return name.hashCode() & 0xf; 
  } 
} 

Controller代碼,使用上述定義的FileUploadingUtil類(lèi)實(shí)現(xiàn)文件上傳之功能。

package com.emerson.cwms.web;  


import java.io.IOException; 
import java.util.Iterator; 
import java.util.Map; 
import java.util.Map.Entry; 
 
import javax.servlet.http.HttpServletRequest; 
import javax.servlet.http.HttpServletResponse; 
 
import org.springframework.stereotype.Controller; 
import org.springframework.ui.Model; 
import org.springframework.web.bind.annotation.RequestMapping; 
import org.springframework.web.bind.annotation.RequestMethod; 
import org.springframework.web.multipart.MultipartHttpServletRequest; 
 
import com.emerson.cwms.utils.FileUploadingUtil; 
 
/** 
 * 文件上傳控制器 
 * 
 * @author Chris Mao(Zibing) 
 * 
 */ 
@Controller 
@RequestMapping(value = "/files") 
public class FileController { 
   
  @RequestMapping(value = "/", method = RequestMethod.GET) 
  public String list(HttpServletRequest request, HttpServletResponse response, Model model) { 
    iniFileDir(request); 
     
    System.out.println(request.getAttribute("files")); 
    model.addAttribute("files", FileUploadingUtil.getFileMap()); 
    return "files/list"; 
  } 
   
  @RequestMapping(value = "/upload", method = RequestMethod.POST) 
  public String doUpload(HttpServletRequest request) { 
    iniFileDir(request); 
     
    try { 
      MultipartHttpServletRequest mRequest = (MultipartHttpServletRequest) request; 
      Map<String, String> uploadedFiles = FileUploadingUtil.upload(mRequest.getFileMap()); 
 
      Iterator<Entry<String, String>> iter = uploadedFiles.entrySet().iterator(); 
      while (iter.hasNext()) { 
        Entry<String, String> each = iter.next(); 
        System.out.print("Uploaded File Name = " + each.getKey()); 
        System.out.println(", Saved Path in Server = " + each.getValue()); 
      } 
    } catch (Exception e) { 
      e.printStackTrace(); 
    } 
    return "redirect:/files/"; 
  } 
 
  private void iniFileDir(HttpServletRequest request) { 
    FileUploadingUtil.FILEDIR = request.getSession().getServletContext().getRealPath("/") + "files/"; 
    if (FileUploadingUtil.FILEDIR == null) { 
      FileUploadingUtil.FILEDIR = request.getSession().getServletContext().getRealPath("/") + "files/"; 
    } 
  } 
} 

注意事項(xiàng):一定要在pom.xml文件中添加對(duì)commons-fileupload的依賴引用,否則代碼在運(yùn)行時(shí)會(huì)提示找不到FileItemFactory類(lèi)的錯(cuò)誤。

依賴引用如下。

<dependency> 
  <groupId>commons-fileupload</groupId> 
  <artifactId>commons-fileupload</artifactId> 
  <version>1.3.1</version> 
</dependency> 

以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。

AI