/*
|
* Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the dreamlu.net developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: Chill 庄骞 (smallchill@163.com)
|
*/
|
package org.sxkj.fw.record.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.ExcelWriter;
|
import com.alibaba.excel.write.metadata.WriteSheet;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import io.swagger.annotations.ApiParam;
|
import lombok.AllArgsConstructor;
|
import org.springblade.core.boot.ctrl.BladeController;
|
import org.springblade.core.excel.util.ExcelUtil;
|
import org.springblade.core.mp.support.Condition;
|
import org.springblade.core.mp.support.Query;
|
import org.springblade.core.secure.BladeUser;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.constant.BladeConstant;
|
import org.springblade.core.tool.utils.BeanUtil;
|
import org.springblade.core.tool.utils.DateUtil;
|
import org.springblade.core.tool.utils.Func;
|
import org.springframework.web.bind.annotation.*;
|
import org.sxkj.fw.record.dto.FwDroneFlightRecordDTO;
|
import org.sxkj.fw.record.entity.FwDroneFlightRecordEntity;
|
import org.sxkj.fw.record.excel.FwDroneAlarmRecordExcel;
|
import org.sxkj.fw.record.excel.FwDroneFlightRecordExcel;
|
import org.sxkj.fw.record.param.FwDroneFlightRecordExportParam;
|
import org.sxkj.fw.record.service.IFwDroneFlightRecordService;
|
import org.sxkj.fw.record.vo.FwDroneFlightRecordVO;
|
import org.sxkj.fw.record.wrapper.FwDroneFlightRecordWrapper;
|
import springfox.documentation.annotations.ApiIgnore;
|
|
import javax.servlet.http.HttpServletResponse;
|
import javax.validation.Valid;
|
import java.io.FileOutputStream;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
import java.nio.charset.StandardCharsets;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Objects;
|
|
/**
|
* 无人机飞行记录表 控制器
|
*
|
* @author lw
|
* @since 2026-01-08
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("record/fwDroneFlightRecord")
|
@Api(value = "无人机飞行记录表", tags = "无人机飞行记录表接口")
|
public class FwDroneFlightRecordController extends BladeController {
|
|
private final IFwDroneFlightRecordService fwDroneFlightRecordService;
|
|
/**
|
* 无人机飞行记录表 详情
|
*/
|
@GetMapping("/detail")
|
@ApiOperationSupport(order = 1)
|
@ApiOperation(value = "详情", notes = "传入fwDroneFlightRecord")
|
public R<FwDroneFlightRecordVO> detail(FwDroneFlightRecordEntity fwDroneFlightRecord) {
|
FwDroneFlightRecordEntity detail = fwDroneFlightRecordService.getOne(Condition.getQueryWrapper(fwDroneFlightRecord));
|
return R.data(FwDroneFlightRecordWrapper.build().entityVO(detail));
|
}
|
/**
|
* 无人机飞行记录表 自定义分页
|
*/
|
@GetMapping("/page")
|
@ApiOperationSupport(order = 3)
|
@ApiOperation(value = "分页", notes = "传入fwDroneFlightRecord")
|
public R<IPage<FwDroneFlightRecordVO>> page(FwDroneFlightRecordDTO fwDroneFlightRecord, Query query) {
|
IPage<FwDroneFlightRecordVO> pages = fwDroneFlightRecordService.selectFwDroneFlightRecordPage(Condition.getPage(query), fwDroneFlightRecord);
|
return R.data(pages);
|
}
|
|
/**
|
* 无人机飞行记录表 新增或修改
|
*/
|
@PostMapping("/submit")
|
@ApiOperationSupport(order = 6)
|
@ApiOperation(value = "新增或修改", notes = "传入fwDroneFlightRecord")
|
public R submit(@Valid @RequestBody FwDroneFlightRecordDTO fwDroneFlightRecord) {
|
FwDroneFlightRecordEntity entity = Objects.requireNonNull(BeanUtil.copy(fwDroneFlightRecord, FwDroneFlightRecordEntity.class));
|
return R.status(fwDroneFlightRecordService.saveOrUpdate(entity));
|
}
|
|
/**
|
* 无人机飞行记录表 删除
|
*/
|
@PostMapping("/remove")
|
@ApiOperationSupport(order = 7)
|
@ApiOperation(value = "逻辑删除", notes = "传入ids")
|
public R remove(@ApiParam(value = "主键集合", required = true) @RequestParam String ids) {
|
return R.status(fwDroneFlightRecordService.deleteLogic(Func.toLongList(ids)));
|
}
|
|
|
/**
|
* 导出数据
|
*/
|
@GetMapping("/export-fwDroneFlightRecord")
|
@ApiOperationSupport(order = 9)
|
@ApiOperation(value = "导出数据", notes = "传入fwDroneFlightRecord")
|
public void exportFwDroneFlightRecord(FwDroneFlightRecordExportParam fwDroneFlightRecord, HttpServletResponse response) throws IOException {
|
Map<String,List<FwDroneFlightRecordExcel>> list = fwDroneFlightRecordService.exportFwDroneFlightRecord(fwDroneFlightRecord);
|
exportMultipleSheets(response, "无人机飞行记录表数据" + DateUtil.time(), list, FwDroneFlightRecordExcel.class);
|
}
|
|
// ... existing code ...
|
|
/**
|
* 导出多个sheet到Excel文件
|
* @param response HTTP响应对象
|
* @param fileName 文件名
|
* @param sheetDataMap 包含sheet名称和对应数据列表的映射
|
* @param clazz 数据类型对应的类
|
*/
|
public static <T> void exportMultipleSheets(HttpServletResponse response, String fileName,
|
java.util.Map<String, List<T>> sheetDataMap, Class<T> clazz) throws IOException {
|
try {
|
response.setContentType("application/vnd.ms-excel");
|
response.setCharacterEncoding(StandardCharsets.UTF_8.name());
|
fileName = URLEncoder.encode(fileName, StandardCharsets.UTF_8.name());
|
response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
|
|
ExcelWriter excelWriter = null;
|
try {
|
excelWriter = EasyExcel.write(response.getOutputStream(), clazz).build();
|
int sheetIndex = 0;
|
for (java.util.Map.Entry<String, List<T>> entry : sheetDataMap.entrySet()) {
|
String sheetName = entry.getKey();
|
List<T> dataList = entry.getValue();
|
if (dataList != null && !dataList.isEmpty()) {
|
com.alibaba.excel.write.metadata.WriteSheet writeSheet = EasyExcel.writerSheet(sheetIndex++, sheetName).build();
|
excelWriter.write(dataList, writeSheet);
|
}
|
}
|
} finally {
|
if (excelWriter != null) {
|
excelWriter.finish();
|
}
|
}
|
} catch (Throwable $ex) {
|
throw $ex;
|
}
|
}
|
|
|
|
}
|