VehicleHeaderController.java 5.41 KB
package com.huaheng.pc.system.model.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
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.common.support.Convert;
import com.huaheng.common.utils.StringUtils;
import com.huaheng.common.utils.poi.ExcelUtil;
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.pc.config.material.domain.Material;
import com.huaheng.pc.receipt.receiptHeader.domain.ReceiptHeader;
import com.huaheng.pc.system.model.domain.Vehicle;
import com.huaheng.pc.system.model.domain.VehicleDetail;
import com.huaheng.pc.system.model.domain.VehicleHeader;
import com.huaheng.pc.system.model.service.IVehicleDetailService;
import com.huaheng.pc.system.model.service.IVehicleHeaderService;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

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

/**
 * 车型 信息操作处理
 *
 * @author huaheng
 * @date 2021-12-27
 */
@Controller
@RequestMapping("/system/model")
public class VehicleHeaderController extends BaseController {
    private String prefix = "system/model/modelHeader";

    @Resource
    private IVehicleHeaderService vehicleHeaderService;
    @Resource
    private IVehicleDetailService vehicleDetailService;

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

    /**
     * 查询车型列表
     */
    @RequiresPermissions("model:vehicleHeader:list")
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list() {
        LambdaQueryWrapper<VehicleHeader> lambdaQueryWrapper = Wrappers.lambdaQuery();
        PageDomain pageDomain = TableSupport.buildPageRequest();
        Integer pageNum = pageDomain.getPageNum();
        Integer pageSize = pageDomain.getPageSize();
        if (StringUtils.isNotNull(pageNum) && StringUtils.isNotNull(pageSize)) {
            /*使用分页查询*/
            Page<VehicleHeader> page = new Page<>(pageNum, pageSize);
            IPage<VehicleHeader> iPage = vehicleHeaderService.page(page, lambdaQueryWrapper);
            return getMpDataTable(iPage.getRecords(), iPage.getTotal());
        } else {
            List<VehicleHeader> list = vehicleHeaderService.list(lambdaQueryWrapper);
            return getDataTable(list);
        }
    }

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

    /**
     * 新增保存车型
     */
    @RequiresPermissions("model:vehicleHeader:add")
    @Log(title = "车型", action = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(VehicleHeader vehicleHeader) {
        vehicleHeader.setCreatedBy("管理员");
        vehicleHeader.setLastUpdatedBy("管理员");
        return toAjax(vehicleHeaderService.save(vehicleHeader));
    }

    /**
     * 修改车型
     */
    @GetMapping("/edit/{id}")
    public String edit(@PathVariable("id") Long id, ModelMap map) {
        VehicleHeader vehicleHeader = vehicleHeaderService.getById(id);
        map.put("vehicleHeader", vehicleHeader);
        return prefix + "/edit";
    }

    /**
     * 修改保存车型
     */
    @RequiresPermissions("model:vehicleHeader:edit")
    @Log(title = "车型", action = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(VehicleHeader vehicleHeader) {
        return toAjax(vehicleHeaderService.updateById(vehicleHeader));
    }

    /**
     * 删除车型
     */
    @RequiresPermissions("model:vehicleHeader:remove")
    @Log(title = "车型", action = BusinessType.DELETE)
    @PostMapping("/remove")
    @ResponseBody
    public AjaxResult remove(String ids) {
        if (StringUtils.isEmpty(ids)) {
            return AjaxResult.error("id不能为空");
        }
        LambdaQueryWrapper<VehicleDetail> lambdaQueryWrapper = Wrappers.lambdaQuery();
        lambdaQueryWrapper.eq(VehicleDetail::getVehicleId,ids);
        List<VehicleDetail> list = vehicleDetailService.list(lambdaQueryWrapper);
        if (list.size()>0)
        {
            return AjaxResult.error("请先删除车型明细");
        }

        return toAjax(vehicleHeaderService.removeByIds(Arrays.asList(Convert.toIntArray(ids))));
    }



    @PostMapping("/importData")
    @ResponseBody
    public AjaxResult importData (MultipartFile file) throws Exception {
        ExcelUtil<Vehicle> util = new ExcelUtil<>(Vehicle.class);
        List<Vehicle> vehicle = util.importExcel(file.getInputStream());
        String operName = ShiroUtils.getLoginName();
        String message = vehicleHeaderService.importMaterial(vehicle, false, operName);
        return AjaxResult.success(0);
    }


}