Reuse coercion on binary to varchar for non-orc partitioned table

trinodb · Nov 27, 2024 · 48d8464 · 48d8464
1 parent 124f296
commit 48d8464
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 34 deletions.
diff --git a/...n/trino-hive/src/main/java/io/trino/plugin/hive/coercions/VarbinaryToVarcharCoercers.java b/...n/trino-hive/src/main/java/io/trino/plugin/hive/coercions/VarbinaryToVarcharCoercers.java
@@ -26,7 +26,6 @@
 import java.util.HexFormat;
 
 import static io.trino.plugin.hive.HiveStorageFormat.ORC;
-import static io.trino.plugin.hive.HiveStorageFormat.PARQUET;
 import static io.trino.spi.type.VarbinaryType.VARBINARY;
 import static io.trino.spi.type.Varchars.truncateToLength;
 import static java.nio.charset.CodingErrorAction.REPLACE;
@@ -41,9 +40,6 @@ public static TypeCoercer<VarbinaryType, VarcharType> createVarbinaryToVarcharCo
         if (storageFormat == ORC) {
             return new OrcVarbinaryToVarcharCoercer(toType);
         }
-        if (storageFormat == PARQUET) {
-            return new ParquetVarbinaryToVarcharCoercer(toType);
-        }
         return new VarbinaryToVarcharCoercer(toType);
     }
 
@@ -55,31 +51,6 @@ public VarbinaryToVarcharCoercer(VarcharType toType)
             super(VARBINARY, toType);
         }
 
-        @Override
-        protected void applyCoercedValue(BlockBuilder blockBuilder, Block block, int position)
-        {
-            try {
-                Slice decodedValue = fromType.getSlice(block, position);
-                if (toType.isUnbounded()) {
-                    toType.writeSlice(blockBuilder, decodedValue);
-                    return;
-                }
-                toType.writeSlice(blockBuilder, truncateToLength(decodedValue, toType.getBoundedLength()));
-            }
-            catch (RuntimeException e) {
-                blockBuilder.appendNull();
-            }
-        }
-    }
-
-    private static class ParquetVarbinaryToVarcharCoercer
-            extends TypeCoercer<VarbinaryType, VarcharType>
-    {
-        public ParquetVarbinaryToVarcharCoercer(VarcharType toType)
-        {
-            super(VARBINARY, toType);
-        }
-
         @Override
         protected void applyCoercedValue(BlockBuilder blockBuilder, Block block, int position)
         {

diff --git a/...g/trino-product-tests/src/main/java/io/trino/tests/product/hive/BaseTestHiveCoercion.java b/...g/trino-product-tests/src/main/java/io/trino/tests/product/hive/BaseTestHiveCoercion.java
@@ -214,16 +214,12 @@ protected void doTestHiveCoercion(HiveTableDefinition tableDefinition)
 
         // Additional assertions for VARBINARY coercion
         if (prestoReadColumns.contains("binary_to_string")) {
-            List<Object> hexRepresentedValue = ImmutableList.of("58F7BFBFBF", "58F7BFBFBF58");
+            List<Object> hexRepresentedValue = ImmutableList.of("58EFBFBDEFBFBDEFBFBDEFBFBD", "58EFBFBDEFBFBDEFBFBDEFBFBD58");
 
             if (tableName.toLowerCase(ENGLISH).contains("orc")) {
                 hexRepresentedValue = ImmutableList.of("3538206637206266206266206266", "3538206637206266206266206266203538");
             }
 
-            if (tableName.toLowerCase(ENGLISH).contains("parquet")) {
-                hexRepresentedValue = ImmutableList.of("58EFBFBDEFBFBDEFBFBDEFBFBD", "58EFBFBDEFBFBDEFBFBDEFBFBD58");
-            }
-
             assertQueryResults(
                     Engine.TRINO,
                     format("SELECT to_hex(cast(binary_to_string as varbinary)) as hex_representation FROM %s", tableName),