clear vectors in close

Author: tianchen92

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroBooleanConsumer.java

@@ -63,4 +63,9 @@ public FieldVector getVector() {
     return this.vector;
   }
 
+  @Override
+  public void close() throws Exception {
+    writer.close();
+  }
+
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroBytesConsumer.java

@@ -81,9 +81,7 @@ public FieldVector getVector() {
   }
 
   @Override
-  public void close() {
-    if (cacheBuffer != null) {
-      cacheBuffer.clear();
-    }
+  public void close() throws Exception {
+    writer.close();
   }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroDoubleConsumer.java

@@ -62,4 +62,9 @@ public void setPosition(int index) {
   public FieldVector getVector() {
     return vector;
   }
+
+  @Override
+  public void close() throws Exception {
+    writer.close();
+  }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroFloatConsumer.java

@@ -62,4 +62,9 @@ public void setPosition(int index) {
   public FieldVector getVector() {
     return this.vector;
   }
+
+  @Override
+  public void close() throws Exception {
+    writer.close();
+  }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroIntConsumer.java

@@ -62,4 +62,9 @@ public void setPosition(int index) {
   public FieldVector getVector() {
     return this.vector;
   }
+
+  @Override
+  public void close() throws Exception {
+    writer.close();
+  }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroLongConsumer.java

@@ -62,4 +62,9 @@ public void setPosition(int index) {
   public FieldVector getVector() {
     return this.vector;
   }
+
+  @Override
+  public void close() throws Exception {
+    writer.close();
+  }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroNullConsumer.java

@@ -48,4 +48,9 @@ public void setPosition(int index) {}
   public FieldVector getVector() {
     return this.vector;
   }
+
+  @Override
+  public void close() {
+    vector.close();
+  }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroStringConsumer.java

@@ -82,9 +82,7 @@ public FieldVector getVector() {
   }
 
   @Override
-  public void close() {
-    if (cacheBuffer != null) {
-      cacheBuffer.clear();
-    }
+  public void close() throws Exception {
+    writer.close();
   }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/AvroUnionsConsumer.java

@@ -82,9 +82,10 @@ public FieldVector getVector() {
   }
 
   @Override
-  public void close() {
-    for (Consumer consumer : delegates) {
-      consumer.close();
+  public void close() throws Exception {
+    writer.close();
+    for (Consumer delegate: delegates) {
+      delegate.close();
     }
   }
 }

java/adapter/avro/src/main/java/org/apache/arrow/consumers/CompositeAvroConsumer.java

@@ -59,7 +59,11 @@ public void consume(Decoder decoder, VectorSchemaRoot root) throws IOException {
   public void close() {
     // clean up
     for (Consumer consumer : consumers) {
-      consumer.close();
+      try {
+        consumer.close();
+      } catch (Exception e) {
+        throw new RuntimeException("Error occurs in close.", e);
+      }
     }
   }
 }