SimulationService/src/main/java/com/simulationservice/controller/InferenceController.java
2025-09-14 12:49:22 +08:00

38 lines
1.4 KiB
Java

package com.simulationservice.controller;
import com.simulationservice.common.R;
import com.simulationservice.service.InferenceTaskService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import java.util.HashMap;
import java.util.Map;
/**
* 想定控制,与前端进行交互的接口
* 想定控制,包括想定的加载,开始,暂停,停止,加速,减速
*/
@RestController
public class InferenceController {
@Autowired
private InferenceTaskService inferenceTaskService;
@GetMapping("/api/loadScenario")
public R<Map<String, Object>> loadScenario(@RequestParam(value = "roomId") String roomId, @RequestParam(value = "scenarioId") String scenarioId) {
// 服务端加载想定
Map<String, Object> retMap = new HashMap<>();
boolean ret = inferenceTaskService.loadScenario(roomId, scenarioId);
return ret ? R.success(retMap) : R.error(retMap);
}
@GetMapping("/api/startScenario")
public String startTask(@RequestParam(value = "roomId") String roomId, @RequestParam(value = "scenarioId") String scenarioId) {
// 启动后台任务
inferenceTaskService.executeTask();
return "任务已启动";
}
}