ORC-562:Don't wrap readerSchema in acidSchema, if readerSchema is alr…

…eady acid Fixes apache#442 Signed-off-by: Owen O'Malley <omalley@apache.org>
abstractdog · Nov 5, 2019 · 2db32a8 · 2db32a8
1 parent a6e7f5f
commit 2db32a8
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/java/core/src/java/org/apache/orc/impl/SchemaEvolution.java b/java/core/src/java/org/apache/orc/impl/SchemaEvolution.java
@@ -86,10 +86,11 @@ public SchemaEvolution(TypeDescription fileSchema,
     this.isOnlyImplicitConversion = true;
     this.fileSchema = fileSchema;
     this.isAcid = checkAcidSchema(fileSchema);
+    boolean readerSchemaIsAcid = readerSchema == null ? false : checkAcidSchema(readerSchema);
     this.includeAcidColumns = options.getIncludeAcidColumns();
-    this.readerColumnOffset = isAcid ? acidEventFieldNames.size() : 0;
+    this.readerColumnOffset = isAcid && !readerSchemaIsAcid ? acidEventFieldNames.size() : 0;
     if (readerSchema != null) {
-      if (isAcid) {
+      if (isAcid && !readerSchemaIsAcid) {
         this.readerSchema = createEventSchema(readerSchema);
       } else {
         this.readerSchema = readerSchema;

diff --git a/java/core/src/test/org/apache/orc/impl/TestSchemaEvolution.java b/java/core/src/test/org/apache/orc/impl/TestSchemaEvolution.java
@@ -2381,4 +2381,24 @@ public void testCheckAcidSchema() {
     assertTrue("Schema (" + ccSchema +") was found to be non-acid ", evoCc.isAcid());
     assertTrue("Schema (" + lcSchema +") was found to be non-acid ", evoLc.isAcid());
   }
+
+  @Test
+  public void testAcidReaderSchema() {
+    String acidSchema = "struct<operation:int,originalTransaction:bigint,bucket:int," +
+        "rowId:bigint,currentTransaction:bigint," +
+        "row:struct<a:int,b:int>>";
+
+    TypeDescription fileSchema = TypeDescription.fromString(acidSchema);
+    TypeDescription readerSchema = TypeDescription.fromString(acidSchema);
+    SchemaEvolution schemaEvolution = new SchemaEvolution(fileSchema, readerSchema, options);
+
+    assertEquals(String.format("Reader schema %s is not acid", schemaEvolution.getReaderSchema().toString()),
+        acidSchema, schemaEvolution.getReaderSchema().toString());
+
+    String notAcidSchema ="struct<a:int,b:int>";
+    readerSchema = TypeDescription.fromString(notAcidSchema);
+    schemaEvolution = new SchemaEvolution(fileSchema, readerSchema, options);
+    assertEquals(String.format("Reader schema %s is not acid", schemaEvolution.getReaderSchema().toString()),
+        acidSchema, schemaEvolution.getReaderSchema().toString());
+  }
 }