package com.itl.project.common.attachment.domain;

import com.baomidou.mybatisplus.annotation.TableField;
import com.fasterxml.jackson.annotation.JsonFormat;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import org.apache.commons.lang3.builder.ToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;
import com.itl.framework.aspectj.lang.annotation.Excel;
import com.itl.framework.web.domain.BaseEntity;

import java.util.Date;

/**
 * 文件信息對(duì)象 sys_attachment
 *
 * @author itl
 */
@Data
public class Attachment extends BaseEntity
{
    private static final long serialVersionUID = 1L;

    /** 文件主鍵 */
    private Long fileId;

    /** 文件業(yè)務(wù)id */
    @Excel(name = "文件業(yè)務(wù)id")
    @TableField(exist = false)
    private String fileBusinessId;

    /** 文件key */
    @Excel(name = "文件key")
    private String fileKey;

    /** 文件名稱 */
    @Excel(name = "文件名稱")
    private String fileName;

    /** mime類型 */
    @Excel(name = "mime類型")
    private String mimeType;

    /** 文件大小 */
    @Excel(name = "文件大小")
    private Long fileSize;

    /** 文件路徑 */
    @Excel(name = "文件路徑")
    private String filePath;

    /** 文件地址 */
    @Excel(name = "文件地址")
    private String fileUrl;

    /** 文件后綴 */
    @Excel(name = "文件后綴")
    private String suffix;

    /** 是否存在 */
	private boolean exists = false;

    /** 創(chuàng)建時(shí)間 */
    @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
    private Date createTime;


    /**
     * 版本號(hào)
     */
    private String version;


    /**
     * 文件備注
     */
    private String comment;

    @TableField(exist = false)
    private String fileIds;

}

package com.itl.common.utils;

import com.itl.project.common.attachment.domain.Attachment;
import java.io.BufferedOutputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.List;
import java.util.Objects;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * 文件壓縮下載工具類
 * 功能：將多個(gè)文件壓縮到指定輸出流中，用于ZIP包下載
 */
public class CompressDownloadUtil {

    /**
     * 將多個(gè)文件壓縮到指定輸出流中
     *
     * @param lists 需要壓縮的文件列表
     * @param outputStream  壓縮后的輸出流（直接關(guān)聯(lián)HTTP響應(yīng)輸出流）
     */
    public static void compressZip(List<Attachment> lists, OutputStream outputStream) {
        ZipOutputStream zipOutStream = null;
        try {
            // 包裝成ZIP格式輸出流，提升寫入效率
            zipOutStream = new ZipOutputStream(new BufferedOutputStream(outputStream));
            // 設(shè)置壓縮方法為DEFLATED（默認(rèn)，有壓縮效果）
            zipOutStream.setMethod(ZipOutputStream.DEFLATED);
            
            // 循環(huán)處理每個(gè)文件
            for (Attachment file : lists) {
                java.io.File localFile = new java.io.File(file.getFilePath());
                // 校驗(yàn)文件是否存在，避免空指針或文件找不到異常
                if (localFile.exists() && localFile.isFile()) {
                    try (FileInputStream fileInputStream = new FileInputStream(localFile)) {
                        // 讀取文件字節(jié)數(shù)據(jù)
                        byte[] data = new byte[(int) localFile.length()];
                        fileInputStream.read(data);
                        
                        // 創(chuàng)建ZIP條目（壓縮包內(nèi)的文件名稱）
                        ZipEntry zipEntry = new ZipEntry(file.getFileName());
                        zipEntry.setSize(localFile.length());
                        
                        // 將條目寫入ZIP流
                        zipOutStream.putNextEntry(zipEntry);
                        zipOutStream.write(data);
                        // 關(guān)閉當(dāng)前條目
                        zipOutStream.closeEntry();
                    }
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException("文件壓縮失敗：" + e.getMessage());
        } finally {
            // 關(guān)閉流（注意關(guān)閉順序：先關(guān)ZIP流，再關(guān)基礎(chǔ)輸出流）
            try {
                if (Objects.nonNull(zipOutStream)) {
                    zipOutStream.flush();
                    zipOutStream.close();
                }
                if (Objects.nonNull(outputStream)) {
                    outputStream.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

import com.itl.common.utils.CompressDownloadUtil;
import com.itl.project.common.attachment.domain.Attachment;
import com.itl.project.common.attachment.service.AttachmentService;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.UnsupportedEncodingException;
import java.util.List;
import java.util.UUID;

/**
 * 文件下載控制器
 */
@RestController
@RequestMapping("/file")
public class FileDownloadController {

    // 注入文件附件服務(wù)（實(shí)際項(xiàng)目中通過@Autowired注入）
    private AttachmentService attachmentService;

    /**
     * 多文件下載為ZIP壓縮包
     * @param fileid 多個(gè)文件ID，建議用逗號(hào)分隔（如：1,2,3）
     * @param request HTTP請(qǐng)求
     * @param response HTTP響應(yīng)
     * @throws UnsupportedEncodingException 編碼異常
     */
    @Log(title = "多文件下載", businessType = BusinessType.DOWNLOAD)
    @CrossOrigin // 解決跨域問題（前后端分離必加）
    @GetMapping("/downloadZip/{fileid}")
    public void downloadZip(@PathVariable("fileid") String fileid, 
                           HttpServletRequest request,
                           HttpServletResponse response) throws UnsupportedEncodingException {
        // 1. 根據(jù)文件ID查詢文件列表
        Attachment attachment = new Attachment();
        attachment.setFileIds(fileid);
        List<Attachment> attachmentList = attachmentService.selectAttachmentList(attachment);
        
        // 2. 校驗(yàn)文件列表是否為空
        if (attachmentList.isEmpty()) {
            response.setStatus(HttpServletResponse.SC_NOT_FOUND);
            return;
        }
        
        // 3. 設(shè)置響應(yīng)頭，告訴瀏覽器下載文件
        // 生成唯一的ZIP包名稱（避免重名）
        String downloadName = UUID.randomUUID().toString().replaceAll("-", "") + ".zip";
        // 解決文件名中文亂碼問題
        String fileName = new String(downloadName.getBytes("UTF-8"), "ISO8859-1");
        
        // 設(shè)置響應(yīng)頭
        response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
        // 設(shè)置響應(yīng)內(nèi)容類型為二進(jìn)制流（通用文件下載類型）
        response.setContentType("application/octet-stream");
        // 禁用緩存
        response.setHeader("Cache-Control", "no-cache");
        
        // 4. 調(diào)用工具類壓縮文件并寫入響應(yīng)輸出流
        try {
            CompressDownloadUtil.compressZip(attachmentList, response.getOutputStream());
        } catch (Exception e) {
            e.printStackTrace();
            response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
        }
    }
}

String fileName = new String(downloadName.getBytes("UTF-8"), "ISO8859-1");

// 前端下載ZIP包示例
function downloadZip(fileIds) {
    // 創(chuàng)建a標(biāo)簽，通過GET請(qǐng)求下載
    const a = document.createElement('a');
    a.href = `/file/downloadZip/${fileIds}`;
    a.download = '文件包.zip'; // 可選，優(yōu)先使用后端返回的文件名
    a.click();
    // 移除a標(biāo)簽
    a.remove();
}

// 調(diào)用示例：下載ID為1,2,3的文件
downloadZip("1,2,3");