Merge pull request #37 from tzzzoz/main

Support partition key in type: timestamp with time zone
qonto · Sep 5, 2024 · 8ed3d4d · 8ed3d4d
2 parents 2b19d8c + 84314a3
commit 8ed3d4d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/internal/infra/postgresql/column.go b/internal/infra/postgresql/column.go
@@ -39,6 +39,8 @@ func (p Postgres) GetColumnDataType(schema, table, column string) (ColumnType, e
 		return DateTime, nil
 	case "timestamp without time zone":
 		return DateTime, nil
+	case "timestamp with time zone":
+		return DateTime, nil
 	case "uuid":
 		return UUID, nil
 	default:

diff --git a/internal/infra/postgresql/column_internal_test.go b/internal/infra/postgresql/column_internal_test.go
@@ -47,6 +47,11 @@ func TestGetColumn(t *testing.T) {
 			"timestamp without time zone",
 			DateTime,
 		},
+		{
+			"Date time with time zone",
+			"timestamp with time zone",
+			DateTime,
+		},
 		{
 			"UUID",
 			"uuid",