springboot文件上傳下載實戰 ——文件上傳、下載、在線打開、刪除

GitHub：https://github.com/szluyu99/springboot_files

登錄功能與展示所有文件請看這個：springboot文件上傳下載實戰 —— 登錄功能、展示所有文件

文件上傳

com.yusael.dao 包下：UserFileDAO.java 接口中增加一個方法：

// 保存用戶的文件記錄
void save(UserFile userFile);

resources/com/yusael/mapper 目錄下：UserFileDAOMapper.xml 中增加代碼：

<!--保存用戶的文件記錄-->
<insert id="save" parameterType="UserFile">
    insert into t_files
    values(#{id}, #{oldFileName}, #{newFileName},
        #{ext}, #{path}, #{size}, #{type}, #{isImg},
        #{downcounts}, #{uploadTime}, #{userId})
</insert>

---

com.yusael.service 包下：UserFileService.java 接口中增加一個方法：

void save(UserFile userFile);

com.yusael.service 包下：UserFileServiceImpl.java 實現類中增加代碼：

@Override
public void save(UserFile userFile) {
    // userFile.setIsImg(); // 是否是圖片: 解決方案: 當類型中含有image時說明當前類型一定爲圖片類型
    String isImage = userFile.getType().startsWith("image") ? "是" : "否";
    userFile.setIsImg(isImage);
    userFile.setDowncounts(0);
    userFile.setUploadTime(new Date());
    userFileDAO.save(userFile);
}

---

文件上傳核心 UserFileController

com.yusael.controller 包下：UserFileController.java 中增加方法：

@PostMapping("/upload")
public String upload(MultipartFile aaa, HttpSession session) throws IOException {
    // 獲取上傳用戶的id
    User user = (User) session.getAttribute("user");

    // 獲取文件的原始名稱
    String oldFileName = aaa.getOriginalFilename();
	// 獲取文件的拓展名
    String extension = "." + FilenameUtils.getExtension(oldFileName);
    // 生成新的文件名稱
    String newFileName = new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) +
            UUID.randomUUID().toString().replace("-", "") + extension;
    // 獲取文件的大小
    long size = aaa.getSize();
    // 獲取文件類型
    String type = aaa.getContentType();

    // 根據日期生成文件目錄
    // ResourceUtils.getURL("classpath:") 是獲取當前文件的resources路徑
    String realPath = ResourceUtils.getURL("classpath:").getPath() + "/static/files";
    // 日期文件夾
    String date = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
    String dateDirPath = realPath + "/" + date;
    File dateDir = new File(dateDirPath);
    if (!dateDir.exists()) dateDir.mkdirs();

    // 處理文件上傳
    aaa.transferTo(new File(dateDir, newFileName));

    // 將文件信息放入數據庫
    UserFile userFile = new UserFile();
    userFile.setOldFileName(oldFileName).setNewFileName(newFileName).
            setExt(extension).setSize(String.valueOf(size)).setType(type).
            setPath("/files/" + date).setUserId(user.getId());
    System.out.println(userFile);
    userFileService.save(userFile);

    return "redirect:/file/showAll";
}

文件上傳測試

之前我們完成了展示所有文件功能，但是由於數據庫中沒有數據，因此顯示的頁面爲空，現在我們利用文件上傳功能往數據庫中添加數據。

點擊 選擇文件，選擇電腦上的任意文件，然後點擊 上傳文件。

點擊上傳文件後頁面自動再次跳到顯示所有頁面，此時可以看到頁面上已經有數據了。

文件下載與在線打開

com.yusael.dao 包下：UserFileDAO.java 接口中增加一個方法：

// 根據id找到文件信息
UserFile findById(Integer id);
// 更新下載次數
void update(UserFile userFile);

resources/com/yusael/mapper 目錄下：UserFileDAOMapper.xml 中增加代碼：

<!--根據id找到文件信息-->
<select id="findById" parameterType="Integer" resultType="UserFile">
    select id,oldFileName,newFileName,ext,path,size,type,isImg,downcounts,uploadTime,userId
    from t_files
    where id = #{id}
</select>

<!--更新文件下載次數-->
<update id="update" parameterType="UserFile">
    update t_files set downcounts = #{downcounts} where id = #{id}
</update>

---

com.yusael.service 包下：UserFileService.java 接口中增加方法：

UserFile findById(Integer id);
void update(UserFile userFile);

com.yusael.service 包下：UserFileServiceImpl.java 實現類中增加代碼：

@Override
public UserFile findById(Integer id) {
    return userFileDAO.findById(id);
}

@Override
public void update(UserFile userFile) {
    userFileDAO.update(userFile);
}

---

文件下載、在線打開核心 UserFileController

com.yusael.controller 包下：UserFileController.java 中增加方法：

// 文件下載
@GetMapping("/download")
public void download(Integer id, String openStyle, HttpServletResponse response) throws IOException {
    // attachement是以附件形式下載, inline是在線打開
    openStyle = "inline".equals(openStyle) ? "inline" : "attachment";
    // 獲取文件信息
    UserFile userFile = userFileService.findById(id);
    if ("attachment".equals(openStyle)) {
        // 更新下載次數
        userFile.setDowncounts(userFile.getDowncounts() + 1); // 下載次數+1
        userFileService.update(userFile);
    }
    // 根據 文件信息中文件名字 和 文件存儲路徑 獲取文件真實路徑
    String realPath = ResourceUtils.getURL("classpath:").getPath() + "/static" + userFile.getPath();
    // 獲取文件輸入流
    InputStream is = new FileInputStream(new File(realPath, userFile.getNewFileName()));
    // 獲取響應輸出流
    response.setHeader("content-disposition", openStyle + ";fileName=" + URLEncoder.encode(userFile.getOldFileName(), "UTF-8"));
    ServletOutputStream os = response.getOutputStream();
    // 文件拷貝
    IOUtils.copy(is, os);
    IOUtils.closeQuietly(is);
    IOUtils.closeQuietly(os);
}

文件下載、在線打開測試

前面我們已經完成了文件的上傳功能，經過上傳了圖片，現在數據庫中已經有了幾張圖片信息，我們點擊下載，可以跳出下載界面，並且文件的下載次數會 + 1。

點擊在線打開，如果是圖片格式的文件，則會直接打開，否則依舊是以附件的形式下載，但是不增加下次次數。

文件刪除

com.yusael.dao 包下：UserFileDAO.java 接口中增加一個方法：

// 刪除數據庫中的文件
void delete(Integer id);

resources/com/yusael/mapper 目錄下：UserFileDAOMapper.xml 中增加代碼：

<!--刪除數據庫中的文件-->
<delete id="delete" parameterType="Integer">
    delete from t_files
    where id = #{id}
</delete>

---

com.yusael.service 包下：UserFileService.java 接口中增加方法：

void delete(Integer id);

com.yusael.service 包下：UserFileServiceImpl.java 實現類中增加代碼：

@Override
public void delete(Integer id) {
    userFileDAO.delete(id);
}

---

com.yusae.controller 包下：UserFileController.java 中增加方法：

// 文件刪除
@GetMapping("/delete")
public String delete(Integer id) throws FileNotFoundException {
    // 根據id查詢信息
    UserFile userFile = userFileService.findById(id);
    // 刪除服務器上的文件
    // 通過相對路徑獲取絕對路徑
    String realPath = ResourceUtils.getURL("classpath:").getPath() + "/static" + userFile.getPath();
    File file = new File(realPath + "/" + userFile.getNewFileName());
    if (file.exists()) file.delete(); // 如果服務器上文件存在則刪除文件

    // 刪除數據庫中的文件
    userFileService.delete(id);

    return "redirect:/file/showAll";
}

文件刪除測試

點擊刪除按鈕即可刪除文件。

至此，整個 springboot文件上傳下載實戰項目 完成了。