From 57201788ef219e67b44b5b2d7574bc5658b3efee Mon Sep 17 00:00:00 2001
From: Billy Chan <ccw.billy.123@gmail.com>
Date: Mon, 20 Sep 2021 10:33:29 +0800
Subject: [PATCH] Codegen `TimestampWithTimeZone` fixup

---
 sea-orm-codegen/src/entity/column.rs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/sea-orm-codegen/src/entity/column.rs b/sea-orm-codegen/src/entity/column.rs
index 2bab8766..df286fc7 100644
--- a/sea-orm-codegen/src/entity/column.rs
+++ b/sea-orm-codegen/src/entity/column.rs
@@ -43,7 +43,6 @@ impl Column {
             ColumnType::Uuid => "Uuid",
             ColumnType::Binary(_) => "Vec<u8>",
             ColumnType::Boolean => "bool",
-            _ => unimplemented!(),
         }
         .parse()
         .unwrap();
@@ -91,6 +90,7 @@ impl Column {
             },
             ColumnType::DateTime(_) => quote! { ColumnType::DateTime.def() },
             ColumnType::Timestamp(_) => quote! { ColumnType::Timestamp.def() },
+            ColumnType::TimestampWithTimeZone(_) => quote! { ColumnType::TimestampWithTimeZone.def() },
             ColumnType::Time(_) => quote! { ColumnType::Time.def() },
             ColumnType::Date => quote! { ColumnType::Date.def() },
             ColumnType::Binary(_) => quote! { ColumnType::Binary.def() },
@@ -106,7 +106,6 @@ impl Column {
                 let s = s.to_string();
                 quote! { ColumnType::Custom(#s.to_owned()).def() }
             }
-            _ => unimplemented!(),
         };
         if !self.not_null {
             col_def.extend(quote! {