package com.linghu.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.linghu.mapper.SectionalizationMapper;
|
import com.linghu.mapper.UserMapper;
|
import com.linghu.model.entity.Sectionalization;
|
import com.linghu.model.entity.User;
|
import com.linghu.model.vo.SectionalizationUserVO;
|
import com.linghu.service.SectionalizationService;
|
import com.linghu.service.UserService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.util.Collections;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* @author xy
|
* @description 针对表【user】的数据库操作Service实现
|
* @createDate 2025-07-07 10:26:00
|
*/
|
@Service
|
public class SectionalizationServiceImpl extends ServiceImpl<SectionalizationMapper, Sectionalization>
|
implements SectionalizationService {
|
|
@Autowired
|
private UserMapper userMapper;
|
@Override
|
public List<SectionalizationUserVO> SectionalizationUser() {
|
// 1. 查询所有状态为"正常"的用户
|
List<User> userList = userMapper.selectList(
|
new LambdaQueryWrapper<User>()
|
.eq(User::getStatus, "正常")
|
);
|
|
// 2. 提取所有分组ID并去重
|
List<Integer> sectionalizationIdList = userList.stream()
|
.map(User::getSectionalization_id)
|
.distinct()
|
.collect(Collectors.toList());
|
|
// 3. 查询这些分组的基本信息
|
List<Sectionalization> sectionalizationList = this.baseMapper.selectList(
|
new LambdaQueryWrapper<Sectionalization>()
|
.in(Sectionalization::getSectionalization_id, sectionalizationIdList)
|
);
|
|
// 4. 创建分组ID到分组名称的映射
|
Map<Integer, String> sectionalizationNameMap = sectionalizationList.stream()
|
.collect(Collectors.toMap(
|
Sectionalization::getSectionalization_id,
|
Sectionalization::getSectionalization_name
|
));
|
|
// 5. 按分组ID对用户进行分组
|
Map<Integer, List<User>> usersBySectionalization = userList.stream()
|
.collect(Collectors.groupingBy(User::getSectionalization_id));
|
|
// 6. 构建最终返回结果
|
return sectionalizationIdList.stream()
|
.map(id -> {
|
SectionalizationUserVO vo = new SectionalizationUserVO();
|
vo.setSectionalization_id(id);
|
vo.setSectionalization_name(sectionalizationNameMap.get(id));
|
vo.setSectionalization_users(usersBySectionalization.getOrDefault(id, Collections.emptyList()));
|
return vo;
|
})
|
.collect(Collectors.toList());
|
}
|
}
|
