模板下载

package com.cango.erp.cs.controller;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.InputStream;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Map;

@RestController
@RequestMapping("/templateFile")
@Slf4j
public class TemplateFileController {

    @Autowired
    ResourceLoader resourceLoader;

    /**
     * 模板名称Map key: 模板ID  value：模板名称
     */
    private static Map<String, String> templateName = new HashMap<>();

    static {
        // 外包案件分配导入模板
        templateName.put("1", "外包案件分配导入模板.xlsx");
    }

    @RequestMapping(value = "/download.do", method = RequestMethod.GET)
    public void declareInfoExport(HttpServletRequest request, HttpServletResponse response) throws Exception {
        final String templateId = request.getParameter("templateId");
        if (!templateName.containsKey(templateId)) {
            throw new RuntimeException("模板不存在");
        }
        // 文件名
        String filename = templateName.get(templateId);
        Resource resource = resourceLoader.getResource("classpath:/template/download/" + filename);
        try (InputStream inputStream = resource.getInputStream()) {
            response.setContentType("application/force-download");
            //保证下载时文件名为指定文件名，避免中文文件名乱码
            response.addHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode(filename, "utf-8"));
            ServletOutputStream outputStream = response.getOutputStream();
            byte[] b = new byte[2048];
            int len;
            //从输入流中读取一定数量的字节，并将其存储在缓冲区字节数组中，读到末尾返回-1
            while ((len = inputStream.read(b)) > 0) {
                outputStream.write(b, 0, len);
            }
        } catch (Exception e) {
            log.error("【CCS--文件下载】异常:{}", e);
            throw new RuntimeException("文件下载失败");
        }
    }
}