Merge pull request #8788 from geoand/#8785

Fix case where MongoDB and MariaDB extensions fail in native

Author: gsmet

extensions/jdbc/jdbc-mariadb/runtime/src/main/java/io/quarkus/jdbc/mariadb/runtime/graal/SharedMemorySocket_Removal.java

@@ -0,0 +1,10 @@
+package io.quarkus.jdbc.mariadb.runtime.graal;
+
+import com.oracle.svm.core.annotate.Delete;
+import com.oracle.svm.core.annotate.TargetClass;
+
+@TargetClass(className = "org.mariadb.jdbc.internal.io.socket.SharedMemorySocket")
+@Delete
+public final class SharedMemorySocket_Removal {
+
+}

extensions/jdbc/jdbc-mariadb/runtime/src/main/java/io/quarkus/jdbc/mariadb/runtime/graal/SimpleSocketHandlerFunction.java

@@ -0,0 +1,27 @@
+package io.quarkus.jdbc.mariadb.runtime.graal;
+
+import java.io.IOException;
+import java.net.Socket;
+
+import org.mariadb.jdbc.internal.io.socket.SocketHandlerFunction;
+import org.mariadb.jdbc.internal.util.Utils;
+import org.mariadb.jdbc.util.Options;
+
+public class SimpleSocketHandlerFunction implements SocketHandlerFunction {
+    @Override
+    public Socket apply(Options options, String host) throws IOException {
+        if (options.pipe != null) {
+            throw new IllegalArgumentException(getErrorMessage("pipe"));
+        } else if (options.localSocket != null) {
+            throw new IllegalArgumentException(getErrorMessage("localSocket"));
+        } else if (options.sharedMemory != null) {
+            throw new IllegalArgumentException(getErrorMessage("sharedMemory"));
+        }
+
+        return Utils.standardSocket(options, host);
+    }
+
+    private String getErrorMessage(String option) {
+        return "Option '" + option + "' is not available for MariaDB in native mode";
+    }
+}

extensions/jdbc/jdbc-mariadb/runtime/src/main/java/io/quarkus/jdbc/mariadb/runtime/graal/SocketUtility_Substitutions.java

@@ -0,0 +1,16 @@
+package io.quarkus.jdbc.mariadb.runtime.graal;
+
+import org.mariadb.jdbc.internal.io.socket.SocketHandlerFunction;
+
+import com.oracle.svm.core.annotate.Substitute;
+import com.oracle.svm.core.annotate.TargetClass;
+
+@TargetClass(org.mariadb.jdbc.internal.io.socket.SocketUtility.class)
+public final class SocketUtility_Substitutions {
+
+    // Ensure that JNA is never used
+    @Substitute
+    public static SocketHandlerFunction getSocketHandler() {
+        return new SimpleSocketHandlerFunction();
+    }
+}

extensions/jdbc/jdbc-mariadb/runtime/src/main/java/io/quarkus/jdbc/mariadb/runtime/graal/Utils_socketHandler.java

@@ -0,0 +1,16 @@
+package io.quarkus.jdbc.mariadb.runtime.graal;
+
+import org.mariadb.jdbc.internal.io.socket.SocketHandlerFunction;
+import org.mariadb.jdbc.internal.util.Utils;
+
+import com.oracle.svm.core.annotate.Alias;
+import com.oracle.svm.core.annotate.RecomputeFieldValue;
+import com.oracle.svm.core.annotate.TargetClass;
+
+@TargetClass(org.mariadb.jdbc.internal.util.Utils.class)
+public final class Utils_socketHandler {
+
+    @Alias
+    @RecomputeFieldValue(kind = RecomputeFieldValue.Kind.FromAlias)
+    private static SocketHandlerFunction socketHandler = new SimpleSocketHandlerFunction();
+}

extensions/mongodb-client/runtime/src/main/java/io/quarkus/mongodb/runtime/graal/MongoClientSubstitutions.java

@@ -173,3 +173,14 @@ public String resolveAdditionalQueryParametersFromTxtRecords(final String host)
         throw new UnsupportedOperationException("mongo+srv:// not supported in native mode");
     }
 }
+
+//TODO: move to a dedicated jna extension that will simply collect JNA substitutions
+@TargetClass(com.sun.jna.Native.class)
+final class JnaNativeSubstitutions {
+
+    // This method can trick GraalVM into thinking that Classloader#findLibrary is getting called
+    @Substitute
+    public static String getWebStartLibraryPath(final String libName) {
+        return null;
+    }
+}