fix: 优化自动过鱼数据接口

This commit is contained in:
tangwei 2026-07-22 13:51:46 +08:00
parent f7de51f6d2
commit aa8013fdb4
3 changed files with 85 additions and 27 deletions

View File

@ -9,13 +9,17 @@ import com.yfd.platform.qgc_base.domain.SdEngInfoBHOperateRequest;
import com.yfd.platform.qgc_base.domain.SdFpssrlR; import com.yfd.platform.qgc_base.domain.SdFpssrlR;
import com.yfd.platform.qgc_base.domain.SdFpssrlRAiRequest; import com.yfd.platform.qgc_base.domain.SdFpssrlRAiRequest;
import com.yfd.platform.qgc_base.service.ISdFpssrlRService; import com.yfd.platform.qgc_base.service.ISdFpssrlRService;
import com.yfd.platform.qgc_data.service.AttachmentUploadService;
import io.swagger.v3.oas.annotations.Operation; import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag; import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource; import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.*; import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.stream.Collectors;
/** /**
* <p> * <p>
@ -34,6 +38,9 @@ public class SdFpssrlRController {
@Resource @Resource
private ObjectMapper objectMapper; private ObjectMapper objectMapper;
@Resource
private AttachmentUploadService attachmentUploadService;
// ==================== CRUD 方法 ==================== // ==================== CRUD 方法 ====================
@PostMapping("/queryPageList") @PostMapping("/queryPageList")
@ -90,34 +97,65 @@ public class SdFpssrlRController {
// ==================== AI 设备识别数据上报接口 ==================== // ==================== AI 设备识别数据上报接口 ====================
/** /**
* AI设备识别数据上报接口 * AI设备图片/视频上传接口
* 接收AI设备自动识别的过鱼数据并入库 * 步骤1先上传图片获取附件URL再将URL填入数据上报接口
*/
@PostMapping("/ai/uploadImage")
@Operation(summary = "AI设备图片上传")
public ResponseResult aiUploadImage(@RequestParam("file") MultipartFile file) {
if (file == null || file.isEmpty()) {
return ResponseResult.error("文件不能为空");
}
try {
List<MultipartFile> files = new ArrayList<>();
files.add(file);
List<String> attachmentIds = attachmentUploadService.uploadMultipartFiles(files);
if (attachmentIds != null && !attachmentIds.isEmpty()) {
return ResponseResult.successData(attachmentIds);
} else {
return ResponseResult.error("图片上传失败");
}
} catch (Exception e) {
log.error("AI图片上传失败", e);
return ResponseResult.error("图片上传失败: " + e.getMessage());
}
}
/**
* AI设备识别数据上报接口支持单条和批量
* 步骤2上传完图片后将附件ID填入JSON数组调用此接口
*/ */
@PostMapping("/ai/report") @PostMapping("/ai/report")
@Operation(summary = "AI设备识别数据上报") @Operation(summary = "AI设备识别数据上报(支持批量)")
public ResponseResult aiReport(@RequestBody SdFpssrlRAiRequest request) { public ResponseResult aiReport(@RequestBody List<SdFpssrlRAiRequest> requests) {
if (request == null) { if (requests == null || requests.isEmpty()) {
return ResponseResult.error("请求数据不能为空"); return ResponseResult.error("请求数据不能为空");
} }
if (request.getStcd() == null || request.getStcd().isEmpty()) {
return ResponseResult.error("过鱼设施编码(stcd)不能为空"); // 逐条校验
} for (int i = 0; i < requests.size(); i++) {
if (request.getTm() == null) { SdFpssrlRAiRequest req = requests.get(i);
return ResponseResult.error("识别时间(tm)不能为空"); if (req == null) {
} return ResponseResult.error("" + (i + 1) + "条数据为空");
if (request.getFtp() == null || request.getFtp().isEmpty()) { }
return ResponseResult.error("鱼种类(ftp)不能为空"); if (req.getStcd() == null || req.getStcd().isEmpty()) {
return ResponseResult.error("" + (i + 1) + "条过鱼设施编码(stcd)不能为空");
}
if (req.getTm() == null) {
return ResponseResult.error("" + (i + 1) + "条识别时间(tm)不能为空");
}
if (req.getFtp() == null || req.getFtp().isEmpty()) {
return ResponseResult.error("" + (i + 1) + "条鱼种类(ftp)不能为空");
}
} }
try { try {
SdFpssrlR entity = sdFpssrlRService.processAiReport(request); List<SdFpssrlR> result = sdFpssrlRService.processAiReportBatch(requests);
if (entity != null) { // List<String> ids = result.stream().map(SdFpssrlR::getId).collect(Collectors.toList());
return ResponseResult.successData(entity.getId()); log.info("AI批量上报成功共{}条", result.size());
} else { return ResponseResult.success();
return ResponseResult.error("数据保存失败");
}
} catch (Exception e) { } catch (Exception e) {
log.error("AI识别数据上报失败", e); log.error("AI识别数据批量上报失败", e);
return ResponseResult.error("上报失败: " + e.getMessage()); return ResponseResult.error("上报失败: " + e.getMessage());
} }
} }

View File

@ -41,7 +41,12 @@ public interface ISdFpssrlRService extends IService<SdFpssrlR> {
SdFpssrlR getById(String id); SdFpssrlR getById(String id);
/** /**
* 处理AI设备上报的识别数据 * 处理AI设备上报的单条识别数据
*/ */
SdFpssrlR processAiReport(SdFpssrlRAiRequest request); SdFpssrlR processAiReport(SdFpssrlRAiRequest request);
/**
* 批量处理AI设备上报的识别数据
*/
List<SdFpssrlR> processAiReportBatch(List<SdFpssrlRAiRequest> requests);
} }

View File

@ -18,6 +18,7 @@ import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional; import org.springframework.transaction.annotation.Transactional;
import java.util.ArrayList;
import java.util.Date; import java.util.Date;
import java.util.List; import java.util.List;
@ -118,12 +119,26 @@ public class SdFpssrlRServiceImpl extends ServiceImpl<SdFpssrlRMapper, SdFpssrlR
entity.setIsDeleted(0); entity.setIsDeleted(0);
entity.setRemark(request.getRemark()); entity.setRemark(request.getRemark());
boolean result = this.save(entity); // boolean result = this.save(entity);
if (result) { // if (result) {
log.info("AI识别数据保存成功, id={}, stcd={}, ftp={}", entity.getId(), entity.getStcd(), entity.getFtp()); // log.info("AI识别数据保存成功, id={}, stcd={}, ftp={}", entity.getId(), entity.getStcd(), entity.getFtp());
} else { // } else {
log.error("AI识别数据保存失败, stcd={}, ftp={}", request.getStcd(), request.getFtp()); // log.error("AI识别数据保存失败, stcd={}, ftp={}", request.getStcd(), request.getFtp());
// }
return entity;
}
@Override
@Transactional(rollbackFor = Exception.class)
public List<SdFpssrlR> processAiReportBatch(List<SdFpssrlRAiRequest> requests) {
List<SdFpssrlR> results = new ArrayList<>();
for (SdFpssrlRAiRequest req : requests) {
SdFpssrlR entity = processAiReport(req);
if (entity != null) {
results.add(entity);
}
} }
return result ? entity : null; this.saveBatch(results);
return results;
} }
} }