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文件上传下载实战项目 完成了。