Support citext data type

sqldef · Mar 12, 2021 · 1bc2ed9 · 1bc2ed9
1 parent ea5603d
commit 1bc2ed9
Show file tree

Hide file tree

Showing 5 changed files with 2,384 additions and 2,358 deletions.
diff --git a/adapter/postgres/postgres.go b/adapter/postgres/postgres.go
@@ -200,7 +200,7 @@ func (c *column) GetDataType() string {
 
 func (d *PostgresDatabase) getColumns(table string) ([]column, error) {
 	query := `SELECT s.column_name, s.column_default, s.is_nullable, s.character_maximum_length,
-	CASE WHEN s.data_type = 'ARRAY' THEN format_type(f.atttypid, f.atttypmod) ELSE s.data_type END,
+	CASE WHEN s.data_type IN ('ARRAY', 'USER-DEFINED') THEN format_type(f.atttypid, f.atttypmod) ELSE s.data_type END,
 	CASE WHEN p.contype = 'u' THEN true ELSE false END AS uniquekey,
 	CASE WHEN pc.contype = 'c' THEN pg_get_constraintdef(pc.oid, true) ELSE NULL END AS check
 FROM pg_attribute f

diff --git a/cmd/psqldef/psqldef_test.go b/cmd/psqldef/psqldef_test.go
@@ -86,6 +86,23 @@ func TestPsqldefCreateTableAlterColumn(t *testing.T) {
 	assertApplyOutput(t, createTable, nothingModified)
 }
 
+func TestPsqldefCitextExtension(t *testing.T) {
+	resetTestDatabase()
+	mustExecute("psql", "-Upostgres", "psqldef_test", "-c", "CREATE EXTENSION citext;")
+
+	createTable := stripHeredoc(`
+		CREATE TABLE users (
+		  name citext
+		);
+		`,
+	)
+	assertApplyOutput(t, createTable, applyPrefix+createTable)
+	assertApplyOutput(t, createTable, nothingModified)
+
+	mustExecute("psql", "-Upostgres", "psqldef_test", "-c", "DROP TABLE users;")
+	mustExecute("psql", "-Upostgres", "psqldef_test", "-c", "DROP EXTENSION citext;")
+}
+
 func TestPsqldefIgnoreExtension(t *testing.T) {
 	resetTestDatabase()
 	mustExecute("psql", "-Upostgres", "psqldef_test", "-c", "CREATE EXTENSION pg_buffercache;")