Add getColumnTypes API to RecordSink

gerashegalov · Dec 22, 2014 · 4770f04 · 4770f04
1 parent 0511e6b
commit 4770f04
Show file tree

Hide file tree

Showing 5 changed files with 30 additions and 0 deletions.
diff --git a/presto-base-jdbc/src/main/java/com/facebook/presto/plugin/jdbc/JdbcRecordSink.java b/presto-base-jdbc/src/main/java/com/facebook/presto/plugin/jdbc/JdbcRecordSink.java
@@ -171,6 +171,12 @@ public String commit()
  return ""; // the committer does not need any additional info
  }
 
+ @Override
+ public List<Type> getColumnTypes()
+ {
+ return columnTypes;
+ }
+
  private int next()
  {
  checkState(field != -1, "not in record");

diff --git a/presto-cassandra/src/main/java/com/facebook/presto/cassandra/CassandraRecordSink.java b/presto-cassandra/src/main/java/com/facebook/presto/cassandra/CassandraRecordSink.java
@@ -148,6 +148,12 @@ public String commit()
  return "";
  }
 
+ @Override
+ public List<Type> getColumnTypes()
+ {
+ return columnTypes;
+ }
+
  private void append(Object value)
  {
  checkState(field != -1, "not in record");

diff --git a/presto-hive/src/main/java/com/facebook/presto/hive/HiveRecordSink.java b/presto-hive/src/main/java/com/facebook/presto/hive/HiveRecordSink.java
@@ -205,6 +205,12 @@ public String commit()
  return ""; // the committer can list the directory
  }
 
+ @Override
+ public List<Type> getColumnTypes()
+ {
+ return columnTypes;
+ }
+
  private void append(Object value)
  {
  checkState(field != -1, "not in record");

diff --git a/presto-raptor/src/main/java/com/facebook/presto/raptor/RaptorRecordSink.java b/presto-raptor/src/main/java/com/facebook/presto/raptor/RaptorRecordSink.java
@@ -151,6 +151,12 @@ public String commit()
  return Joiner.on(':').join(nodeId, outputHandle.getShardUuid());
  }
 
+ @Override
+ public List<Type> getColumnTypes()
+ {
+ return columnTypes;
+ }
+
  private Type currentType()
  {
  return columnTypes.get(rowSink.currentField());

diff --git a/presto-spi/src/main/java/com/facebook/presto/spi/RecordSink.java b/presto-spi/src/main/java/com/facebook/presto/spi/RecordSink.java
@@ -13,6 +13,10 @@
  */
 package com.facebook.presto.spi;
 
+import com.facebook.presto.spi.type.Type;
+
+import java.util.List;
+
 public interface RecordSink
 {
  /**
@@ -34,4 +38,6 @@ public interface RecordSink
  void appendString(byte[] value);
 
  String commit();
+
+ List<Type> getColumnTypes();
 }