electripper-v2/ruoyi-web/src/main/java/com/ruoyi/web/app/AppFaultController.java

package com.ruoyi.web.app;

import com.ruoyi.bst.device.domain.DeviceVO;
import com.ruoyi.bst.device.service.DeviceService;
import com.ruoyi.bst.fault.domain.Fault;
import com.ruoyi.bst.fault.domain.FaultQuery;
import com.ruoyi.bst.fault.domain.FaultVO;
import com.ruoyi.bst.fault.service.FaultService;
import com.ruoyi.bst.order.service.OrderService;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;

/**
 * 故障申报前台
 */
@RestController
@RequestMapping("/app/fault")
public class AppFaultController extends BaseController {

    @Autowired
    private FaultService  faultService;

    @Autowired
    private DeviceService deviceService;

    @Autowired
    private OrderService orderService;

    /**
     * 新增
     * @param fault
     * @return
     */
    @PostMapping
    public AjaxResult add(@RequestBody Fault fault) {
        faultService.insertFault(fault);
        return success();
    }


    @GetMapping
    public AjaxResult get(FaultQuery query) {
        query.setUserId(getUserId());
        List<FaultVO> list = faultService.selectFaultList(query);
        List<FaultVO> validList = new ArrayList<>();

        list.forEach(faultVO -> {
            DeviceVO deviceVO = deviceService.selectDeviceBySn(faultVO.getVehicleCode());
            if (deviceVO == null){
                return;
            }
            Long orderId = deviceVO.getOrderId();
            if (orderId  == null) {
                return;
            }
            LocalDateTime orderEndTime = orderService.selectOrderById(orderId).getEndTime();
            if (orderEndTime == null) {
                return;
            }
            faultVO.setOrderEndTime(orderEndTime);
            validList.add(faultVO);
        });
        return success(validList);
        }
    }