Prevent deadlock when importing database (#8648)

* lock all tables at beginning of transaction to avoid deadlocks * fix lock statement format
airbytehq · Dec 8, 2021 · 973c043 · 973c043
1 parent 07287ca
commit 973c043
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...ler/persistence/src/main/java/io/airbyte/scheduler/persistence/DefaultJobPersistence.java b/...ler/persistence/src/main/java/io/airbyte/scheduler/persistence/DefaultJobPersistence.java
@@ -640,6 +640,10 @@ private void importDatabase(final String airbyteVersion,
     if (!data.isEmpty()) {
       createSchema(BACKUP_SCHEMA);
       jobDatabase.transaction(ctx -> {
+        // obtain locks on all tables first, to prevent deadlocks
+        for (final JobsDatabaseSchema tableType : data.keySet()) {
+          ctx.execute(String.format("LOCK TABLE %s IN ACCESS EXCLUSIVE MODE", tableType.name()));
+        }
         for (final JobsDatabaseSchema tableType : data.keySet()) {
           if (!incrementalImport) {
             truncateTable(ctx, targetSchema, tableType.name(), BACKUP_SCHEMA);