import com.alibaba.excel.EasyExcel;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.List;

@RestController
@RequestMapping("/api/excel")
public class ExcelController {

    @GetMapping("/download")
    public void downloadExcel(HttpServletResponse response) {
        List<BmsComparisonExcelReportDto> excelList = fetchData(); // 假设这是获取数据的方法
        writeExcel(response, "template", excelList);
    }

    private void writeExcel(HttpServletResponse response, String fileName, List<?> excelList) {
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setCharacterEncoding("utf-8");
        
        try {
            fileName = URLEncoder.encode(fileName, "UTF-8").replaceAll("\\+", "%20");
        } catch (UnsupportedEncodingException e) {
            log.info("【编码异常】 '{}'", e);
        }
        response.setHeader("Content-Disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
        try {
            EasyExcel.write(response.getOutputStream(), BmsComparisonExcelReportDto.class).sheet("模板").doWrite(excelList);
        } catch (IOException e) {
            log.info("【写入excel失败】", e);
        }
    }
    
    private List<BmsComparisonExcelReportDto> fetchData() {
        // 获取数据的逻辑
        return List.of(
            new BmsComparisonExcelReportDto(1, "数据1"),
            new BmsComparisonExcelReportDto(2, "数据2")
        );
    }
}

@Data
@AllArgsConstructor
public class BmsComparisonExcelReportDto {
    private Integer id;
    private String name;
}