MemberExportService.java

  1. package no.nav.data.team.member;

  3. import lombok.RequiredArgsConstructor;
  4. import no.nav.data.common.export.ExcelBuilder;
  5. import no.nav.data.common.utils.DateUtil;
  6. import no.nav.data.common.utils.StreamUtils;
  7. import no.nav.data.common.utils.StringUtils;
  8. import no.nav.data.team.cluster.ClusterService;
  9. import no.nav.data.team.cluster.domain.Cluster;
  10. import no.nav.data.team.member.MemberExportService.Member.Relation;
  11. import no.nav.data.team.member.dto.MemberResponse;
  12. import no.nav.data.team.po.ProductAreaService;
  13. import no.nav.data.team.po.domain.ProductArea;
  14. import no.nav.data.team.resource.NomGraphClient;
  15. import no.nav.data.team.shared.Lang;
  16. import no.nav.data.team.shared.domain.Membered;
  17. import no.nav.data.team.team.TeamService;
  18. import no.nav.data.team.team.domain.Team;
  19. import org.springframework.stereotype.Service;

  21. import java.time.LocalDate;
  22. import java.util.Comparator;
  23. import java.util.List;
  24. import java.util.UUID;
  25. import java.util.stream.Stream;

  27. import static java.util.Comparator.comparing;
  28. import static java.util.Objects.isNull;
  29. import static java.util.Optional.ofNullable;
  30. import static java.util.stream.Collectors.toList;
  31. import static no.nav.data.common.utils.StreamUtils.convert;
  32. import static no.nav.data.common.utils.StreamUtils.filter;
  33. import static no.nav.data.common.utils.StreamUtils.tryFind;
  34. import static org.apache.commons.lang3.StringUtils.EMPTY;

  36. @Service
  37. @RequiredArgsConstructor
  38. public class MemberExportService {

  40.     public enum SpreadsheetType {
  41.         ALL,
  42.         AREA,
  43.         CLUSTER,
  44.         TEAM,
  45.         ROLE,
  46.         LEADER
  47.     }

  49.     private final TeamService teamService;
  50.     private final ProductAreaService productAreaService;
  51.     private final ClusterService clusterService;
  52.     private final NomGraphClient nomGraphClient;

  54.     public byte[] generateSpreadsheet(SpreadsheetType type, String filter) {
  55.         var pas = productAreaService.getAll();
  56.         var clusters = clusterService.getAll();
  57.         var members = switch (type) {
  58.             case ALL -> getAll(pas, clusters);
  59.             case AREA -> getForProductArea(StringUtils.toUUID(filter), pas, clusters);
  60.             case CLUSTER -> getForCluster(StringUtils.toUUID(filter), pas, clusters);
  61.             case TEAM -> mapTeamMembers(List.of(teamService.get(StringUtils.toUUID(filter))), pas, clusters).collect(toList());
  62.             case ROLE -> filter(getAll(pas, clusters), m -> convert(m.member().getRoles(), Enum::name).contains(filter));
  63.             case LEADER -> filter(getAll(pas, clusters), m -> {
  64.                 var leaderMembers = nomGraphClient.getLeaderMembersActiveOnly(filter);
  65.                 return leaderMembers.contains(m.member().getNavIdent());
  66.             });
  67.         };
  68.         return generateFor(members);
  69.     }

  71.     private List<Member> getAll(List<ProductArea> pas, List<Cluster> clusters) {
  72.         return Stream.concat(
  73.                 Stream.concat(
  74.                         mapTeamMembers(teamService.getAll(), pas, clusters),
  75.                         mapPaMembers(pas)
  76.                 ),
  77.                 mapClusterMembers(clusters, pas)
  78.         ).collect(toList());
  79.     }

  81.     private List<Member> getForProductArea(UUID id, List<ProductArea> pas, List<Cluster> clusters) {
  82.         ProductArea productArea = productAreaService.get(id);
  83.         return Stream.concat(Stream.concat(
  84.                 mapPaMembers(List.of(productArea)),
  85.                 mapTeamMembers(teamService.findByProductArea(id).stream().filter(team -> team.getStatus().isActive()).toList(), pas, clusters))
  86.                 , mapClusterMembers(filter(clusters, cl -> productArea.getId().equals(cl.getProductAreaId())), pas)
  87.         ).collect(toList());
  88.     }

  90.     private List<Member> getForCluster(UUID id, List<ProductArea> pas, List<Cluster> clusters) {
  91.         return Stream.concat(
  92.                 mapClusterMembers(List.of(clusterService.get(id)), pas),
  93.                 mapTeamMembers(teamService.findByCluster(id), pas, clusters)
  94.         ).collect(toList());
  95.     }

  97.     private Stream<Member> mapPaMembers(List<ProductArea> productAreas) {
  98.         return productAreas.stream().flatMap(pa -> pa.getMembers().stream().map(m -> new Member(Relation.PA, m.convertToResponse(), null, pa, List.of())));
  99.     }

  101.     private Stream<Member> mapClusterMembers(List<Cluster> clusters, List<ProductArea> productAreas) {
  102.         return clusters.stream().flatMap(cluster -> cluster.getMembers().stream()
  103.                 .map(m -> new Member(Relation.CLUSTER, m.convertToResponse(), null, tryFind(productAreas, pa -> pa.getId().equals(cluster.getProductAreaId())).orElse(null),
  104.                         List.of(cluster))));
  105.     }

  107.     private Stream<Member> mapTeamMembers(List<Team> teams, List<ProductArea> pas, List<Cluster> clusters) {
  108.         return teams.stream().flatMap(t -> t.getMembers().stream().map(m -> {
  109.             ProductArea productArea = t.getProductAreaId() != null ? StreamUtils.find(pas, pa -> pa.getId().equals(t.getProductAreaId())) : null;
  110.             List<Cluster> clustersForTeam = filter(clusters, cluster -> t.getClusterIds().contains(cluster.getId()));
  111.             return new Member(Relation.TEAM, m.convertToResponse(), t, productArea, clustersForTeam);
  112.         }));
  113.     }

  115.     private byte[] generateFor(List<Member> members) {
  116.         var doc = new ExcelBuilder(Lang.MEMBERS);
  117.         doc.addRow()
  118.                 .addCell(Lang.RELATION)
  119.                 .addCell(Lang.AREA)
  120.                 .addCell(Lang.CLUSTER)
  121.                 .addCell(Lang.TEAM)
  122.                 .addCell(Lang.IDENT)
  123.                 .addCell(Lang.GIVEN_NAME)
  124.                 .addCell(Lang.FAMILY_NAME)
  125.                 .addCell(Lang.RESOURCE_TYPE)
  126.                 .addCell(Lang.ROLES)
  127.                 .addCell(Lang.OTHER)
  128.                 .addCell(Lang.EMAIL)
  129.                 .addCell(Lang.START_DATE)
  130.                 .addCell(Lang.END_DATE);

  132.         Comparator<Member> c1 = comparing(m -> ofNullable(m.member.getResource().getFamilyName()).orElse(""));
  133.         Comparator<Member> c2 = c1.thenComparing(m -> ofNullable(m.member.getResource().getGivenName()).orElse(""));
  134.         members.sort(c2);
  135.         members.forEach(m -> add(doc, m));

  137.         return doc.build();
  138.     }

  140.     private void add(ExcelBuilder doc, Member member) {
  141.         doc.addRow()
  142.                 .addCell(member.relationType())
  143.                 .addCell(member.productAreaName())
  144.                 .addCell(member.clusterName())
  145.                 .addCell(member.teamName())
  146.                 .addCell(member.member.getNavIdent())
  147.                 .addCell(member.member.getResource().getGivenName())
  148.                 .addCell(member.member.getResource().getFamilyName())
  149.                 .addCell(member.memberType())
  150.                 .addCell(member.roles())
  151.                 .addCell(member.member.getDescription())
  152.                 .addCell(member.member.getResource().getEmail())
  153.                 .addCell(DateUtil.formatDate(member.member.getResource().getStartDate()))
  154.                 .addCell(shouldHideEndDateIfBeforeNow(member.member.getResource().getEndDate()))
  155.         ;
  156.     }

  158.     private String shouldHideEndDateIfBeforeNow(LocalDate endDate) {
  159.         if (isNull(endDate) || endDate.isAfter(LocalDate.now())) return null;
  160.         else return DateUtil.formatDate(endDate);
  161.     }

  163.     record Member(Relation relation, MemberResponse member, Team team, ProductArea pa, List<Cluster> clusters) {

  165.         enum Relation {
  166.             TEAM(Team.class),
  167.             PA(ProductArea.class),
  168.             CLUSTER(Cluster.class);

  170.             private final Class<? extends Membered> type;

  172.             Relation(Class<? extends Membered> type) {
  173.                 this.type = type;
  174.             }
  175.         }

  177.         public String relationType() {
  178.             return Lang.objectType(relation.type);
  179.         }

  181.         public String productAreaName() {
  182.             return pa != null ? pa.getName() : EMPTY;
  183.         }

  185.         public String clusterName() {
  186.             return clusters.isEmpty() ? EMPTY : String.join(", ", convert(clusters, Cluster::getName));
  187.         }

  189.         public String teamName() {
  190.             return switch (relation) {
  191.                 case TEAM -> team.getName();
  192.                 case PA, CLUSTER -> EMPTY;
  193.             };
  194.         }

  196.         public String memberType() {
  197.             if (member.getResource().getResourceType() == null) {
  198.                 return EMPTY;
  199.             }
  200.             return Lang.memberType(member.getResource().getResourceType());
  201.         }

  203.         public String roles() {
  204.             return String.join(", ", convert(member.getRoles(), Lang::roleName));
  205.         }
  206.     }
  207. }
Created with JaCoCo 0.8.12.202403310830