wms / hhwms · Files

package com.huaheng.pc.config.company.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import com.huaheng.common.utils.StringUtils;

import com.huaheng.common.utils.security.ShiroUtils;

import com.huaheng.framework.web.domain.AjaxResult;

import com.huaheng.pc.config.company.domain.Company;
import com.huaheng.pc.config.company.mapper.CompanyMapper;
import com.huaheng.pc.config.warehouseCompany.domain.WarehouseCompany;
import com.huaheng.pc.config.warehouseCompany.service.WarehouseCompanyService;

import com.huaheng.pc.system.user.domain.UserCompany;
import com.huaheng.pc.system.user.mapper.UserCompanyMapper;
import org.springframework.stereotype.Service;

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

import java.util.Map;

@Service("companyService")

public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> implements CompanyService {

    @Resource
    private CompanyMapper companyMapper;
    @Resource

    private UserCompanyMapper userCompanyMapper;

    @Resource
    private WarehouseCompanyService warehouseCompanyService;

    @Override

    public List<Company> selectCompanyByCurrentUserId() {

        return companyMapper.selectCompanyByUserId(ShiroUtils.getUserId(), ShiroUtils.getWarehouseCode());

    }

    /**
     *  查询出仓库所有的货主，并且根据用户id查询对应的货主，包含的就将flag赋为true

     * @param

     * @return
     */
    @Override

    public List<Company> selectCompanyByUserId(Integer id) {

        if (id == 0) {
            id = ShiroUtils.getUserId();
        }

        UserCompany condition = new UserCompany();

        condition.setUserId(id);

        List<UserCompany> userCompanys = userCompanyMapper.selectListEntityByEqual(condition);
        Company company = new Company();

        company.setWarehouseCode(ShiroUtils.getWarehouseCode());
        List<Company> companys = companyMapper.selectByWarehouseCode(company);

        for (Company companyItem : companys)
        {
            for (UserCompany userCompanyItem : userCompanys)
            {

                if (companyItem.getId().equals(userCompanyItem.getCompanyId()))

                {
                    companyItem.setFlag(true);
                    break;
                }
            }
        }
        return companys;
    }

    @Override
    public AjaxResult addCompany(Company company) {
        String[] warehouse = company.getWarehouseCode().split(",");
        //更新货主
        company.setCreatedBy(ShiroUtils.getLoginName());
        company.setLastUpdatedBy(ShiroUtils.getLoginName());
        this.save(company);
        //添加货主仓库关联
        WarehouseCompany record = new WarehouseCompany();
        for (int i=0; i<warehouse.length-1; i++)
        {
            record.setCompanyId(company.getId());
            record.setCompanyCode(company.getCode());
            record.setWarehouseCode(warehouse[i]);
            warehouseCompanyService.save(record);
        }
        return AjaxResult.success("更新成功！");
    }

    @Override
    public AjaxResult updateCompany(Company company) {

        if (company.getId() == null) {

            return AjaxResult.error("id为空！");

        }

        String[] warehouse = company.getWarehouseCode().split(",");
        LambdaQueryWrapper<WarehouseCompany> lambdaQueryWrapper = Wrappers.lambdaQuery();
        warehouseCompanyService.remove(lambdaQueryWrapper.eq(WarehouseCompany::getCompanyId, company.getId()));
        //添加货主仓库关联
        WarehouseCompany record = new WarehouseCompany();

        for (int i=0; i<warehouse.length; i++)

        {
            record.setCompanyId(company.getId());
            record.setCompanyCode(company.getCode());

            record.setWarehouseCode(warehouse[i]);

            warehouseCompanyService.save(record);
        }
        //更新货主
        company.setLastUpdatedBy(ShiroUtils.getLoginName());
        this.updateById(company);
        return AjaxResult.success("更新成功！");
    }

    @Override

    public List<Map<String, Object>> getCode(){
        LambdaQueryWrapper<Company> lambda = Wrappers.lambdaQuery();

        lambda.select(Company::getCode, Company::getId, Company::getName)

                .eq(Company::getWarehouseCode, ShiroUtils.getWarehouseCode());
        return this.listMaps(lambda);
    }

    @Override
    public List<Company> selectListEntityByEqual(Company condition) {
        return companyMapper.selectListEntityByEqual(condition);
    }

}