| | |
| | | import org.springframework.transaction.annotation.Transactional; |
| | | |
| | | import java.math.BigDecimal; |
| | | import java.util.ArrayList; |
| | | import java.util.Date; |
| | | import java.util.List; |
| | | import java.util.stream.Collectors; |
| | |
| | | |
| | | return ids; |
| | | } |
| | | |
| | | @Override |
| | | public List<MallMember> findParentMemberList(List<String> inviteIds, Integer level, boolean containsParent) { |
| | | List<MallMember> mallMembers = this.baseMapper.selectByInviteIds(inviteIds); |
| | | if(CollUtil.isEmpty(mallMembers)) { |
| | | return mallMembers; |
| | | } |
| | | |
| | | |
| | | List<MallMember> result = new ArrayList<>(); |
| | | int index = containsParent ? 0 : 1; |
| | | int count = 0; |
| | | while(true) { |
| | | if (level != null && count == level) { |
| | | break; |
| | | } |
| | | |
| | | if (index >= mallMembers.size()) { |
| | | break; |
| | | } |
| | | |
| | | String inviteId = inviteIds.get(index); |
| | | List<MallMember> collect = mallMembers.stream().filter(item -> item.getInviteId().equals(inviteId)).collect(Collectors.toList()); |
| | | result.add(collect.get(0)); |
| | | |
| | | index++; |
| | | count++; |
| | | } |
| | | return result; |
| | | } |
| | | } |