fix thrift remote funciton

prestodb · Jul 19, 2024 · 30c4804 · 30c4804
1 parent 7bd2b82
commit 30c4804
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/presto-function-namespace-managers/pom.xml b/presto-function-namespace-managers/pom.xml
@@ -177,7 +177,6 @@
         <dependency>
             <groupId>com.facebook.drift</groupId>
             <artifactId>drift-transport-netty</artifactId>
-            <scope>test</scope>
         </dependency>
 
         <dependency>

diff --git a/...ava/com/facebook/presto/functionNamespace/mysql/MySqlFunctionNamespaceManagerFactory.java b/...ava/com/facebook/presto/functionNamespace/mysql/MySqlFunctionNamespaceManagerFactory.java
@@ -14,6 +14,7 @@
 package com.facebook.presto.functionNamespace.mysql;
 
 import com.facebook.airlift.bootstrap.Bootstrap;
+import com.facebook.drift.transport.netty.client.DriftNettyClientModule;
 import com.facebook.presto.functionNamespace.execution.SimpleAddressSqlFunctionExecutorsModule;
 import com.facebook.presto.spi.function.FunctionHandleResolver;
 import com.facebook.presto.spi.function.FunctionNamespaceManager;
@@ -50,6 +51,7 @@ public FunctionNamespaceManager<?> create(String catalogName, Map<String, String
     {
         try {
             Bootstrap app = new Bootstrap(
+                    new DriftNettyClientModule(),
                     new MySqlFunctionNamespaceManagerModule(catalogName),
                     new MySqlConnectionModule(),
                     new SimpleAddressSqlFunctionExecutorsModule());

diff --git a/...sting-udf-server/src/main/java/com/facebook/presto/udf/thrift/TestingThriftUdfServer.java b/...sting-udf-server/src/main/java/com/facebook/presto/udf/thrift/TestingThriftUdfServer.java
@@ -33,7 +33,10 @@ public static void start(Map<String, String> properties)
                         .add(new DriftNettyServerModule())
                         .add(new TestingThriftUdfServerModule())
                         .build());
-        app.setRequiredConfigurationProperties(properties).initialize();
+        if (!properties.isEmpty()) {
+            app.setRequiredConfigurationProperties(properties);
+        }
+        app.initialize();
     }
 
     public static void main(String[] args)