CompanyController.java 6.6 KB
package com.huaheng.pc.config.company.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.exception.service.ServiceException;
import com.huaheng.common.support.Convert;
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.pc.config.company.domain.Company;
import com.huaheng.pc.config.company.service.CompanyService;
import com.huaheng.pc.config.warehouse.domain.Warehouse;
import com.huaheng.pc.config.warehouse.service.WarehouseService;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.stereotype.Controller;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;

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

@Controller
@RequestMapping("/config/company")
public class CompanyController extends BaseController {
    @Resource
    private CompanyService companyService;
    @Resource
    private WarehouseService warehouseService;

    private String prefix = "config/company";

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

    /**
     * 查询货主列表
     */
    @RequiresPermissions("config:company:list")
    @Log(title = "配置-库存资料-货主管理", operating = "查看货主列表", action = BusinessType.GRANT)
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(Company company, String createdBegin, String createdEnd) {
        LambdaQueryWrapper<Company> lambdaQueryWrapper = Wrappers.lambdaQuery();
        PageDomain pageDomain = TableSupport.buildPageRequest();
        Integer pageNum = pageDomain.getPageNum();
        Integer pageSize = pageDomain.getPageSize();

        lambdaQueryWrapper.gt(StringUtils.isNotEmpty(createdBegin), Company::getCreated, createdBegin)
                .lt(StringUtils.isNotEmpty(createdEnd), Company::getCreated, createdEnd)
                .eq(Company::getWarehouseCode,ShiroUtils.getWarehouseCode())
                .like(StringUtils.isNotNull(company.getCode())&&StringUtils.isNotEmpty(company.getCode()), Company::getCode, company.getCode())
                .like(StringUtils.isNotNull(company.getCode())&&StringUtils.isNotEmpty(company.getName()), Company::getName, company.getName())
                .eq(Company::getDeleted,false)
                .orderByDesc(Company::getId);

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

    /**
     * 新增货主
     */
    @GetMapping("/add")
    public String add(ModelMap modelMap) {
        /*LambdaQueryWrapper<Warehouse> lambdaQueryWrapper = Wrappers.lambdaQuery();
        lambdaQueryWrapper.select(Warehouse::getCode,Warehouse::getName,Warehouse::getEnable);
        List<Map<String, Object>> warehouseList = warehouseService.listMaps(lambdaQueryWrapper);
        for (Map<String, Object> item : warehouseList){
            item.put("value",item.get("code").toString());
        }
        modelMap.put("warehouseList",warehouseList);*/
        return prefix + "/add";
    }

    /**
     * 新增保存货主
     */
    @RequiresPermissions("config:company:add")
    @Log(title = "配置-库存资料-货主管理", operating = "新增货主", action = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(Company company) {
        if(StringUtils.isEmpty(company.getCode()) && StringUtils.isEmpty(company.getName())){
            return AjaxResult.error("货主编码和名称不能为空!");
        }
        company.setWarehouseCode(ShiroUtils.getWarehouseCode());
        company.setCreated(new Date());
        company.setCreatedBy(ShiroUtils.getLoginName());
        company.setLastUpdated(new Date());
        company.setLastUpdatedBy(ShiroUtils.getLoginName());
        return companyService.addCompany(company);
    }

    /**
     * 修改货主
     */
    @GetMapping("/edit/{id}")
    public String edit(@PathVariable("id") Integer id, ModelMap mmap) {
        Company company = companyService.getById(id);
        mmap.put("company", company);
        mmap.put("warehouseList",warehouseService.getWarehouseList(id));
        return prefix + "/edit";
    }

    /**
     * 修改保存货主
     */
    @RequiresPermissions("config:company:edit")
    @Log(title = "配置-库存资料-货主管理", operating = "修改货主", action = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(Company company) {
        company.setLastUpdated(new Date());
        company.setLastUpdatedBy(ShiroUtils.getLoginName());
        return companyService.updateCompany(company);
    }

    /**
     * 删除货主
     */
    @RequiresPermissions("config:company:remove")
    @Log(title = "配置-库存资料-货主管理", operating ="删除货主", action = BusinessType.DELETE)
    @PostMapping( "/remove")
    @ResponseBody
    @Transactional
    public AjaxResult remove(String ids) {
        if (StringUtils.isEmpty(ids)){
            return AjaxResult.error("id不能为空");
        }

        for (Integer id : Convert.toIntArray(ids)){
            Company company = new Company();
            company.setId(id);
            company.setDeleted(true);
            company.setLastUpdatedBy(ShiroUtils.getLoginName());
            companyService.updateById(company);
        }
        return AjaxResult.success("删除成功!");
//        List<Integer> idList =  Arrays.asList(Convert.toIntArray(ids));
//        return toAjax(companyService.removeByIds(idList));
    }
}