获取文件并下载

方式1: 本地文件

@PostMapping("/importTemplate")
    public ResponseEntity<byte[]> importTemplate() {
        String fileName = "company_template.xlsx";
        String fileUrl = properties.getFile() + fileName;


        try {
            File f = new File(fileUrl);
            InputStream in = Files.newInputStream(f.toPath());

            ByteArrayOutputStream out = new ByteArrayOutputStream();
            byte[] buffer = new byte[1024];
            int bytesRead;

            while ((bytesRead = in.read(buffer)) != -1) {
                out.write(buffer, 0, bytesRead);
            }

            byte[] fileData = out.toByteArray();

            HttpHeaders headers = new HttpHeaders();
            headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
            headers.setContentDispositionFormData("attachment", fileName);
            headers.setContentLength(fileData.length);

            return new ResponseEntity<>(fileData, headers, HttpStatus.OK);
        } catch (IOException e) {
            e.printStackTrace();
            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
        }
    }

方式2:远程文件

 @PostMapping("/importTemplate")
 public ResponseEntity<byte[]> importTemplate() {
     String fileName = "company_template.xlsx";
     String fileUrl = properties.getFile() + fileName;

     try {
         URL url = new URL(fileUrl);
         URLConnection conn = url.openConnection();
         InputStream in = conn.getInputStream();

         ByteArrayOutputStream out = new ByteArrayOutputStream();
         byte[] buffer = new byte[1024];
         int bytesRead;

         while ((bytesRead = in.read(buffer)) != -1) {
             out.write(buffer, 0, bytesRead);
         }

         byte[] fileData = out.toByteArray();

         HttpHeaders headers = new HttpHeaders();
         headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
         headers.setContentDispositionFormData("attachment", fileName);
         headers.setContentLength(fileData.length);

         return new ResponseEntity<>(fileData, headers, HttpStatus.OK);
     } catch (IOException e) {
         e.printStackTrace();
         return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
     }
 }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值