文件的上传和下载功能

@Slf4j
@RestController
@RequestMapping("/common")
public class CommonController {
    //这里是在application.yml 配置了一个文件路径,然后就可以通过@Value来读取值
    @Value("${reggie.path}") 
    private String basePath;
    @PostMapping("/upload")
    //这里参数路径必须写成MultipratFile file 规定
    public R<String> upload(MultipartFile file){ 
        log.info(file.toString());
        //file是一个临时文件,需要转存到指定位置,否则本次请求完成后临时文件会删除
        // 这个是获取file文件的原始名称的方法
        String originalFilename = file.getOriginalFilename();//abc.jpg   
        //这个是通过截取获得文件后缀
        String suffix = originalFilename.substring(originalFilename.lastIndexOf("."));         
        //使用UUID来重新生成文件名称,防止文件名重复造成的覆盖
        //UUID是一个内置的工具类,可以直接调用
        String fileName = UUID.randomUUID().toString()+suffix; 

        //创建一个目录对象
        File dir =new File(basePath);
        //判断当前目录是否存在
        if(!dir.exists()){
        //目录不存在,则需要创建
            dir.mkdirs();
        }
        //将临时文件转存到指定位置
        try {
         //调方法就完事了
            file.transferTo(new File(basePath+fileName));
        } catch (IOException e) {
            e.getMessage();
        }
        return R.success(fileName);  //最后不要忘了给前端传回文件的文件名,前端要用
    }

    @GetMapping("/download")
    public void download(String name, HttpServletResponse response){ //response参数用来获取输出流
        try {
            //创建一个输入流,通过输入流读取文件内容
            FileInputStream fileInputStream=new FileInputStream(new File(basePath+name));
            //创建一个输出流,通过输出流将文件写回浏览器,在浏览器展示图片
            ServletOutputStream outputStream=response.getOutputStream();
            //设定返回回去的文件类型是什么image/jpeg
            response.setContentType("image/jpeg");
            int len=0;
            //采取byte数组形式读取,每次读1024 byte
            byte[] bytes=new byte[1024];
            while ((len=fileInputStream.read(bytes))!=-1){
                //写入bytes中,从下标0开始
                outputStream.write(bytes,0,len);
                outputStream.flush();
            }
            outputStream.close();
            fileInputStream.close();
        } catch (Exception e) {
            e.getMessage();
        }
    }
}


    }
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值