package com.ruoyi.mock.controller;
|
|
import com.ruoyi.framework.web.domain.R;
|
import com.ruoyi.mock.dto.DataCheckRequest;
|
import com.ruoyi.mock.dto.DataGenerateRequest;
|
import com.ruoyi.mock.service.DataCheckService;
|
import com.ruoyi.mock.service.DataGenerateService;
|
import com.ruoyi.mock.vo.DataCheckResult;
|
import com.ruoyi.mock.vo.DataCheckResult.CheckItem;
|
import com.ruoyi.mock.vo.DataGenerateResult;
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
import lombok.RequiredArgsConstructor;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
@RestController
|
@RequestMapping("/mock")
|
@RequiredArgsConstructor
|
@Tag(name = "数据模拟")
|
public class DataCheckController {
|
|
private final DataCheckService dataCheckService;
|
private final DataGenerateService dataGenerateService;
|
|
@PostMapping("/dataCheck")
|
@Operation(summary = "检测基础数据")
|
public R<DataCheckResult> dataCheck(@RequestBody DataCheckRequest request) {
|
DataCheckResult result = dataCheckService.check(request);
|
return R.ok(result);
|
}
|
|
@PostMapping("/generate")
|
@Operation(summary = "AI生成模拟数据")
|
public R<?> generate(@RequestBody DataGenerateRequest request) {
|
// 先校验基础数据
|
DataCheckRequest checkRequest = new DataCheckRequest();
|
checkRequest.setModules(request.getModules());
|
DataCheckResult checkResult = dataCheckService.check(checkRequest);
|
|
boolean allPassed = checkResult.getItems().stream().allMatch(CheckItem::isPassed);
|
if (!allPassed) {
|
DataGenerateResult failResult = new DataGenerateResult();
|
failResult.setStatus("CHECK_FAILED");
|
failResult.setCheckResult(checkResult);
|
return R.fail(failResult, "基础数据不足，请先补充后再生成");
|
}
|
|
DataGenerateResult result = dataGenerateService.generate(request);
|
return R.ok(result);
|
}
|
}
|
