Changes to add missing class for ann

Author: bibith4

presto-iceberg/src/main/java/com/facebook/presto/iceberg/IcebergHiveMetadata.java

@@ -722,5 +722,4 @@ protected void updateIcebergViewProperties(
     {
         throw new PrestoException(NOT_SUPPORTED, "Iceberg Hive catalog does not support native Iceberg views for materialized views.");
     }
-
 }

presto-iceberg/src/main/java/com/facebook/presto/iceberg/IcebergNativeMetadata.java

@@ -529,5 +529,4 @@ protected void updateIcebergViewProperties(
 
         icebergViews.remove(viewName);
     }
-
 }

presto-iceberg/src/main/java/com/facebook/presto/iceberg/tvf/ApproxNearestNeighborsFunction.java

@@ -196,7 +196,7 @@ public static class IcebergAnnTableHandle
         private final int limit;
         public IcebergAnnTableHandle(List<Float> queryVector, int limit, String schema, String table)
         {
-            super(schema, IcebergTableName.from(table), false, Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), ImmutableList.of());
+            super(schema, IcebergTableName.from(table), false, Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), Optional.empty(), ImmutableList.of(), ImmutableList.of(), Optional.empty());
             this.queryVector = queryVector;
             this.limit = limit;
         }

presto-spi/src/main/java/com/facebook/presto/spi/connector/classloader/ClassLoaderSafeConnectorTableFunction.java

@@ -0,0 +1,83 @@
+/*
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.facebook.presto.spi.connector.classloader;
+
+import com.facebook.presto.spi.ConnectorSession;
+import com.facebook.presto.spi.classloader.ThreadContextClassLoader;
+import com.facebook.presto.spi.connector.ConnectorTransactionHandle;
+import com.facebook.presto.spi.function.table.Argument;
+import com.facebook.presto.spi.function.table.ArgumentSpecification;
+import com.facebook.presto.spi.function.table.ConnectorTableFunction;
+import com.facebook.presto.spi.function.table.ReturnTypeSpecification;
+import com.facebook.presto.spi.function.table.TableFunctionAnalysis;
+
+import java.util.List;
+import java.util.Map;
+
+import static java.util.Objects.requireNonNull;
+
+public class ClassLoaderSafeConnectorTableFunction
+        implements ConnectorTableFunction
+{
+    private final ConnectorTableFunction delegate;
+    private final ClassLoader classLoader;
+
+    public ClassLoaderSafeConnectorTableFunction(ConnectorTableFunction delegate, ClassLoader classLoader)
+    {
+        this.delegate = requireNonNull(delegate, "delegate is null");
+        this.classLoader = requireNonNull(classLoader, "classLoader is null");
+    }
+
+    @Override
+    public String getSchema()
+    {
+        try (ThreadContextClassLoader a = new ThreadContextClassLoader(classLoader)) {
+            return delegate.getSchema();
+        }
+    }
+
+    @Override
+    public String getName()
+    {
+        try (ThreadContextClassLoader a = new ThreadContextClassLoader(classLoader)) {
+            return delegate.getName();
+        }
+    }
+
+    @Override
+    public List<ArgumentSpecification> getArguments()
+    {
+        try (ThreadContextClassLoader a = new ThreadContextClassLoader(classLoader)) {
+            return delegate.getArguments();
+        }
+    }
+
+    @Override
+    public ReturnTypeSpecification getReturnTypeSpecification()
+    {
+        try (ThreadContextClassLoader a = new ThreadContextClassLoader(classLoader)) {
+            return delegate.getReturnTypeSpecification();
+        }
+    }
+
+    @Override
+    public TableFunctionAnalysis analyze(ConnectorSession session,
+                                         ConnectorTransactionHandle transaction,
+                                         Map<String, Argument> arguments)
+    {
+        try (ThreadContextClassLoader a = new ThreadContextClassLoader(classLoader)) {
+            return delegate.analyze(session, transaction, arguments);
+        }
+    }
+}