NomGraphClient.java

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

  3. import com.fasterxml.jackson.databind.node.ArrayNode;
  4. import com.github.benmanes.caffeine.cache.Cache;
  5. import com.github.benmanes.caffeine.cache.Caffeine;
  6. import lombok.RequiredArgsConstructor;
  7. import lombok.SneakyThrows;
  8. import lombok.extern.slf4j.Slf4j;
  9. import no.nav.data.common.security.SecurityProperties;
  10. import no.nav.data.common.security.TokenProvider;
  11. import no.nav.data.common.utils.DateUtil;
  12. import no.nav.data.common.utils.JsonUtils;
  13. import no.nav.data.common.utils.MetricUtils;
  14. import no.nav.data.common.utils.StreamUtils;
  15. import no.nav.data.team.integration.process.GraphQLRequest;
  16. import no.nav.data.team.org.OrgUrlId;
  17. import no.nav.data.team.resource.domain.Resource;
  18. import no.nav.data.team.resource.dto.NomGraphQlResponse.MultiRessurs;
  19. import no.nav.data.team.resource.dto.NomGraphQlResponse.SingleOrg;
  20. import no.nav.data.team.resource.dto.NomGraphQlResponse.SingleRessurs;
  21. import no.nav.data.team.resource.dto.NomGraphQlResponse.SingleRessurs.DataWrapper;
  22. import no.nav.data.team.resource.dto.ResourceUnitsResponse;
  23. import no.nav.nom.graphql.model.*;
  24. import org.springframework.boot.web.client.RestTemplateBuilder;
  25. import org.springframework.http.HttpHeaders;
  26. import org.springframework.http.client.ClientHttpRequestInterceptor;
  27. import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;
  28. import org.springframework.lang.Nullable;
  29. import org.springframework.stereotype.Service;
  30. import org.springframework.util.ReflectionUtils;
  31. import org.springframework.web.client.RestOperations;
  32. import org.springframework.web.client.RestTemplate;

  34. import java.lang.reflect.Method;
  35. import java.time.Duration;
  36. import java.time.LocalDate;
  37. import java.util.ArrayList;
  38. import java.util.Collection;
  39. import java.util.List;
  40. import java.util.Map;
  41. import java.util.Objects;
  42. import java.util.Optional;
  43. import java.util.UUID;
  44. import java.util.stream.Stream;

  46. import static java.util.Objects.requireNonNull;
  47. import static no.nav.data.common.utils.StreamUtils.distinctByKey;
  48. import static no.nav.data.common.web.TraceHeaderRequestInterceptor.correlationInterceptor;

  50. /**
  51.  * Cannot be used in dev atm, as teamkat runs as nav.no, and nom as trygdeetaten.no
  52.  */
  53. @Slf4j
  54. @Service
  55. @RequiredArgsConstructor
  56. public class NomGraphClient {

  58.     private RestTemplate restTemplate;
  59.     private final RestTemplateBuilder restTemplateBuilder;
  60.     private final SecurityProperties securityProperties;
  61.     private final TokenProvider tokenProvider;
  62.     private final NomGraphQLProperties properties;

  64.     private static final String getResourceQuery = StreamUtils.readCpFile("nom/graphql/queries/get_org_for_ident.graphql");
  65.     private static final String getOrgQuery = StreamUtils.readCpFile("nom/graphql/queries/get_org_with_organiseringer.graphql");
  66.     private static final String getLeaderMemberQuery = StreamUtils.readCpFile("nom/graphql/queries/get_personer_for_org.graphql");
  67.     private static final String scopeTemplate = "api://%s-gcp.nom.nom-api/.default";

  69.     private static final Cache<String, RessursDto> ressursCache = MetricUtils.register("nomRessursCache",
  70.             Caffeine.newBuilder().recordStats()
  71.                     .expireAfterWrite(Duration.ofMinutes(10))
  72.                     .maximumSize(1000).build());

  74.     private static final Cache<String, OrgEnhetDto> orgCache = MetricUtils.register("nomOrgCache",
  75.             Caffeine.newBuilder().recordStats()
  76.                     .expireAfterWrite(Duration.ofMinutes(10))
  77.                     .maximumSize(1000).build());

  79.     private static final Cache<String, List<String>> leaderCache = MetricUtils.register("nomLeaderCache",
  80.             Caffeine.newBuilder().recordStats()
  81.                     .expireAfterWrite(Duration.ofMinutes(10))
  82.                     .maximumSize(1000).build());

  84.     public Optional<RessursDto> getRessurs(String navIdent) {
  85.         return Optional.ofNullable(getRessurser(List.of(navIdent)).get(navIdent));
  86.     }

  88.     public Optional<OrgEnhetDto> getOrgEnhet(String orgUrl) {
  89.         var org = orgCache.get(orgUrl, key -> {
  90.             var orgUrlData = new OrgUrlId(orgUrl);
  91.             Map<String,Object> orgMap = Map.of("agressoId",orgUrlData.getAgressoId(), "orgNiv", orgUrlData.getOrgNiv());

  93.             var req = new GraphQLRequest(getOrgQuery, orgMap);

  95.             var res = template().postForEntity(properties.getUrl(), req, SingleOrg.class);
  96.             logErrors("getOrgWithOrganiseringer", res.getBody());
  97.             var orgEnhet = requireNonNull(res.getBody()).getData().getOrgEnhet();
  98.             if (orgEnhet != null) {
  99.                 orgEnhet.setOrganiseringer(distinctByKey(orgEnhet.getOrganiseringer(), o -> o.getOrgEnhet().getAgressoId()));
  100.             }
  101.             return orgEnhet;
  102.         });
  103.         return Optional.ofNullable(org);
  104.     }

  106.     public Optional<ResourceUnitsResponse> getUnits(String navIdent) {
  107.         return getRessurs(navIdent)
  108.                 .map(r -> ResourceUnitsResponse.from(r, getLeaderMembersActiveOnly(navIdent), this::getOrgEnhet));
  109.     }

  111.     private Map<String, RessursDto> getRessurser(List<String> navIdents) {
  112.         return ressursCache.getAll(navIdents, idents -> {
  113.             var req = new GraphQLRequest(getResourceQuery, Map.of("navIdenter", idents));
  114.             var res = template().postForEntity(properties.getUrl(), req, MultiRessurs.class);
  115.             logErrors("getDepartments", res.getBody());
  116.             return requireNonNull(res.getBody()).getData().getRessurserAsMap();
  117.         });
  118.     }

  120.     public List<String> getLeaderMembersActiveOnly(String navIdent) {
  121.         var nomClient = NomClient.getInstance();
  122.         return getLeaderMembers(navIdent).stream()
  123.                 .map(nomClient::getByNavIdent)
  124.                 .filter(Optional::isPresent).map(Optional::get)
  125.                 .filter(it -> !it.isInactive()).map(Resource::getNavIdent).toList();
  126.     }

  128.     public List<String> getLeaderMembers(String navIdent) {
  129.         return leaderCache.get(navIdent, ident -> {
  130.             var req = new GraphQLRequest(getLeaderMemberQuery, Map.of("navident", navIdent));
  131.             var res = template().postForEntity(properties.getUrl(), req, SingleRessurs.class);
  132.             logErrors("getLeaderMembers", res.getBody());
  133.             var orgenheter = Optional.ofNullable(res.getBody())
  134.                     .map(SingleRessurs::getData)
  135.                     .map(DataWrapper::getRessurs)
  136.                     .stream()
  137.                     .map(RessursDto::getLederFor)
  138.                     .flatMap(Collection::stream)
  139.                     .map(LederOrgEnhetDto::getOrgEnhet)
  140.                     .filter(org -> DateUtil.isNow(org.getGyldigFom(), org.getGyldigTom())).toList();

  142.             var directMembers = new ArrayList<String>();
  143.             for (var org : orgenheter) {
  144.                 var refId = org.getId();
  145.                 var ressurser = org.getKoblinger().stream().map(OrgEnhetsKoblingDto::getRessurs);
  146.                 var okRessurser = ressurser.filter(it -> !it.getNavident().equals(navIdent) && this.ressursHarEnRelevantOrgtilknytning(it, refId));
  147.                 directMembers.addAll(okRessurser.map(RessursDto::getNavident).filter(Objects::nonNull).toList());
  148.             }

  150.             var subDepMembers = orgenheter.stream()
  151.                     .map(OrgEnhetDto::getOrganiseringer)
  152.                     .flatMap(Collection::stream)
  153.                     .map(OrganiseringDto::getOrgEnhet)
  154.                     .map(OrgEnhetDto::getLeder)
  155.                     .flatMap(Collection::stream)
  156.                     .map(OrgEnhetsLederDto::getRessurs)
  157.                     .map(RessursDto::getNavident)
  158.                     .filter(Objects::nonNull)
  159.                     .filter(id -> !id.equals(navIdent))
  160.                     .toList();


  163.             var x = UUID.randomUUID();
  164.             log.debug("{}: getLeaderMembers for {}: orgenheter size {}, directMembers size {}, subDepartmentMembers size {}",x, navIdent, orgenheter.size(), directMembers.size(), subDepMembers);
  165.             log.debug("{}\n{}",x,res.getBody().toString());
  166.             return Stream.concat(directMembers.stream(), subDepMembers.stream())
  167.                     .distinct()
  168.                     .toList();
  169.         });
  170.     }

  172.     @SneakyThrows
  173.     public void logErrors(String query, @Nullable Object body) {
  174.         if (body == null) {
  175.             return;
  176.         }
  177.         Method getErrors = requireNonNull(ReflectionUtils.findMethod(body.getClass(), "getErrors"));
  178.         var errors = Optional.ofNullable((ArrayNode) getErrors.invoke(body));
  179.         errors.ifPresent(errorArray -> log.error("Error during graphql query {} {}", query, JsonUtils.toJson(errorArray)));
  180.     }

  182.     private RestOperations template() {
  183.         if (restTemplate == null) {
  184.             restTemplate = restTemplateBuilder
  185.                     .additionalInterceptors(correlationInterceptor(), tokenInterceptor())
  186.                     .messageConverters(new MappingJackson2HttpMessageConverter())
  187.                     .build();
  188.         }
  189.         return restTemplate;
  190.     }

  192.     private boolean ressursHarEnRelevantOrgtilknytning(RessursDto ressursDto, String akseptertOrgenhetId){
  193.         var out = false;
  194.         for(var orgTilknytning : ressursDto.getOrgTilknytning()){
  195.             var idErRelevant = orgTilknytning.getOrgEnhet().getId().equals(akseptertOrgenhetId);
  196.             if(idErRelevant && orgTilknytning.getErDagligOppfolging()){
  197.                 var intervallOkBefore = orgTilknytning.getGyldigFom().isBefore(LocalDate.now().plusDays(1));
  198.                 var intervallOkAfter = (orgTilknytning.getGyldigTom() == null || LocalDate.now().minusDays(1).isBefore(orgTilknytning.getGyldigTom()));
  199.                 out |= intervallOkBefore && intervallOkAfter;
  200.             }
  201.         }
  202.         return out;
  203.     }

  205.     @SneakyThrows
  206.     private ClientHttpRequestInterceptor tokenInterceptor() {
  207.         return (request, body, execution) -> {
  208.             String token = tokenProvider.getConsumerToken(getScope());
  209.             log.debug("tokenInterceptor adding token: %s... for scope '%s'".formatted( (token != null && token.length() > 12 ? token.substring(0,11) : token ), getScope()));
  210.             request.getHeaders().add(HttpHeaders.AUTHORIZATION, token);
  211.             return execution.execute(request, body);
  212.         };
  213.     }

  215.     private String getScope() {
  216.         return scopeTemplate.formatted(securityProperties.isDev() ? "dev" : "prod");
  217.     }
  218. }
Created with JaCoCo 0.8.12.202403310830