feat: enhance constraint handling by adding support for UNIQUE KEY in…

… diagram generation
Gausix · Feb 25, 2025 · 7f04080 · 7f04080
1 parent c9c722d
commit 7f04080
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/handlers/diagram_handlers.rs b/src/handlers/diagram_handlers.rs
@@ -119,7 +119,7 @@ impl DiagramHandlers {
         for line in &column_lines {
             let line_upper = line.to_uppercase();
 
-            if line_upper.starts_with("PRIMARY KEY") || line_upper.starts_with("FOREIGN KEY") || line_upper.starts_with("KEY") || line_upper.starts_with("CONSTRAINT") {
+            if line_upper.starts_with("PRIMARY KEY") || line_upper.starts_with("FOREIGN KEY") || line_upper.starts_with("KEY") || line_upper.starts_with("UNIQUE KEY") || line_upper.starts_with("CONSTRAINT") {
                 constraints.push(*line);
                 continue;
             }
@@ -157,6 +157,8 @@ impl DiagramHandlers {
                                 column.key = Some("FK".to_string());
                             } else if cons_line_upper.starts_with("KEY") && column.key.is_none() {
                                 column.key = Some("KEY".to_string());
+                            } else if cons_line_upper.starts_with("UNIQUE KEY") {
+                                column.key = Some("UNI".to_string());
                             }
                         }
                     }