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;

import static com.ydtech.modules.base.controller.BaseController.getUser;


@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);
    }

    @PostMapping(value = "/checkCurrentUserEdit")
    @ApiOperation(value = "查询")
    public HttpResult queryOrderOperatorId(@RequestSingleParam(value = "orderNo") String orderNo) {
        return orderTaskPoolsService.checkCurrentUserEdit(orderNo);
    }


    @PostMapping(value = "/querySubOrders")
    @ApiOperation(value = "子订单任务查询")
    public HttpResult querySubOrders(@RequestBody OrderTaskPoolsVo orderTaskPoolsVo) {
        return orderTaskPoolsService.querySubOrders(orderTaskPoolsVo);
    }

    @PostMapping(value = "/forwardingSubOrders")
    @ApiOperation(value = "子订单任务转发")
    public HttpResult forwardingSubOrders(@RequestBody TaskForwardingVo taskForwardingVo) {
        return orderTaskPoolsService.forwardingSubOrders(taskForwardingVo,getUser());
    }

    @PostMapping(value = "/recover")
    @ApiOperation(value = "子订单任务收回")
    public HttpResult recover(@RequestBody TaskForwardingVo taskForwardingVo) {
        return orderTaskPoolsService.recover(taskForwardingVo,getUser());
    }
}