yunsu/cloud-model/src/main/java/com/hssx/cloudmodel/service/impl/CompanyServiceImpl.java

package com.hssx.cloudmodel.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.hssx.cloudmodel.constant.Constant;
import com.hssx.cloudmodel.entity.Company;
import com.hssx.cloudmodel.entity.User;
import com.hssx.cloudmodel.mapper.CompanyMapper;
import com.hssx.cloudmodel.mapper.UserMapper;
import com.hssx.cloudmodel.service.CompanyService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hssx.cloudmodel.util.HttpRespMsg;
import com.hssx.cloudmodel.util.PageUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 吴涛涛
 * @since 2019-07-26
 */
@Service
public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> implements CompanyService {
    @Autowired
    CompanyMapper companyMapper;
    @Autowired
    UserMapper userMapper;

    @Override
    public HttpRespMsg addAndUpdateRole(Company company, Integer flag) {
        HttpRespMsg msg = new HttpRespMsg();
        if (flag == 0) {
            //添加公司
            QueryWrapper<Company> qw = new QueryWrapper<>();
            qw.eq("company_name", company.getCompanyName());
            int count = companyMapper.selectCount(qw);
            if (count > 0) {
                msg.setError("公司已存在，请勿重复添加");
            } else {
                companyMapper.insert(company);
            }
        } else if (flag == 1) {
            //更新公司信息
            companyMapper.updateById(company);
        }
        return msg;
    }

    @Override
    public HttpRespMsg pageList(PageUtil page, String keyName) {
        HttpRespMsg msg = new HttpRespMsg();
        QueryWrapper<Company> qw = new QueryWrapper<>();
        if(keyName !=null && !"".equals(keyName)){
            qw.like("company_name", keyName);
        }
        Integer start = (page.getPageNum()-1)*page.getPageSize();
        Integer count = companyMapper.selectCount(qw);
        page.setTotal(count);
        page.setPages(page.getTotal());
        List<Company> list = companyMapper.getListByKeyName(keyName, start, page.getPageSize());
        page.setList(list);
        msg.data = page;
        return msg;
    }

    @Override
    public List<Company> getIdAndNamelist(User user) {
        QueryWrapper<Company> qw = new QueryWrapper<>();
        qw.select("id","company_name","company_type");
        if(Constant.SYS_ID == user.getParentId()){
            //此时是admin创建用户,返回可选的生产方公司
            qw.eq("company_type",Constant.PRODUCER_COMPANY);
        }else if(Constant.SYS_PARENT_ID == user.getParentId()){
            //系统管理员创建admin,返回资产方公司
            qw.eq("company_type",Constant.ASSETS_COMPANY);
        }else{
            //此时是项目经理创建其他用户
            User admin = userMapper.selectOne(new QueryWrapper<User>().eq("id", user.getParentId()));
            qw.eq("company_type",Constant.ASSETS_COMPANY).or().eq("id",admin.getCompanyId());
        }
        return companyMapper.selectList(qw);
    }

    @Override
    public HttpRespMsg deleteById(Integer id) {
        HttpRespMsg msg = new HttpRespMsg();
        Integer count = userMapper.selectCount(new QueryWrapper<User>().eq("company_id", id));
        if (count > 0) {
            msg.setError("该公司已被应用到用户中，暂不提供删除操作");
        } else {
            companyMapper.deleteById(id);
        }
        return msg;
    }
}