- Add logic for preferredRoots API endpoint

EBISPOT · Aug 7, 2024 · faa8b44 · faa8b44
1 parent 478b7f6
commit faa8b44
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/backend/src/main/java/uk/ac/ebi/spot/ols/repository/v1/V1TermRepository.java b/backend/src/main/java/uk/ac/ebi/spot/ols/repository/v1/V1TermRepository.java
@@ -194,7 +194,17 @@ public Page<V1Term> getRoots(String ontologyId, boolean obsolete, String lang, P
     //    @Query (countQuery = "MATCH (n:PreferredRootTerm) WHERE n.ontology_name = {0} AND n.is_obsolete = {1} RETURN count(n)",
     //            value = "MATCH (n:PreferredRootTerm) WHERE n.ontology_name = {0} AND n.is_obsolete = {1} RETURN n")
     public Page<V1Term> getPreferredRootTerms(String ontologyId, boolean obsolete, String lang, Pageable pageable) {
-        throw new RuntimeException();
+
+        OlsSolrQuery query = new OlsSolrQuery();
+        query.addFilter("type", List.of("class"), SearchType.WHOLE_FIELD);
+        query.addFilter("ontologyId", List.of(ontologyId), SearchType.WHOLE_FIELD);
+        query.addFilter("isPreferredRoot", List.of("true"), SearchType.WHOLE_FIELD);
+
+        if (!obsolete)
+            query.addFilter(IS_OBSOLETE.getText(), List.of(Boolean.toString(obsolete)), SearchType.WHOLE_FIELD);
+
+        return solrClient.searchSolrPaginated(query, pageable)
+                .map(result -> V1TermMapper.mapTerm(result, lang));
     }
 
     //    @Query (value = "MATCH (n:PreferredRootTerm) WHERE n.ontology_name = {0} AND n.is_obsolete = {1} RETURN count(n)")