Use flags to detect MySQL enums

In some cases the returned value is of `ColumnType::String`, but it has the `ENUM` flag set.
launchbadge · Oct 13, 2020 · 7a70717 · 7a70717
1 parent 277d041
commit 7a70717
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sqlx-core/src/mysql/protocol/text/column.rs b/sqlx-core/src/mysql/protocol/text/column.rs
@@ -167,6 +167,7 @@ impl ColumnType {
     ) -> &'static str {
         let is_binary = char_set == 63;
         let is_unsigned = flags.contains(ColumnFlags::UNSIGNED);
+        let is_enum = flags.contains(ColumnFlags::ENUM);
 
         match self {
             ColumnType::Tiny if max_size == Some(1) => "BOOLEAN",
@@ -196,6 +197,7 @@ impl ColumnType {
             ColumnType::Json => "JSON",
 
             ColumnType::String if is_binary => "BINARY",
+            ColumnType::String if is_enum => "ENUM",
             ColumnType::VarChar | ColumnType::VarString if is_binary => "VARBINARY",
 
             ColumnType::String => "CHAR",