ReservationController.java 6.8 KB
package com.huaheng.pc.receipt.reservation.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.huaheng.api.general.domain.Receipt;
import com.huaheng.common.utils.StringUtils;
import com.huaheng.common.utils.security.ShiroUtils;
import com.huaheng.framework.aspectj.lang.annotation.Log;
import com.huaheng.framework.aspectj.lang.constant.BusinessType;
import com.huaheng.framework.web.controller.BaseController;
import com.huaheng.framework.web.domain.AjaxResult;
import com.huaheng.framework.web.page.PageDomain;
import com.huaheng.framework.web.page.TableDataInfo;
import com.huaheng.framework.web.page.TableSupport;
import com.huaheng.mobile.shipment.Shipment;
import com.huaheng.pc.receipt.receiptHeader.domain.ReceiptHeader;
import com.huaheng.pc.receipt.receiptHeader.service.ReceiptHeaderService;
import com.huaheng.pc.receipt.reservation.domain.Reservation;
import com.huaheng.pc.receipt.reservation.service.ReservationService;
import com.huaheng.pc.shipment.shipmentHeader.domain.ShipmentHeader;
import com.huaheng.pc.shipment.shipmentHeader.service.ShipmentHeaderService;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.util.List;

@ApiModel(value = "预约")
@RequestMapping("/reservation")
@Controller
public class ReservationController extends BaseController {

    @Resource
    private ReservationService reservationService;
    @Resource
    private ReceiptHeaderService receiptHeaderService;
    @Resource
    private ShipmentHeaderService shipmentHeaderService;

    private String prefix = "/receipt/reservation";

    @RequiresPermissions("reservation:view")
    @GetMapping()
    public String reservation() {
        return prefix + "/reservation";
    }

    /**
     * 查询预约
     */
    @RequiresPermissions("reservation:list")
    @Log(title = "预约-预约查讯", operating = "预约查讯", action = BusinessType.GRANT)
    @PostMapping("/list")
    @ResponseBody
    @ApiOperation("预约查询接口")
    public TableDataInfo list(Reservation reservation, String begin, String end) {
        LambdaQueryWrapper<Reservation> lambdaQueryWrapper = Wrappers.lambdaQuery();
        PageDomain pageDomain = TableSupport.buildPageRequest();
        Integer pageNum = pageDomain.getPageNum();
        Integer pageSize = pageDomain.getPageSize();

        lambdaQueryWrapper.like(StringUtils.isNotEmpty(reservation.getCarPlate()), Reservation::getCarPlate, reservation.getCarPlate())
            .like(StringUtils.isNotEmpty(reservation.getName()), Reservation::getName, reservation.getName())
            .eq(StringUtils.isNotEmpty(reservation.getStatus()), Reservation::getStatus, reservation.getStatus())
            .like(StringUtils.isNotEmpty(reservation.getCode()), Reservation::getCode, reservation.getCode())
            .gt(StringUtils.isNotEmpty(begin), Reservation::getAppointment, begin)
            .lt(StringUtils.isNotEmpty(end), Reservation::getAppointment, end)
            .eq(StringUtils.isNotEmpty(reservation.getStationCode()), Reservation::getStationCode, reservation.getStationCode());

        if (StringUtils.isNotNull(pageNum) && StringUtils.isNotNull(pageSize)){
            /**
             * 使用分页查询
             */
            Page<Reservation> page = new Page<>(pageNum, pageSize);
            IPage<Reservation> iPage = reservationService.page(page, lambdaQueryWrapper);
            return getMpDataTable(iPage.getRecords(),iPage.getTotal());
        } else {
            List<Reservation> list = reservationService.list(lambdaQueryWrapper);
            return getDataTable(list);
        }
    }

    /**
     * 新增预约
     */
    @GetMapping("/add")
    public String add() {
        return prefix + "/add";
    }

    /**
     * 新增保存预约
     */
    @ApiOperation(value="新增预约 ", notes="新增预约 ", httpMethod = "POST")
    @Log(title = "预约 ",operating = "新增预约 ", action = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(Reservation reservation) {
        LambdaQueryWrapper<ReceiptHeader> queryWrapper= Wrappers.lambdaQuery();
        queryWrapper.eq(ReceiptHeader::getCode, reservation.getCode());
        LambdaQueryWrapper<ShipmentHeader> queryWrapper1 = Wrappers.lambdaQuery();
        queryWrapper1.eq(ShipmentHeader::getCode, reservation.getCode());
        ReceiptHeader receiptHeader = receiptHeaderService.getOne(queryWrapper);
        ShipmentHeader shipmentHeader = shipmentHeaderService.getOne(queryWrapper1);
        if (StringUtils.isNull(receiptHeader) & StringUtils.isNull(shipmentHeader)) {
            return AjaxResult.error("单据不存在");
        }
        if (StringUtils.isNotNull(receiptHeader)) {
            reservation.setCompanyCode(receiptHeader.getCompanyCode());
            reservation.setWarehouseCode(receiptHeader.getWarehouseCode());
        } else if (StringUtils.isNotNull(shipmentHeader)) {
            reservation.setCompanyCode(shipmentHeader.getCompanyCode());
            reservation.setWarehouseCode(shipmentHeader.getWarehouseCode());
        }
        return toAjax(reservationService.save(reservation));
    }

    /**
     * 修改预约
     */
    @GetMapping("/edit/{id}")
    public String edit(@PathVariable("id") Integer id, ModelMap mmap) {
        Reservation reservation = reservationService.getById(id);
        mmap.put("reservation", reservation);
        return prefix + "/edit";
    }

    /**
     * 修改保存预约
     */
    @ApiOperation(value="修改预约", notes="修改预约", httpMethod = "POST")
    @RequiresPermissions("reservation:edit")
    @Log(title = "预约 ",operating = "修改预约 ", action = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(Reservation reservation) {
        return toAjax(reservationService.updateById(reservation));
    }

    @ApiOperation(value="修改预约状态", notes="修改预约状态", httpMethod = "POST")
    @RequiresPermissions("reservation:change")
    @PostMapping("/change")
    public AjaxResult changeStatus(Integer id, String status) {
        LambdaUpdateWrapper<Reservation> updateWrapper = Wrappers.lambdaUpdate();
        updateWrapper.eq(Reservation::getId, id)
                .set(Reservation::getStatus, status);
        return toAjax(reservationService.update(updateWrapper));
    }
}