[backend/frontend] Fix import

openbas-api/src/main/java/io/openbas/migration/V3_16__Add_array_union_agg_method.java

@@ -1,32 +1,45 @@
 package io.openbas.migration;
 
+import lombok.extern.java.Log;
 import org.flywaydb.core.api.migration.BaseJavaMigration;
 import org.flywaydb.core.api.migration.Context;
 import org.springframework.stereotype.Component;
 
 import java.sql.Connection;
+import java.sql.ResultSet;
 import java.sql.Statement;
 
+@Log
 @Component
 public class V3_16__Add_array_union_agg_method extends BaseJavaMigration {
 
   @Override
   public void migrate(Context context) throws Exception {
     Connection connection = context.getConnection();
-    Statement select = connection.createStatement();
-    select.execute("CREATE FUNCTION array_union(a ANYARRAY, b ANYARRAY)"
-        + "    RETURNS ANYARRAY AS"
-        + "    $$"
-        + "SELECT array_agg(DISTINCT x)"
-        + "FROM ("
-        + "         SELECT unnest(a) x"
-        + "         UNION ALL SELECT unnest(b)"
-        + "     ) AS u"
-        + "    $$ LANGUAGE SQL;"
-        + "CREATE AGGREGATE array_union_agg(ANYARRAY) ("
-        + "  SFUNC = array_union,"
-        + "  STYPE = ANYARRAY,"
-        + "  INITCOND = '{}'"
-        + ");");
+    try (Statement statement = connection.createStatement()) {
+      ResultSet rs = statement.executeQuery(
+          "SELECT EXISTS (SELECT 1 FROM pg_proc WHERE proname = 'array_union');");
+
+      boolean functionExists = false;
+      if (rs.next()) {
+        functionExists = rs.getBoolean(1);
+      }
+      if (!functionExists) {
+        statement.execute("CREATE FUNCTION array_union(a ANYARRAY, b ANYARRAY)"
+            + "    RETURNS ANYARRAY AS"
+            + "    $$"
+            + "SELECT array_agg(DISTINCT x)"
+            + "FROM ("
+            + "         SELECT unnest(a) x"
+            + "         UNION ALL SELECT unnest(b)"
+            + "     ) AS u"
+            + "    $$ LANGUAGE SQL;"
+            + "CREATE AGGREGATE array_union_agg(ANYARRAY) ("
+            + "  SFUNC = array_union,"
+            + "  STYPE = ANYARRAY,"
+            + "  INITCOND = '{}'"
+            + ");");
+      }
+    }
   }
 }

openbas-api/src/main/java/io/openbas/rest/exercise/ExerciseApi.java

@@ -800,7 +800,11 @@
       importExport.setUsers(players);
       objectMapper.addMixIn(User.class, ExerciseExportMixins.User.class);
       // organizations
-      List<Organization> organizations = players.stream().map(User::getOrganization).filter(Objects::nonNull).toList();
+      List<Organization> organizations = players.stream()
+          .map(User::getOrganization)
+          .filter(Objects::nonNull)
+          .distinct()
+          .toList();
       exerciseTags.addAll(organizations.stream().flatMap(org -> org.getTags().stream()).toList());
       importExport.setOrganizations(organizations);
       objectMapper.addMixIn(Organization.class, ExerciseExportMixins.Organization.class);