java-insagent-pc/src/main/java/com/ydtech/modules/order/controller/OrderTaskPoolsController.java

package com.ydtech.modules.order.controller;


import com.ydtech.aop.request.RequestSingleParam;
import com.ydtech.core.page.HttpResult;
import com.ydtech.modules.order.entity.InsAreaCompany;
import com.ydtech.modules.order.entity.InsOrders;
import com.ydtech.modules.order.entity.vo.OrderTaskPoolsVo;
import com.ydtech.modules.order.entity.vo.TaskForwardingVo;
import com.ydtech.modules.order.service.InsAreaCompanyService;
import com.ydtech.modules.order.service.InsOrdersService;
import com.ydtech.modules.order.service.OrderTaskPoolsService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
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;

import java.util.List;


@Api(tags = "任务池")
@RestController
@RequestMapping(value = "/order/taskPools")
public class OrderTaskPoolsController {


    private final OrderTaskPoolsService orderTaskPoolsService;

    private final InsOrdersService insOrdersService;

    private final InsAreaCompanyService insAreaCompanyService;

    public OrderTaskPoolsController(OrderTaskPoolsService orderTaskPoolsService, InsOrdersService insOrdersService, InsAreaCompanyService insAreaCompanyService) {
        this.orderTaskPoolsService = orderTaskPoolsService;
        this.insOrdersService = insOrdersService;
        this.insAreaCompanyService = insAreaCompanyService;
    }

    @PostMapping(value = "/query")
    @ApiOperation(value = "查询")
    public HttpResult query(@RequestBody OrderTaskPoolsVo orderTaskPoolsVo) {
        return orderTaskPoolsService.query(orderTaskPoolsVo);
    }

    @PostMapping(value = "/forwarding")
    @ApiOperation(value = "任务转发")
    public HttpResult forwarding(@RequestBody TaskForwardingVo taskForwardingVo) {
        return orderTaskPoolsService.forwarding(taskForwardingVo);
    }

    @PostMapping("/queryOrderInfo")
    @ApiOperation(value = "通过主订单ID查询子订单")
    public HttpResult queryInsAreaCompany(@RequestSingleParam(value = "orderNo") String orderNo) {
        InsOrders insOrders = this.insOrdersService.existOrder(orderNo);
        List<InsAreaCompany> insAreaCompanies = this.insAreaCompanyService.query().eq("orderno", orderNo).list();
        insOrders.setInsAreaCompanyList(insAreaCompanies);
        return HttpResult.ok("success", insOrders);
    }

}