代码:文件上传下载,服务端controller

本文介绍了如何在SpringMVC应用中,使用Lombok和MultipartFile处理文件上传和下载,包括文件名处理、路径配置以及异常处理。通过`@RestController`和`@RequestMapping`注解定义了上传和下载的接口。
摘要由CSDN通过智能技术生成

1.application.yml

fileSavePath:
  path: D:\文件

2.CommonController

//import R对象所在包 
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.UUID;

/**
 * 文件上传和下载
 */
@RestController
@RequestMapping("/common")  //路由
@Slf4j  //记录日志
public class CommonController {

    @Value("$fileSavePath.path")  //注入配置文件
    private String basePath;

    /**
     * 文件上传
     * @param file
     * @return
     */
    @PostMapping("/upload")
    public R<String> upload(MultipartFile file){

        log.info(file.toString());

        //获取原始文件名
        String originalFilename = file.getOriginalFilename();
        //截取文件名后缀:.jpg、.png ……
        String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));

        //使用UUID重新生成文件名,防止文件名称重复造成文件覆盖
        String newFileName = UUID.randomUUID().toString() + suffix;

        //创建一个文件目录对象
        File dir = new File(basePath);

        //判断当前文件目录是否存在
        if (!dir.exists()){
            //如果文件夹不存在,创建一个新的文件目录
            dir.mkdirs();
        }

        try {
            //将临时文件转存到另外一个位置
            //转存的路径不写死,通过配置文件指定,实现动态修改存储位置----》application.yml
            file.transferTo(new File(basePath + newFileName));
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        return R.success(newFileName);
    }

    /**
     * 文件下载
     * @param name 文件名
     * @param response
     */
    @GetMapping("/download")
    public void download(String name, HttpServletResponse response){    //输出流需要通过response获得

        try {
            //输入流,通过输入流读取文件内容
            FileInputStream fileInputStream = new FileInputStream(new File(basePath + name));
            //输出流,通过输出流将文件写会浏览器,在浏览器展示图片
            ServletOutputStream outputStream = response.getOutputStream();

            //指定相应文件类型文图片
            response.setContentType("image/jpeg");

            //每次读取的长度
            int len = 0;
            byte[] bytes = new byte[1024];
            while ( (len = fileInputStream.read(bytes)) != -1 ){
                //写数据
                outputStream.write(bytes,0,len);
                //刷新流
                outputStream.flush();
            }
            //关闭资源
            outputStream.close();
            fileInputStream.close();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }

    }
}

3.R对象

import lombok.Data;
import java.util.HashMap;
import java.util.Map;

/**
 * 通用返回结果类,服务端响应的数据最终都会封装成此对象
 * @param <T>
 */
@Data
public class R<T> {

    private Integer code; //编码:1成功,0和其它数字为失败

    private String msg; //错误信息

    private T data; //数据

    private Map map = new HashMap(); //动态数据

    public static <T> R<T> success(T object) {
        R<T> r = new R<T>();
        r.data = object;
        r.code = 1;
        return r;
    }

    public static <T> R<T> error(String msg) {
        R r = new R();
        r.msg = msg;
        r.code = 0;
        return r;
    }

    public R<T> add(String key, Object value) {
        this.map.put(key, value);
        return this;
    }

}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值