Fixes to CIDR ban model

Signed-off-by: Elizabeth Myers <[email protected]>
Elizafox · Mar 13, 2024 · dade023 · dade023
1 parent 458c55c
commit dade023
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/entity/src/cidr_ban.rs b/entity/src/cidr_ban.rs
@@ -11,8 +11,8 @@ pub struct Model {
     pub range_begin: Vec<u8>,
     #[sea_orm(column_type = "Binary(16)", unique)]
     pub range_end: Vec<u8>,
-    #[sea_orm(unique)]
-    pub reason: String,
+    pub reason: Option<String>,
+    pub created_at: DateTime,
     pub user_created_id: Option<i64>,
 }
 

diff --git a/migration/src/m20240312_184421_create_ip_filter_table.rs b/migration/src/m20240312_184421_create_ip_filter_table.rs
@@ -34,7 +34,7 @@ impl MigrationTrait for Migration {
                     )
                     .col(binary_len(CidrBan::RangeBegin, 16).unique_key().not_null())
                     .col(binary_len(CidrBan::RangeEnd, 16).unique_key().not_null())
-                    .col(string(CidrBan::Reason).unique_key().not_null())
+                    .col(string(CidrBan::Reason))
                     .col(
                         ColumnDef::new(CidrBan::CreatedAt)
                             .date_time()

diff --git a/service/src/mutation.rs b/service/src/mutation.rs
@@ -25,7 +25,7 @@ impl Mutation {
     pub async fn create_cidr_ban(
         db: &DbConn,
         network: IpNetwork,
-        reason: String,
+        reason: Option<String>,
         user: &user::Model,
     ) -> Result<cidr_ban::ActiveModel, DbErr> {
         let (start, end) = match network {