From fac1031de1aebb090e5f6e615697f15184a4a341 Mon Sep 17 00:00:00 2001 From: Matt Corallo Date: Sun, 16 Jul 2023 17:25:10 +0000 Subject: [PATCH] Correct upgrade logic to ignore missing indexes If we upgrade multiple versions in one go, the indexes we're trying to drop may not exist, and that shouldn't cause us to fail to upgrade. --- src/config.rs | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/src/config.rs b/src/config.rs index 6249940..b0e3a7c 100644 --- a/src/config.rs +++ b/src/config.rs @@ -213,31 +213,31 @@ pub(crate) async fn upgrade_db(schema: i32, client: &mut tokio_postgres::Client) } if schema >= 1 && schema <= 7 { let tx = client.transaction().await.unwrap(); - tx.execute("DROP INDEX channels_seen", &[]).await.unwrap(); - tx.execute("DROP INDEX channel_updates_scid", &[]).await.unwrap(); - tx.execute("DROP INDEX channel_updates_direction", &[]).await.unwrap(); - tx.execute("DROP INDEX channel_updates_seen", &[]).await.unwrap(); - tx.execute("DROP INDEX channel_updates_scid_seen", &[]).await.unwrap(); - tx.execute("DROP INDEX channel_updates_scid_dir_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channels_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_scid", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_direction", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_scid_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_scid_dir_seen", &[]).await.unwrap(); tx.execute("UPDATE config SET db_schema = 8 WHERE id = 1", &[]).await.unwrap(); tx.commit().await.unwrap(); } if schema >= 1 && schema <= 8 { let tx = client.transaction().await.unwrap(); - tx.execute("DROP INDEX channel_updates_seen", &[]).await.unwrap(); - tx.execute("DROP INDEX channel_updates_scid_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_scid_seen", &[]).await.unwrap(); tx.execute("UPDATE config SET db_schema = 9 WHERE id = 1", &[]).await.unwrap(); tx.commit().await.unwrap(); } if schema >= 1 && schema <= 9 { let tx = client.transaction().await.unwrap(); - tx.execute("DROP INDEX channel_updates_scid_dir_seen", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_scid_dir_seen", &[]).await.unwrap(); tx.execute("UPDATE config SET db_schema = 10 WHERE id = 1", &[]).await.unwrap(); tx.commit().await.unwrap(); } if schema >= 1 && schema <= 10 { let tx = client.transaction().await.unwrap(); - tx.execute("DROP INDEX channel_updates_id_with_scid_dir_blob", &[]).await.unwrap(); + tx.execute("DROP INDEX IF EXISTS channel_updates_id_with_scid_dir_blob", &[]).await.unwrap(); tx.execute("UPDATE config SET db_schema = 11 WHERE id = 1", &[]).await.unwrap(); tx.commit().await.unwrap(); } -- 2.39.5