fix(database): set correct collection id column type

Fixes #855.
IgnisDa · Jun 3, 2024 · 03d1905 · 03d1905
1 parent 7b21d13
commit 03d1905
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 7 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/apps/backend/Cargo.toml b/apps/backend/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "ryot"
-version = "5.5.4"
+version = "5.5.5"
 edition = "2021"
 repository = "https://github.com/IgnisDa/ryot"
 license = "GPL-3.0"

diff --git a/libs/database/src/migrations/m20240509_create_user_to_collection.rs b/libs/database/src/migrations/m20240509_create_user_to_collection.rs
@@ -1,5 +1,7 @@
 use sea_orm_migration::prelude::*;
 
+use crate::get_whether_column_is_text;
+
 use super::{m20230417_create_user::User, m20230504_create_collection::Collection};
 
 #[derive(DeriveMigrationName)]
@@ -15,16 +17,19 @@ pub enum UserToCollection {
 #[async_trait::async_trait]
 impl MigrationTrait for Migration {
     async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+        let db = manager.get_connection();
+        let mut base = ColumnDef::new(UserToCollection::CollectionId);
+        let collection_column = if get_whether_column_is_text("collection", "id", db).await? {
+            base.text()
+        } else {
+            base.integer()
+        };
         manager
             .create_table(
                 Table::create()
                     .table(UserToCollection::Table)
                     .col(ColumnDef::new(UserToCollection::UserId).integer())
-                    .col(
-                        ColumnDef::new(UserToCollection::CollectionId)
-                            .text()
-                            .not_null(),
-                    )
+                    .col(collection_column.not_null())
                     .foreign_key(
                         ForeignKey::create()
                             .name("user_to_collection-fk1")