Infer the domain types type as the underlying type

Fixes singer-io#97
Limess · Jul 19, 2020 · 3419067 · 3419067
1 parent e990e88
commit 3419067
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tap_postgres/__init__.py b/tap_postgres/__init__.py
@@ -235,7 +235,6 @@ def produce_table_info(conn):
     with conn.cursor(cursor_factory=psycopg2.extras.DictCursor, name='stitch_cursor') as cur:
         cur.itersize = post_db.cursor_iter_size
         table_info = {}
-          # SELECT CASE WHEN $2.typtype = 'd' THEN $2.typbasetype ELSE $1.atttypid END
         cur.execute("""
 SELECT
   pg_class.reltuples::BIGINT                            AS approximate_row_count,
@@ -244,7 +243,7 @@ def produce_table_info(conn):
   pg_class.relname                                      AS table_name,
   attname                                               AS column_name,
   i.indisprimary                                        AS primary_key,
-  format_type(a.atttypid, NULL::integer)                AS data_type,
+  format_type(CASE WHEN pgt.typtype = 'd' THEN pgt.typbasetype ELSE a.atttypid END, NULL::integer) AS data_type,
   information_schema._pg_char_max_length(CASE WHEN COALESCE(subpgt.typtype, pgt.typtype) = 'd'
                                               THEN COALESCE(subpgt.typbasetype, pgt.typbasetype) ELSE COALESCE(subpgt.oid, pgt.oid)
                                           END,