Adding check for existing institutions (gbif/registry#579)

gbif · Jun 20, 2024 · 1f3db1d · 1f3db1d
1 parent e87ef79
commit 1f3db1d
Show file tree

Hide file tree

Showing 6 changed files with 86 additions and 17 deletions.
diff --git a/src/main/java/org/gbif/collections/sync/SyncResult.java b/src/main/java/org/gbif/collections/sync/SyncResult.java
@@ -1,6 +1,7 @@
 package org.gbif.collections.sync;
 
 import org.gbif.api.model.collections.Collection;
+import org.gbif.api.model.collections.CollectionEntity;
 import org.gbif.api.model.collections.Contact;
 import org.gbif.api.model.collections.Institution;
 
@@ -10,6 +11,7 @@
 import lombok.Builder;
 import lombok.Data;
 import lombok.Singular;
+import org.gbif.api.model.collections.suggestions.ChangeSuggestion;
 
 /**
  * Holds the result of a sync.
@@ -72,6 +74,7 @@ public static class NoEntityMatch {
     private Institution newInstitution;
     private Collection newCollection;
     private ContactMatch contactMatch;
+    private int newChangeSuggestion;
   }
 
   @Data

diff --git a/src/main/java/org/gbif/collections/sync/clients/http/GrSciCollHttpClient.java b/src/main/java/org/gbif/collections/sync/clients/http/GrSciCollHttpClient.java
@@ -4,6 +4,7 @@
 import org.gbif.api.model.collections.Contact;
 import org.gbif.api.model.collections.Institution;
 import org.gbif.api.model.collections.MasterSourceMetadata;
+import org.gbif.api.model.collections.suggestions.ChangeSuggestion;
 import org.gbif.api.model.common.paging.PagingResponse;
 import org.gbif.api.model.registry.Identifier;
 import org.gbif.api.model.registry.MachineTag;
@@ -121,6 +122,26 @@ public List<Institution> getIhInstitutions() {
     return result;
   }
 
+  public List<Institution> getInstitutionsByName(String name) {
+    List<Institution> result = new ArrayList<>();
+
+    boolean endRecords = false;
+    int offset = 0;
+    while (!endRecords) {
+      PagingResponse<Institution> response =
+          syncCall(api.listInstitutionsByName(name, 1000, offset));
+      endRecords = response.isEndOfRecords();
+      offset += response.getLimit();
+      result.addAll(response.getResults());
+    }
+
+    return result;
+  }
+
+  public int createChangeSuggestion(ChangeSuggestion changeSuggestion) {
+    return syncCall(api.createChangeSuggestion(changeSuggestion));
+  }
+
   public Institution getInstitution(UUID key) {
     return syncCall(api.getInstitution(key));
   }
@@ -259,6 +280,12 @@ Call<PagingResponse<Institution>> listInstitutions(
         @Query("limit") int limit,
         @Query("offset") int offset);
 
+    @GET("institution")
+    Call<PagingResponse<Institution>> listInstitutionsByName(
+        @Query("name") String name,
+        @Query("limit") int limit,
+        @Query("offset") int offset);
+
     @GET("institution/{key}")
     Call<Institution> getInstitution(@Path("key") UUID key);
 
@@ -357,6 +384,9 @@ Call<Void> addMasterSourceMetadataToInstitution(
     @POST("collection/{collectionKey}/masterSourceMetadata")
     Call<Void> addMasterSourceMetadataToCollection(
         @Path("collectionKey") UUID collectionKey, @Body MasterSourceMetadata masterSourceMetadata);
+
+    @POST("changeSuggestion")
+    Call<Integer> createChangeSuggestion(@Body ChangeSuggestion createSuggestion);
   }
 
   /** Adapter necessary for retrofit due to versioning. */

diff --git a/src/main/java/org/gbif/collections/sync/clients/proxy/BaseProxyClient.java b/src/main/java/org/gbif/collections/sync/clients/proxy/BaseProxyClient.java
@@ -1,6 +1,8 @@
 package org.gbif.collections.sync.clients.proxy;
 
+import java.util.List;
 import org.gbif.api.model.collections.Collection;
+import org.gbif.api.model.collections.CollectionEntity;
 import org.gbif.api.model.collections.Contact;
 import org.gbif.api.model.collections.Institution;
 import org.gbif.collections.sync.clients.http.GrSciCollHttpClient;
@@ -57,6 +59,16 @@ public void removeContactFromInstitution(UUID entityKey, int contactKey) {
     institutionHandler.removeContactFromEntityCall(entityKey, contactKey);
   }
 
+  @Override
+  public List<Institution> findInstitutionByName(String institutionName) {
+    return institutionHandler.listInstitutionsByName(institutionName);
+  }
+
+  @Override
+  public int createChangeSuggestion(CollectionEntity createSuggestion) {
+    return institutionHandler.createChangeSuggestion(createSuggestion);
+  }
+
   public Integer addContactToCollection(UUID entityKey, Contact contact) {
     return collectionHandler.addContactToEntityCall(entityKey, contact);
   }

diff --git a/src/main/java/org/gbif/collections/sync/clients/proxy/GrSciCollProxyClient.java b/src/main/java/org/gbif/collections/sync/clients/proxy/GrSciCollProxyClient.java
@@ -1,6 +1,8 @@
 package org.gbif.collections.sync.clients.proxy;
 
+import java.util.List;
 import org.gbif.api.model.collections.Collection;
+import org.gbif.api.model.collections.CollectionEntity;
 import org.gbif.api.model.collections.Institution;
 
 public interface GrSciCollProxyClient {
@@ -12,4 +14,8 @@ public interface GrSciCollProxyClient {
   Collection createCollection(Collection newCollection);
 
   boolean updateCollection(Collection oldCollection, Collection newCollection);
+
+  List<Institution> findInstitutionByName(String institutionName);
+
+  int createChangeSuggestion(CollectionEntity createSuggestion);
 }
diff --git a/src/main/java/org/gbif/collections/sync/common/BaseSynchronizer.java b/src/main/java/org/gbif/collections/sync/common/BaseSynchronizer.java
@@ -3,6 +3,7 @@
 import org.gbif.api.model.collections.Collection;
 import org.gbif.api.model.collections.Contact;
 import org.gbif.api.model.collections.Institution;
+import org.gbif.api.model.collections.suggestions.ChangeSuggestion;
 import org.gbif.collections.sync.SyncResult.*;
 import org.gbif.collections.sync.clients.proxy.GrSciCollProxyClient;
 import org.gbif.collections.sync.common.converter.EntityConverter;
@@ -128,23 +129,30 @@ public InstitutionAndCollectionMatch handleInstAndCollMatch(MatchResult<S, R> ma
   public NoEntityMatch handleNoMatch(MatchResult<S, R> matchResult) {
     // create institution
     Institution newInstitution = entityConverter.convertToInstitution(matchResult.getSource());
-
-    Institution createdInstitution = proxyClient.createInstitution(newInstitution);
-
-    // create collection
-    Collection createdCollection = createCollection(matchResult.getSource(), createdInstitution);
-
-    // same staff for both entities
-    ContactMatch contactMatchInstitution =
-        staffResultHandler.handleStaff(matchResult, createdInstitution);
-    ContactMatch contactMatchCollection =
-        staffResultHandler.handleStaff(matchResult, createdCollection);
-
-    return NoEntityMatch.builder()
-        .newCollection(createdCollection)
-        .newInstitution(createdInstitution)
-        .contactMatch(mergeContactMatches(contactMatchInstitution, contactMatchCollection))
-        .build();
+    if (proxyClient.findInstitutionByName(newInstitution.getName()).isEmpty()) {
+      Institution createdInstitution = proxyClient.createInstitution(newInstitution);
+      // create collection
+      Collection createdCollection = createCollection(matchResult.getSource(), createdInstitution);
+
+      // same staff for both entities
+      ContactMatch contactMatchInstitution =
+          staffResultHandler.handleStaff(matchResult, createdInstitution);
+      ContactMatch contactMatchCollection =
+          staffResultHandler.handleStaff(matchResult, createdCollection);
+
+      return NoEntityMatch.builder()
+          .newCollection(createdCollection)
+          .newInstitution(createdInstitution)
+          .contactMatch(mergeContactMatches(contactMatchInstitution, contactMatchCollection))
+          .build();
+    }
+    else {
+      // create change suggestion
+      int suggestionCreated = proxyClient.createChangeSuggestion(newInstitution);
+      return NoEntityMatch.builder()
+          .newChangeSuggestion(suggestionCreated)
+          .build();
+    }
   }
 
   @VisibleForTesting

diff --git a/src/main/java/org/gbif/collections/sync/common/handler/InstitutionHandler.java b/src/main/java/org/gbif/collections/sync/common/handler/InstitutionHandler.java
@@ -1,5 +1,7 @@
 package org.gbif.collections.sync.common.handler;
 
+import java.util.List;
+import org.gbif.api.model.collections.CollectionEntity;
 import org.gbif.api.model.collections.Contact;
 import org.gbif.api.model.collections.Institution;
 import org.gbif.api.model.collections.MasterSourceMetadata;
@@ -75,4 +77,12 @@ public void removeContactFromEntityCall(UUID entityKey, int contactKey) {
         () -> grSciCollHttpClient.removeContactFromInstitution(entityKey, contactKey),
         exceptionHandler(contactKey, "Failed to remove contact from institution " + entityKey));
   }
+
+  public List<Institution> listInstitutionsByName(String name) {
+    return grSciCollHttpClient.getInstitutionsByName(name);
+  }
+
+  public int createChangeSuggestion(CollectionEntity changeSuggestion) {
+    return grSciCollHttpClient.createChangeSuggestion(changeSuggestion);
+  }
 }