|
@@ -24,6 +24,7 @@ import com.qunzhixinxi.hnqz.admin.api.entity.SysUserRole;
|
|
|
import com.qunzhixinxi.hnqz.admin.api.entity.WmPlatformQuizTestResult;
|
|
|
import com.qunzhixinxi.hnqz.admin.api.model.WmQuizResultModel;
|
|
|
import com.qunzhixinxi.hnqz.admin.api.vo.UserVO;
|
|
|
+import com.qunzhixinxi.hnqz.admin.entity.SysCsmUserRelation;
|
|
|
import com.qunzhixinxi.hnqz.admin.entity.SysDeptSub;
|
|
|
import com.qunzhixinxi.hnqz.admin.entity.SysUserSub;
|
|
|
import com.qunzhixinxi.hnqz.admin.entity.dto.CommonUserDTO;
|
|
@@ -88,6 +89,7 @@ public class SysUserController {
|
|
|
private final SysUserSubService sysUserSubService;
|
|
|
private final SysDictItemService sysDictItemService;
|
|
|
private final WmPlatformQuizTestResultService platformQuizTestResultService;
|
|
|
+ private final SysCsmUserRelationMapper sysCsmUserRelationMapper;
|
|
|
|
|
|
|
|
|
private boolean isZbOrDbUsers(Integer[] roles) {
|
|
@@ -1071,6 +1073,14 @@ public class SysUserController {
|
|
|
locations = list.stream().map(item -> Enum.valueOf(SubjectLocation.class, item.getValue())).collect(Collectors.toList());
|
|
|
}
|
|
|
|
|
|
+ Map<Integer, List<SysCsmUserRelation>> csmParentUserMap = new HashMap<>();
|
|
|
+ if (CollUtil.isNotEmpty(userIds)) {
|
|
|
+ // 上级csm
|
|
|
+ List<SysCsmUserRelation> csmUserRelations = sysCsmUserRelationMapper.selectList(Wrappers.<SysCsmUserRelation>lambdaQuery()
|
|
|
+ .in(SysCsmUserRelation::getUserId, userIds));
|
|
|
+ csmParentUserMap.putAll(csmUserRelations.stream().collect(Collectors.groupingBy(SysCsmUserRelation::getUserId)));
|
|
|
+ }
|
|
|
+
|
|
|
userVosPage.getRecords().forEach(userVO -> {
|
|
|
// 查询所属企业渠道
|
|
|
SysDeptSub sysDeptSub = new SysDeptSub();
|
|
@@ -1132,6 +1142,11 @@ public class SysUserController {
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ // 上级csm
|
|
|
+ List<SysCsmUserRelation> csmUserRelations = csmParentUserMap.get(userVO.getUserId());
|
|
|
+ if (CollUtil.isNotEmpty(csmUserRelations)) {
|
|
|
+ userVO.setParentIdList(csmUserRelations.stream().map(SysCsmUserRelation::getParentId).collect(Collectors.toList()));
|
|
|
+ }
|
|
|
});
|
|
|
|
|
|
return R.ok(userVosPage);
|