Mark primary key columns as unique in sea-orm-codegen

SeaQL · May 24, 2024 · d9e278c · d9e278c
1 parent 2b6e724
commit d9e278c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sea-orm-codegen/src/entity/transformer.rs b/sea-orm-codegen/src/entity/transformer.rs
@@ -50,7 +50,7 @@ impl EntityTransformer {
                     col.unique = table_create
                         .get_indexes()
                         .iter()
-                        .filter(|index| index.is_unique_key())
+                        .filter(|index| index.is_unique_key() || index.is_primary_key())
                         .map(|index| index.get_index_spec().get_column_names())
                         .filter(|col_names| col_names.len() == 1 && col_names[0] == col.name)
                         .count()