Add licenses, fix features config

tdcmeehan · tdcmeehan · commit 39f050502ccc · 2025-10-10T11:56:17.000-04:00
diff --git a/presto-main-base/src/main/java/com/facebook/presto/metadata/CodecDeserializer.java b/presto-main-base/src/main/java/com/facebook/presto/metadata/CodecDeserializer.java
@@ -1,3 +1,16 @@
+/*
+ * 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.metadata;
 
 import com.facebook.presto.spi.ConnectorCodec;
diff --git a/presto-main-base/src/main/java/com/facebook/presto/metadata/CodecSerializer.java b/presto-main-base/src/main/java/com/facebook/presto/metadata/CodecSerializer.java
@@ -1,3 +1,16 @@
+/*
+ * 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.metadata;
 
 import com.facebook.presto.spi.ConnectorCodec;
diff --git a/presto-main-base/src/test/java/com/facebook/presto/catalogserver/TestCatalogServerResponse.java b/presto-main-base/src/test/java/com/facebook/presto/catalogserver/TestCatalogServerResponse.java
@@ -13,8 +13,6 @@
  */
 package com.facebook.presto.catalogserver;
 
-import com.facebook.airlift.configuration.ConfigurationFactory;
-import com.facebook.airlift.configuration.ConfigurationModule;
 import com.facebook.airlift.json.JsonModule;
 import com.facebook.presto.common.QualifiedObjectName;
 import com.facebook.presto.common.transaction.TransactionId;
@@ -30,7 +28,6 @@
 import com.facebook.presto.spi.connector.ConnectorTransactionHandle;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
-import com.google.common.collect.ImmutableMap;
 import com.google.inject.Guice;
 import com.google.inject.Injector;
 import org.testng.annotations.BeforeTest;
@@ -55,8 +52,7 @@ public class TestCatalogServerResponse
     public void setup()
     {
         this.testingCatalogServerClient = new TestingCatalogServerClient();
-        ConfigurationFactory configurationFactory = new ConfigurationFactory(ImmutableMap.of());
-        Injector injector = Guice.createInjector(new ConfigurationModule(configurationFactory), new JsonModule(), new TestingHandleJsonModule());
+        Injector injector = Guice.createInjector(new JsonModule(), new TestingHandleJsonModule());
         this.objectMapper = injector.getInstance(ObjectMapper.class);
     }
 
diff --git a/presto-main-base/src/test/java/com/facebook/presto/metadata/TestingHandleJsonModule.java b/presto-main-base/src/test/java/com/facebook/presto/metadata/TestingHandleJsonModule.java
@@ -13,23 +13,23 @@
  */
 package com.facebook.presto.metadata;
 
+import com.facebook.drift.codec.guice.ThriftCodecModule;
 import com.facebook.presto.connector.ConnectorManager;
 import com.facebook.presto.sql.analyzer.FeaturesConfig;
 import com.google.inject.Binder;
 import com.google.inject.Module;
 import com.google.inject.Scopes;
 
-import static com.facebook.airlift.configuration.ConfigBinder.configBinder;
-
 public class TestingHandleJsonModule
         implements Module
 {
     @Override
     public void configure(Binder binder)
     {
         binder.bind(ConnectorManager.class).toProvider(() -> null).in(Scopes.SINGLETON);
-        configBinder(binder).bindConfig(FeaturesConfig.class);
+        binder.bind(FeaturesConfig.class).toInstance(new FeaturesConfig());
 
+        binder.install(new ThriftCodecModule());
         binder.install(new HandleJsonModule());
     }
 }
diff --git a/presto-main-base/src/test/java/com/facebook/presto/sql/analyzer/TestFeaturesConfig.java b/presto-main-base/src/test/java/com/facebook/presto/sql/analyzer/TestFeaturesConfig.java
@@ -269,7 +269,8 @@ public void testDefaults()
                 .setInEqualityJoinPushdownEnabled(false)
                 .setRewriteMinMaxByToTopNEnabled(false)
                 .setPrestoSparkExecutionEnvironment(false)
-                .setMaxSerializableObjectSize(1000));
+                .setMaxSerializableObjectSize(1000)
+                .setUseConnectorProvidedSerializationCodecs(false));
     }
 
     @Test
@@ -486,6 +487,7 @@ public void testExplicitPropertyMappings()
                 .put("optimizer.utilize-unique-property-in-query-planning", "false")
                 .put("optimizer.add-exchange-below-partial-aggregation-over-group-id", "true")
                 .put("max_serializable_object_size", "50")
+                .put("use-connector-provided-serialization-codecs", "true")
                 .build();
 
         FeaturesConfig expected = new FeaturesConfig()
@@ -700,7 +702,8 @@ public void testExplicitPropertyMappings()
                 .setRewriteMinMaxByToTopNEnabled(true)
                 .setInnerJoinPushdownEnabled(true)
                 .setPrestoSparkExecutionEnvironment(true)
-                .setMaxSerializableObjectSize(50);
+                .setMaxSerializableObjectSize(50)
+                .setUseConnectorProvidedSerializationCodecs(true);
         assertFullMapping(properties, expected);
     }
 
diff --git a/presto-main-base/src/test/java/com/facebook/presto/sql/planner/plan/TestWindowNode.java b/presto-main-base/src/test/java/com/facebook/presto/sql/planner/plan/TestWindowNode.java
@@ -16,11 +16,13 @@
 import com.facebook.airlift.bootstrap.Bootstrap;
 import com.facebook.airlift.json.JsonCodec;
 import com.facebook.airlift.json.JsonModule;
+import com.facebook.drift.codec.guice.ThriftCodecModule;
 import com.facebook.presto.common.block.SortOrder;
 import com.facebook.presto.common.type.Type;
 import com.facebook.presto.common.type.TypeManager;
+import com.facebook.presto.connector.ConnectorManager;
 import com.facebook.presto.metadata.FunctionAndTypeManager;
-import com.facebook.presto.metadata.TestingHandleJsonModule;
+import com.facebook.presto.metadata.HandleJsonModule;
 import com.facebook.presto.server.SliceDeserializer;
 import com.facebook.presto.server.SliceSerializer;
 import com.facebook.presto.spi.VariableAllocator;
@@ -159,8 +161,10 @@ private JsonCodec<WindowNode> getJsonCodec()
             SqlParser sqlParser = new SqlParser();
             FunctionAndTypeManager functionAndTypeManager = createTestFunctionAndTypeManager();
             binder.install(new JsonModule());
+            binder.install(new ThriftCodecModule());
+            binder.install(new HandleJsonModule());
             configBinder(binder).bindConfig(FeaturesConfig.class);
-            binder.install(new TestingHandleJsonModule());
+            binder.bind(ConnectorManager.class).toProvider(() -> null);
             binder.bind(SqlParser.class).toInstance(sqlParser);
             binder.bind(TypeManager.class).toInstance(functionAndTypeManager);
             newSetBinder(binder, Type.class);
diff --git a/presto-main/src/test/java/com/facebook/presto/server/remotetask/TestHttpRemoteTask.java b/presto-main/src/test/java/com/facebook/presto/server/remotetask/TestHttpRemoteTask.java
@@ -36,6 +36,7 @@
 import com.facebook.presto.common.type.Type;
 import com.facebook.presto.common.type.TypeManager;
 import com.facebook.presto.connector.ConnectorCodecManager;
+import com.facebook.presto.connector.ConnectorManager;
 import com.facebook.presto.execution.Lifespan;
 import com.facebook.presto.execution.NodeTaskMap;
 import com.facebook.presto.execution.QueryManagerConfig;
@@ -56,7 +57,7 @@
 import com.facebook.presto.metadata.HandleResolver;
 import com.facebook.presto.metadata.InternalNode;
 import com.facebook.presto.metadata.Split;
-import com.facebook.presto.metadata.TestingHandleJsonModule;
+import com.facebook.presto.metadata.HandleJsonModule;
 import com.facebook.presto.server.InternalCommunicationConfig;
 import com.facebook.presto.server.TaskUpdateRequest;
 import com.facebook.presto.server.thrift.ConnectorSplitThriftCodec;
@@ -361,12 +362,13 @@ private static HttpRemoteTaskFactory createHttpRemoteTaskFactory(TestingTaskReso
                 new JsonModule(),
                 new SmileModule(),
                 new ThriftCodecModule(),
-                new TestingHandleJsonModule(),
+                new HandleJsonModule(),
                 new Module()
                 {
                     @Override
                     public void configure(Binder binder)
                     {
+                        binder.bind(ConnectorManager.class).toProvider(() -> null).in(Scopes.SINGLETON);
                         binder.bind(JsonMapper.class);
                         binder.bind(ThriftMapper.class);
                         configBinder(binder).bindConfig(FeaturesConfig.class);
diff --git a/presto-main/src/test/java/com/facebook/presto/server/remotetask/TestHttpRemoteTaskWithEventLoop.java b/presto-main/src/test/java/com/facebook/presto/server/remotetask/TestHttpRemoteTaskWithEventLoop.java
@@ -35,6 +35,7 @@
 import com.facebook.presto.common.type.Type;
 import com.facebook.presto.common.type.TypeManager;
 import com.facebook.presto.connector.ConnectorCodecManager;
+import com.facebook.presto.connector.ConnectorManager;
 import com.facebook.presto.execution.Lifespan;
 import com.facebook.presto.execution.NodeTaskMap;
 import com.facebook.presto.execution.QueryManagerConfig;
@@ -54,7 +55,7 @@
 import com.facebook.presto.metadata.HandleResolver;
 import com.facebook.presto.metadata.InternalNode;
 import com.facebook.presto.metadata.Split;
-import com.facebook.presto.metadata.TestingHandleJsonModule;
+import com.facebook.presto.metadata.HandleJsonModule;
 import com.facebook.presto.server.InternalCommunicationConfig;
 import com.facebook.presto.server.TaskUpdateRequest;
 import com.facebook.presto.server.thrift.ConnectorSplitThriftCodec;
@@ -369,12 +370,13 @@ private static HttpRemoteTaskFactory createHttpRemoteTaskFactory(TestingTaskReso
                 new JsonModule(),
                 new SmileModule(),
                 new ThriftCodecModule(),
-                new TestingHandleJsonModule(),
+                new HandleJsonModule(),
                 new Module()
                 {
                     @Override
                     public void configure(Binder binder)
                     {
+                        binder.bind(ConnectorManager.class).toProvider(() -> null).in(Scopes.SINGLETON);
                         binder.bind(JsonMapper.class);
                         binder.bind(ThriftMapper.class);
                         configBinder(binder).bindConfig(FeaturesConfig.class);

Original file line number	Diff line number	Diff line change
`@@ -13,23 +13,23 @@`
`13`	`13`	`*/`
`14`	`14`	`package com.facebook.presto.metadata;`
`15`	`15`
	`16`	`+import com.facebook.drift.codec.guice.ThriftCodecModule;`
`16`	`17`	`import com.facebook.presto.connector.ConnectorManager;`
`17`	`18`	`import com.facebook.presto.sql.analyzer.FeaturesConfig;`
`18`	`19`	`import com.google.inject.Binder;`
`19`	`20`	`import com.google.inject.Module;`
`20`	`21`	`import com.google.inject.Scopes;`
`21`	`22`
`22`		`-import static com.facebook.airlift.configuration.ConfigBinder.configBinder;`
`23`		`-`
`24`	`23`	`public class TestingHandleJsonModule`
`25`	`24`	`implements Module`
`26`	`25`	`{`
`27`	`26`	`@Override`
`28`	`27`	`public void configure(Binder binder)`
`29`	`28`	`{`
`30`	`29`	`binder.bind(ConnectorManager.class).toProvider(() -> null).in(Scopes.SINGLETON);`
`31`		`- configBinder(binder).bindConfig(FeaturesConfig.class);`
	`30`	`+ binder.bind(FeaturesConfig.class).toInstance(new FeaturesConfig());`
`32`	`31`
	`32`	`+ binder.install(new ThriftCodecModule());`
`33`	`33`	`binder.install(new HandleJsonModule());`
`34`	`34`	`}`
`35`	`35`	`}`