freemarker通过ftl生成模板并使用ZipOutputStream压缩直接下载

freemarker通过ftl生成模板并使用ZipOutputStream压缩直接下载

代码参考如下：

/**
 *<p>Title:SdkMockDeviceController.java</p>
 *<p>Description:数据模拟设备管理 Action</p>
 * @author ZGN
 * @date 2022/1/7 16:07
 */
@Service
public class SdkMockServiceImpl implements SdkMockService {
    private static final Logger log = LoggerFactory.getLogger(SdkMockServiceImpl.class);

    @Autowired
    private IDeviceApi iDeviceApi;

    @Autowired
    private IProductApi iProductApi;

    @Value("${iot.sdkTemplates.templates-path:C:\\Users\\changzheng\\Desktop\\test\\}")
    private String templatesPath;

    @Value("${emq.mqtt.ip:}")
    private String mqttIp;

    @Value("${emq.mqtt.port}")
    private String mqttPort;



    @Override
    public ResultMap<AResultCode, String> sdkMockdeviceDownload(HttpServletResponse response, String device_id) {
        ResultMap<AResultCode,String> result = new ResultMap<>();
        //判断设备id是否为空
        if (StringUtil.isNull(device_id)) {
            return result.setCode(AResultCode.MISS_ID);
        }
        //通过设备id查询设备三元组信息
        ResultMap<AResultCode, DeviceMockBean> DeviceMockBeanResultMap = iDeviceApi.getBasicInfoByMockDevice(device_id);
        if (AResultCode.isFailed(DeviceMockBeanResultMap)) {
            return result.setCode(DeviceMockBeanResultMap.getCode());
        }
        //获取该设备的三元组信息
        DeviceMockBean deviceMockBean = DeviceMockBeanResultMap.getResult();
        //如果设备三元组信息为null
        if (StringUtil.isNull(deviceMockBean)) {
            return result.setCode(AResultCode.IS_NULL);
        }
        //通过产品id获取该产品属性
        ResultMap<AResultCode, List<MockPropertyBean>> MockPropertyBeanResultMap = iProductApi.getPublishedDefaultBlockFunctionPropertyByProductKey(deviceMockBean.getProduct_key());
        if (AResultCode.isFailed(MockPropertyBeanResultMap)) {
            return result.setCode(MockPropertyBeanResultMap.getCode());
        }
        List<MockPropertyBean> MockPropertyBeanList = MockPropertyBeanResultMap.getResult();
        //存放所有属性
        SdkMockDeviceBean sdkMockDeviceBean = new SdkMockDeviceBean();
        //遍历所有属性
        List<String> propertyList = new ArrayList<String>();
        for(int i=0;i<MockPropertyBeanList.size();i++){
            String property_id = MockPropertyBeanList.get(i).getPid();
            propertyList.add(property_id);
        }
        //组装要生成的DeviceConf.json
        sdkMockDeviceBean.setDeviceName(deviceMockBean.getDevice_name());
        sdkMockDeviceBean.setDeviceSecret(deviceMockBean.getDevice_secret());
        sdkMockDeviceBean.setProductKey(deviceMockBean.getProduct_key());
        sdkMockDeviceBean.setPropertyList(propertyList);

        //新建模板生成工具
        FreemarkerUtil freemarkerUtil = new FreemarkerUtil();
        //新建生成的Template的map
        Map<String,Object> deviceConfMap = new HashMap<>();
        deviceConfMap.put("sdkMockDeviceBean",sdkMockDeviceBean);
        try {
            freemarkerUtil.generateFile(templatesPath,"DeviceConf.ftl",templatesPath+"talkweb_iot_mockDevice","DeviceConf.json",deviceConfMap);
        } catch (IOException e) {
            log.error("SdkMockServiceImpl类生成DeviceConf.json文件发生IOException");
            e.printStackTrace();
        } catch (TemplateException e) {
            log.error("SdkMockServiceImpl类生成DeviceConf.json文件发生TemplateException");
            e.printStackTrace();
        }

        //压缩并导出
        String zipName = "talkweb_iot_mockDevice.zip";
        File zipNameFile = new File(templatesPath+"talkweb_iot_mockDevice");
        try {
            response.setHeader("Content-Disposition", "attachment;filename="+ URLEncoder.encode(zipName, "utf-8"));
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        }
        response.setContentType("application/octet-stream;charset=UTF-8");
        OutputStream outputStream = null;
        ZipOutputStream zipOutputStream = null;
        BufferedOutputStream bufferedOutputStream = null;
        try {
            outputStream = response.getOutputStream();
            //压缩输出流
            zipOutputStream = new ZipOutputStream(outputStream);
            //加快缓存流速度
            bufferedOutputStream = new BufferedOutputStream(zipOutputStream);
            //递归压缩，实现文件夹和文件都以压缩
            zip(zipOutputStream,zipNameFile,zipNameFile.getName(),bufferedOutputStream);

        } catch (IOException e) {
            log.error("SdkMockServiceImpl类生成文件输入输出流发生IOException");
            e.printStackTrace();
        }finally {
            try {
                //关闭输出流
                bufferedOutputStream.close();
                //关闭zip输出流
                zipOutputStream.close();
            } catch (IOException e) {
                log.error("SdkMockServiceImpl类关闭文件输入输出流发生IOException");
                e.printStackTrace();
            }
        }

        result.setCode(AResultCode.OK);
        return result;
    }

    //进行zip压缩
    private void zip(ZipOutputStream zipOutputStream,File targetFile,String targetName,BufferedOutputStream bufferedOutputStream){
        FileInputStream fileInputStream = null;
        BufferedInputStream bufferedInputStream = null;
        if(targetFile.isDirectory()){
            //获取到所有文件
            File[] files = targetFile.listFiles();
            //如果文件为空
            if(files.length == 0){
                try {
                    zipOutputStream.putNextEntry(new ZipEntry(targetName+"/"));
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            //遍历文件列表，实现递归压缩
            for (File file : files){
                zip(zipOutputStream,file, targetName+"/"+file.getName(),bufferedOutputStream);
            }
        }else{
            try {
                //创建压缩条目
                zipOutputStream.putNextEntry(new ZipEntry(targetName));
                //读取文件
                fileInputStream = new FileInputStream(targetFile);
                //读取到缓冲流
                bufferedInputStream = new BufferedInputStream(fileInputStream);
                byte[] bytes = new byte[1024];
                int length = -1;
                while((length = bufferedInputStream.read(bytes))!=-1){
                    bufferedOutputStream.write(bytes,0,length);
                }
            } catch (IOException e) {
                log.error("SdkMockServiceImpl类zip压缩方法生成文件输入输出流发生IOException");
                e.printStackTrace();
            }
        }
        try {
            //关闭输出流
            bufferedOutputStream.flush();
            if(bufferedInputStream!=null){
                bufferedInputStream.close();
            }
        } catch (IOException e) {
            log.error("SdkMockServiceImpl类zip压缩方法关闭文件输入输出流发生IOException");
            e.printStackTrace();
        }
    }
}