From 26d16876d424c8134ecf610a772ebf32bb179096 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Wed, 1 May 2024 18:16:29 -0400
Subject: [PATCH] migrations

---
 .../migrate_media_files_enum_values.rs         | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/crates/cli/production/db-cleanup/src/operations/migrate_media_files_enum_values/migrate_media_files_enum_values.rs b/crates/cli/production/db-cleanup/src/operations/migrate_media_files_enum_values/migrate_media_files_enum_values.rs
index bc45a2c2f9..fc9d8b527a 100644
--- a/crates/cli/production/db-cleanup/src/operations/migrate_media_files_enum_values/migrate_media_files_enum_values.rs
+++ b/crates/cli/production/db-cleanup/src/operations/migrate_media_files_enum_values/migrate_media_files_enum_values.rs
@@ -53,7 +53,7 @@ pub async fn migrate_media_files_enum_values(_args: &Args, mysql: &Pool<MySql>)
   //  |     20570300 |
   //  +--------------+
   //  1 row in set (42.86 sec)
-  let query_pair = QueryPair {
+  let _query_pair = QueryPair {
     count_query: r#"
       select count(*) as record_count
       from media_files
@@ -69,6 +69,22 @@ pub async fn migrate_media_files_enum_values(_args: &Args, mysql: &Pool<MySql>)
     "#.to_string(),
   };
 
+  let query_pair = QueryPair {
+    count_query: r#"
+      select count(*) as record_count
+      from media_files
+      where media_class = "unknown"
+      and media_type = "video"
+    "#.to_string(),
+    migrate_query: r#"
+      update media_files
+      set media_class = "video"
+      where media_class = "unknown"
+      and media_type = "video"
+      limit 100000
+    "#.to_string(),
+  };
+
   query_pair.run_migration(mysql).await?;
 
   Ok(())