后端下載圖片或者文件的方法

直接貼代碼吧

package com.wjs.controller;

import lombok.extern.slf4j.Slf4j;
import org.activiti.engine.RepositoryService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;

/**
 * 獲取流程圖
 */
@Slf4j
@RequestMapping("/image")
@RestController
public class ImageController {
    @Autowired
    private RepositoryService repositoryService;

    @RequestMapping(value = "/getImg",method = RequestMethod.GET)
    public void getImg(@RequestParam String deploymentId, @RequestParam String resourceName, HttpServletRequest request, HttpServletResponse response){
        log.info("deploymentId:{},resourceName:{}",deploymentId,resourceName);

        response.setCharacterEncoding(request.getCharacterEncoding());
        response.setContentType("application/octet-stream");
        InputStream is = null;
        is = repositoryService.getResourceAsStream(deploymentId, resourceName);
        response.setHeader("Content-Disposition","attachment; filename=" + URLEncoder.encode("流程圖.png"));
        OutputStream os = null;
        try {
            // 這兩行代碼可以代替42-48行的代碼。
//            IOUtils.copy(is,response.getOutputStream());
//            response.flushBuffer();
            os = response.getOutputStream();
            byte[] b = new byte[1024];
            int len;
            while ((len = is.read(b)) != -1) {
                os.write(b,0,len);
            }
            response.flushBuffer();
        } catch (IOException e) {
            e.printStackTrace();
            log.error("圖片下載失敗....");
        }finally {
            if (null != is) {
                try {
                    is.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

友情鏈接更多精彩內(nèi)容