Merge pull request #8687 from Mats-SX/remoteprojprogress

Fix remote projection progress logging
neo4j · Feb 9, 2024 · 60fd53d · 60fd53d
2 parents c4afb15 + 9fabe07
commit 60fd53d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/core-test-utils/src/main/java/org/neo4j/gds/assertj/Extractors.java b/core-test-utils/src/main/java/org/neo4j/gds/assertj/Extractors.java
@@ -31,6 +31,17 @@ public final class Extractors {
 
     private Extractors() {}
 
+    public static ThrowingExtractor<String, String, RuntimeException> removingServerAddress() {
+        return Extractors::removingServerAddress;
+    }
+
+    private static String removingServerAddress(String message) {
+        if (message.contains("localhost:")) {
+            return removingServerAddress(message.replaceFirst("localhost:\\d+", "<address>"));
+        }
+        return message;
+    }
+
     public static ThrowingExtractor<String, String, RuntimeException> removingThreadId() {
         return Extractors::removeThreadId;
     }

diff --git a/core/src/main/java/org/neo4j/gds/api/DatabaseInfo.java b/core/src/main/java/org/neo4j/gds/api/DatabaseInfo.java
@@ -31,6 +31,10 @@ static DatabaseInfo of(DatabaseId databaseId, DatabaseLocation databaseLocation)
         return ImmutableDatabaseInfo.of(databaseId, databaseLocation, Optional.empty());
     }
 
+    static DatabaseInfo of(DatabaseId databaseId, DatabaseLocation databaseLocation, DatabaseId remoteDatabaseId) {
+        return ImmutableDatabaseInfo.of(databaseId, databaseLocation, remoteDatabaseId);
+    }
+
     DatabaseId databaseId();
 
     DatabaseLocation databaseLocation();