|
@@ -1,6 +1,7 @@
|
|
|
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;
|
|
@@ -68,7 +69,20 @@ public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> impl
|
|
|
|
|
|
@Override
|
|
|
public List<Company> getIdAndNamelist(User user) {
|
|
|
- return companyMapper.selectIdAndNamelist(user);
|
|
|
+ QueryWrapper<Company> qw = new QueryWrapper<>();
|
|
|
+ qw.select("id","company_name","company_type");
|
|
|
+ if(Constant.SYS_ID == user.getParentId()){
|
|
|
+ //此时是admin创建用户,返回可选的生产方公司
|
|
|
+ qw.eq("company_type",1);
|
|
|
+ }else if(Constant.SYS_PARENT_ID == user.getParentId()){
|
|
|
+ //系统管理员创建admin,返回资产方公司
|
|
|
+ qw.eq("company_type",0);
|
|
|
+ }else{
|
|
|
+ //此时是项目经理创建其他用户
|
|
|
+ User admin = userMapper.selectOne(new QueryWrapper<User>().eq("id", user.getParentId()));
|
|
|
+ qw.eq("company_type",0).or().eq("id",admin.getCompanyId());
|
|
|
+ }
|
|
|
+ return companyMapper.selectList(qw);
|
|
|
}
|
|
|
|
|
|
@Override
|