From 580d70b15ae47bc180a0b579af8c47c506eefac6 Mon Sep 17 00:00:00 2001
From: huliguo <2023611923@qq.com>
Date: 星期五, 11 七月 2025 20:13:38 +0800
Subject: [PATCH] 修改
---
src/main/java/com/linghu/controller/PlatformController.java | 166 +++++++++++++++++++++++++++++++++++++------------------
1 files changed, 112 insertions(+), 54 deletions(-)
diff --git a/src/main/java/com/linghu/controller/PlatformController.java b/src/main/java/com/linghu/controller/PlatformController.java
index e71fb80..468a7f6 100644
--- a/src/main/java/com/linghu/controller/PlatformController.java
+++ b/src/main/java/com/linghu/controller/PlatformController.java
@@ -1,17 +1,37 @@
package com.linghu.controller;
import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
+import com.alibaba.excel.write.handler.SheetWriteHandler;
+import com.alibaba.excel.write.metadata.WriteSheet;
+import com.alibaba.excel.write.metadata.holder.WriteSheetHolder;
+import com.alibaba.excel.write.metadata.holder.WriteWorkbookHolder;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.linghu.listener.PlatformExcelListener;
+import com.linghu.listener.TypeDropdownWriteHandler;
+import com.linghu.mapper.ReferenceMapper;
import com.linghu.model.common.ResponseResult;
import com.linghu.model.entity.Platform;
+import com.linghu.model.entity.Reference;
+import com.linghu.model.entity.Sectionalization;
import com.linghu.model.entity.Type;
+import com.linghu.model.excel.ExcelDataWithRow;
import com.linghu.model.excel.PlatformExcel;
+import com.linghu.model.excel.UserExcel;
+import com.linghu.model.page.CustomPage;
import com.linghu.service.PlatformService;
import com.linghu.service.TypeService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
+import lombok.extern.log4j.Log4j;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.poi.ss.usermodel.DataValidation;
+import org.apache.poi.ss.usermodel.DataValidationConstraint;
+import org.apache.poi.ss.usermodel.DataValidationHelper;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.util.CellRangeAddressList;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
@@ -23,20 +43,26 @@
import java.io.ByteArrayOutputStream;
import java.io.IOException;
+import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
+import java.util.Objects;
import java.util.stream.Collectors;
@RestController
@RequestMapping("/platform")
@Api(value = "平台相关接口", tags = "设置-平台")
+@Slf4j
public class PlatformController {
+
@Autowired
private PlatformService platformService;
@Autowired
private TypeService typeService;
+ @Autowired
+ private ReferenceMapper referenceMapper;
@PostMapping
@ApiOperation(value = "添加平台")
@@ -57,7 +83,13 @@
}
@DeleteMapping("/{platformId}")
+ @ApiOperation(value = "删除平台")
public ResponseResult<Void> delete(@PathVariable Integer platformId) {
+ //平台被引用了没
+ Integer count = referenceMapper.selectCount(new LambdaQueryWrapper<Reference>().eq(Reference::getPlatform_id, platformId));
+ if (count > 0) {
+ return ResponseResult.error("该平台被引用中，不能删除");
+ }
boolean success = platformService.removeById(platformId);
if (success) {
return ResponseResult.success();
@@ -66,6 +98,7 @@
}
@PutMapping
+ @ApiOperation(value = "更新平台")
public ResponseResult<Void> update(@RequestBody Platform platform) {
// 校验平台名称和域名不能为空
if (!StringUtils.hasText(platform.getPlatform_name())) {
@@ -75,14 +108,13 @@
return ResponseResult.error("平台域名不能为空");
}
- boolean success = platformService.updateById(platform);
- if (success) {
- return ResponseResult.success();
- }
- return ResponseResult.error("更新平台失败");
+ platformService.updateById(platform);
+
+ return ResponseResult.success();
}
@GetMapping("/{platformId}")
+ @ApiOperation("根据id获取平台")
public ResponseResult<Platform> getById(@PathVariable Integer platformId) {
Platform platform = platformService.getById(platformId);
if (platform != null) {
@@ -92,24 +124,46 @@
}
@GetMapping("/list")
- public ResponseResult<List<Platform>> list(
+ @ApiOperation("查询平台列表，不传页数和大小就查全部")
+ public ResponseResult<CustomPage<Platform>> list(
@RequestParam(required = false) Integer page,
- @RequestParam(required = false) Integer pageSize) {
+ @RequestParam(required = false) Integer pageSize,
+ @RequestParam(required = false) Integer type_id) {
+ // 构建查询条件并添加排序（按创建时间倒序）
+ LambdaQueryWrapper<Platform> queryWrapper = new LambdaQueryWrapper<>();
+ queryWrapper.orderByDesc(Platform::getCreate_time); // 新增的排序条件
+ if (type_id != null) {
+ queryWrapper.eq(Platform::getType_id, type_id);
+ }
if (page != null && pageSize != null) {
Page<Platform> pageInfo = new Page<>(page, pageSize);
- Page<Platform> result = platformService.page(pageInfo);
- return ResponseResult.success(result.getRecords());
+ Page<Platform> result = platformService.page(pageInfo, queryWrapper);
+ return ResponseResult.success(new CustomPage<>(result));
} else {
- List<Platform> list = platformService.list();
- return ResponseResult.success(list);
+ List<Platform> list = platformService.list(queryWrapper);
+ CustomPage<Platform> customPage = new CustomPage<>(new Page<>());
+ customPage.setRecords(list);
+ customPage.setTotal(list.size());
+ return ResponseResult.success(customPage);
}
}
@GetMapping("/download")
@ApiOperation("下载平台模板")
public ResponseEntity<byte[]> downloadTemplate() throws IOException {
+ // 获取所有类型名称
+ List<Type> typeList = typeService.list();
+ List<String> typeNames = typeList.stream()
+ .map(Type::getType_name)
+ .collect(Collectors.toList());
+
ByteArrayOutputStream out = new ByteArrayOutputStream();
- EasyExcel.write(out, PlatformExcel.class).sheet("平台模板").doWrite(new ArrayList<>());
+
+ // 注册下拉框拦截器
+ EasyExcel.write(out, PlatformExcel.class)
+ .registerWriteHandler(new TypeDropdownWriteHandler(typeNames))
+ .sheet("平台模板")
+ .doWrite(new ArrayList<>());
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=platform_template.xlsx")
@@ -121,71 +175,75 @@
@ApiOperation("导入平台数据")
public ResponseResult<String> importPlatforms(@RequestParam("file") MultipartFile file) {
try {
- // 检查文件是否为空
if (file.isEmpty()) {
return ResponseResult.error("上传文件不能为空");
}
- // 读取Excel数据
- List<PlatformExcel> excelList = EasyExcel.read(file.getInputStream())
+ // 使用自定义监听器读取数据（包含行号）
+ PlatformExcelListener listener = new PlatformExcelListener();
+ EasyExcel.read(file.getInputStream())
.head(PlatformExcel.class)
.sheet()
- .doReadSync();
+ .registerReadListener(listener)
+ .doRead();
- // 数据转换与验证
+ List<ExcelDataWithRow<PlatformExcel>> excelList = listener.getDataList();
List<Platform> platforms = new ArrayList<>();
List<String> errorMessages = new ArrayList<>();
- for (PlatformExcel excel : excelList) {
- // 检查必要字段
+ // 遍历数据并验证（逻辑同方案一）
+ for (ExcelDataWithRow<PlatformExcel> excelData : excelList) {
+ int rowNum = excelData.getRowNumber();
+ PlatformExcel excel = excelData.getData();
+ List<String> rowErrors = new ArrayList<>();
+ // 验证逻辑
if (!StringUtils.hasText(excel.getPlatform_name())) {
- errorMessages.add("平台名称不能为空");
- continue;
- }
- if (!StringUtils.hasText(excel.getType_name())) {
- errorMessages.add("平台类型不能为空");
- continue;
+ rowErrors.add("平台名称不能为空");
}
if (!StringUtils.hasText(excel.getDomain())) {
- errorMessages.add("平台域名不能为空");
- continue;
+ rowErrors.add("平台域名不能为空");
}
+ // ... 其他验证
+ // 查找类型（仅当类型名称不为空时检查，避免空指针）
+ Integer typeId = null;
+ if (StringUtils.hasText(excel.getType_name())) {
+ Type typeByName = typeService.getTypeByName(excel.getType_name());
+ if (typeByName == null) {
+ rowErrors.add("未知的平台类型: " + excel.getType_name());
+ }
+ typeId = typeByName.getType_id();
- // 查找类型
- Type typeByName = typeService.getTypeByName(excel.getType_name());
- if (typeByName == null) {
- errorMessages.add("未知的平台类型: " + excel.getType_name());
- continue;
+ }else {
+ rowErrors.add("平台类型不能为空");
}
-
- // 构建平台对象
- Platform platform = new Platform();
- platform.setPlatform_name(excel.getPlatform_name());
- platform.setDomain(excel.getDomain());
- platform.setType_id(typeByName.getType_id());
-
- // 设置创建时间（解决之前的数据库错误）
- platform.setCreate_time(new Date());
-
- platforms.add(platform);
+ if (!rowErrors.isEmpty()) {
+ errorMessages.add(String.format("第%d行错误: %s", rowNum, String.join("；", rowErrors)));
+ } else {
+ // 构建Platform对象...
+ Platform platform = new Platform();
+ platform.setPlatform_name(excel.getPlatform_name());
+ platform.setDomain(excel.getDomain());
+ platform.setType_id(typeId);
+ platform.setCreate_time(LocalDateTime.now());
+ platforms.add(platform);
+ }
}
- // 处理错误
+ // 错误处理和导入逻辑（同方案一）
if (!errorMessages.isEmpty()) {
- return ResponseResult.error("数据验证失败: " + String.join("; ", errorMessages));
+ // 构建清晰的错误信息
+ StringBuilder errorMsg = new StringBuilder();
+ errorMsg.append("导入失败，共").append(errorMessages.size()).append("条数据存在错误：\n");
+ errorMessages.forEach(msg -> errorMsg.append(msg).append("\n"));
+
+ return ResponseResult.error(400, errorMsg.toString());
}
- // 批量保存
- if (!platforms.isEmpty()) {
- platformService.saveBatch(platforms);
- return ResponseResult.success("成功导入" + platforms.size() + "条数据");
- } else {
- return ResponseResult.error("没有有效数据可导入");
- }
+ // 无错误时导入...
+ return ResponseResult.success("导入成功，共" + platforms.size() + "条数据");
} catch (Exception e) {
- // 记录详细异常信息
-
+ log.error(e.getMessage());
return ResponseResult.error("文件解析失败：" + e.getMessage());
}
}
--
Gitblit v1.7.1