added support for querying views in PostgreSQL

albertocsm · Dec 14, 2015 · 0a75bb2 · 0a75bb2
1 parent df2e649
commit 0a75bb2
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/presto-postgresql/src/main/java/com/facebook/presto/plugin/postgresql/PostgreSqlClient.java b/presto-postgresql/src/main/java/com/facebook/presto/plugin/postgresql/PostgreSqlClient.java
@@ -29,6 +29,8 @@
 import com.google.common.base.Throwables;
 import io.airlift.slice.Slice;
 
+import java.sql.DatabaseMetaData;
+import java.sql.ResultSet;
 import java.sql.Types;
 
 import org.postgresql.Driver;
@@ -117,4 +119,17 @@ protected Type toPrestoType(int jdbcType)
             return VARBINARY;
         }
     }
+
+    @Override
+    protected ResultSet getTables(Connection connection, String schemaName, String tableName)
+        throws SQLException
+    {
+      DatabaseMetaData metadata = connection.getMetaData();
+      String escape = metadata.getSearchStringEscape();
+      return metadata.getTables(
+          connection.getCatalog(),
+          escapeNamePattern(schemaName, escape),
+          escapeNamePattern(tableName, escape),
+          new String[] {"TABLE", "VIEW"});
+    }
 }