Commit fd807119 by huluobin

update

parent 0a92a114
......@@ -108,16 +108,20 @@ public abstract class AbstractCostService implements CostService {
@Override
public void setPrimaryDepartment(List<CostDomain> costDomains) {
List<OaUser> oaUserList = oaUserMapper.selectBatchIds(costDomains.stream().map(CostDomain::getCreateUserid).collect(Collectors.toList()));
List<OaUser> oaUserList = oaUserMapper.selectList(
new LambdaQueryWrapper<OaUser>()
.in(OaUser::getOaUserId, costDomains.stream().map(CostDomain::getCreateUserid).collect(Collectors.toList()))
);
if (ListUtil.isNotEmpty(oaUserList)) {
Map<Integer, OaDepartment> departmentMap = oaDepartmentMapper.selectBatchIds(
oaUserList
.stream()
.map(OaUser::getPrimaryDepartmentId)
.filter(Objects::nonNull)
.collect(Collectors.toList()))
Map<Integer, OaDepartment> departmentMap = oaDepartmentMapper.selectList(
new LambdaQueryWrapper<OaDepartment>()
.in(OaDepartment::getDepartmentId, oaUserList.stream()
.map(OaUser::getPrimaryDepartmentId)
.filter(Objects::nonNull)
.collect(Collectors.toList())))
.stream()
.collect(Collectors.toMap(OaDepartment::getDepartmentId, oaDepartment -> oaDepartment));
Map<Integer, OaUser> oaUserMap = oaUserList.stream()
.peek(oaUser -> oaUser.setPrimaryDepartment(departmentMap.get(oaUser.getPrimaryDepartmentId())))
.collect(Collectors.toMap(OaUser::getOaUserId, oaUser -> oaUser));
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment