mrxu-yolo/src/main/java/com/mrxu/yolo/service/YoloStreamTaskService.java

package com.mrxu.yolo.service;

import com.alibaba.fastjson.JSONObject;
import com.mrxu.framework.boot.bean.LayuiPage;
import com.mrxu.framework.boot.bean.ResponseObj;
import com.mrxu.framework.common.util.BaseCode;
import com.mrxu.framework.common.util.MrxuAssert;
import com.mrxu.yolo.python.YoloV12FeignClient;
import com.mrxu.yolo.python.response.StreamTaskResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

@Service
@Slf4j
public class YoloStreamTaskService {

    @Autowired
    private YoloV12FeignClient yoloV12FeignClient;

    public LayuiPage<StreamTaskResponse> page() {
        ResponseObj<List<StreamTaskResponse>> response = yoloV12FeignClient.streamTaskList();
        log.info("查询任务列表返回结果：{}", JSONObject.toJSONString(response));
        MrxuAssert.isTrue(BaseCode.OK.getCode() == response.getCode(),response.getMsg());
        LayuiPage layuiPage = new LayuiPage<>();
        layuiPage.setData(response.getResult());
        return layuiPage;
    }

    public void cancel(String task_id) {
        ResponseObj<String> response = yoloV12FeignClient.streamTaskCancel(task_id);
        log.info("去掉任务请求返回结果：{}", JSONObject.toJSONString(response));
        MrxuAssert.isTrue(BaseCode.OK.getCode() == response.getCode(),response.getMsg());
    }

}