Use switch expression in IcebergAvroFileWriter

trinodb · Jan 30, 2024 · 76bb60b · 76bb60b
1 parent 7e62f24
commit 76bb60b
Showing 1 changed file with 7 additions and 14 deletions.
diff --git a/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/IcebergAvroFileWriter.java b/plugin/trino-iceberg/src/main/java/io/trino/plugin/iceberg/IcebergAvroFileWriter.java
@@ -34,7 +34,6 @@
 import static io.trino.plugin.iceberg.IcebergAvroDataConversion.toIcebergRecords;
 import static io.trino.plugin.iceberg.IcebergErrorCode.ICEBERG_WRITER_CLOSE_ERROR;
 import static io.trino.plugin.iceberg.IcebergErrorCode.ICEBERG_WRITER_OPEN_ERROR;
-import static io.trino.spi.StandardErrorCode.GENERIC_INTERNAL_ERROR;
 import static java.util.Objects.requireNonNull;
 import static org.apache.iceberg.TableProperties.AVRO_COMPRESSION;
 
@@ -135,19 +134,13 @@ public long getValidationCpuNanos()
 
     private static String toIcebergAvroCompressionName(HiveCompressionCodec hiveCompressionCodec)
     {
-        switch (hiveCompressionCodec) {
-            case NONE:
-                return "UNCOMPRESSED";
-            case SNAPPY:
-                return "SNAPPY";
-            case LZ4:
-                return "LZ4";
-            case ZSTD:
-                return "ZSTD";
-            case GZIP:
-                return "GZIP";
-        }
-        throw new TrinoException(GENERIC_INTERNAL_ERROR, "Unexpected hiveCompressionCodec: " + hiveCompressionCodec);
+        return switch (hiveCompressionCodec) {
+            case NONE -> "UNCOMPRESSED";
+            case SNAPPY -> "SNAPPY";
+            case LZ4 -> "LZ4";
+            case ZSTD -> "ZSTD";
+            case GZIP -> "GZIP";
+        };
     }
 
     @Override