Fix such that there is a rollback() before LogicalLock unlock()

When there is an error during migration and using logical lock
then ensure there is a rollback before the unlock() on the logical
lock.

Author: rbygrave

ebean-migration/src/main/java/io/ebean/migration/runner/MigrationEngine.java

@@ -5,7 +5,6 @@
 import io.ebean.migration.MigrationException;
 import io.ebean.migration.MigrationResource;
 
-import java.io.IOException;
 import java.sql.Connection;
 import java.sql.SQLException;
 import java.util.Collections;
@@ -44,7 +43,6 @@ public List<MigrationResource> run(Connection connection) {
       }
 
       long startMs = System.currentTimeMillis();
-      connection.setAutoCommit(false);
       MigrationTable table = initialiseMigrationTable(connection);
       try {
         List<MigrationResource> result = runMigrations(resources.versions(), table, checkStateOnly);
@@ -58,30 +56,31 @@ public List<MigrationResource> run(Connection connection) {
           }
         }
         return result;
+      } catch (Exception e) {
+        log.log(ERROR, "Perform rollback due to DB migration error", e);
+        rollback(connection);
+        throw new MigrationException("Error running DB migrations", e);
       } finally {
         table.unlockMigrationTable();
       }
-
-    } catch (MigrationException e) {
-      rollback(connection);
-      throw e;
-
-    } catch (Exception e) {
-      rollback(connection);
-      throw new MigrationException("Error running DB migrations", e);
-
     } finally {
       close(connection);
     }
   }
 
-  private MigrationTable initialiseMigrationTable(Connection connection) throws SQLException, IOException {
-    MigrationPlatform platform = derivePlatformName(migrationConfig, connection);
-    new MigrationSchema(migrationConfig, connection).createAndSetIfNeeded();
+  private MigrationTable initialiseMigrationTable(Connection connection) {
+    try {
+      connection.setAutoCommit(false);
+      MigrationPlatform platform = derivePlatformName(migrationConfig, connection);
+      new MigrationSchema(migrationConfig, connection).createAndSetIfNeeded();
 
-    final MigrationTable table = new MigrationTable(migrationConfig, connection, checkStateOnly, platform);
-    table.createIfNeededAndLock();
-    return table;
+      final MigrationTable table = new MigrationTable(migrationConfig, connection, checkStateOnly, platform);
+      table.createIfNeededAndLock();
+      return table;
+    } catch (Exception e) {
+      rollback(connection);
+      throw new MigrationException("Error initialising db migrations table", e);
+    }
   }
 
   /**

ebean-migration/src/main/java/io/ebean/migration/runner/MigrationPlatform.java

@@ -1,6 +1,7 @@
 package io.ebean.migration.runner;
 
 import io.ebean.ddlrunner.DdlDetect;
+import io.ebean.migration.MigrationException;
 
 import java.sql.*;
 import java.util.ArrayList;
@@ -30,7 +31,7 @@ DdlDetect ddlDetect() {
     return DdlDetect.NONE;
   }
 
-  void unlockMigrationTable(String sqlTable, Connection connection) throws SQLException {
+  void unlockMigrationTable(String sqlTable, Connection connection) {
     // do nothing by default for select for update case
   }
 
@@ -118,9 +119,14 @@ void lockMigrationTable(String sqlTable, Connection connection) throws SQLExcept
     }
 
     @Override
-    void unlockMigrationTable(String sqlTable, Connection connection) throws SQLException {
-      releaseLogicalLock(sqlTable, connection);
-      connection.commit();
+    void unlockMigrationTable(String sqlTable, Connection connection) {
+      try {
+        releaseLogicalLock(sqlTable, connection);
+        connection.commit();
+      } catch (SQLException e) {
+        rollback(connection);
+        throw new MigrationException("Error releasing logical lock for ebean migrations");
+      }
     }
 
     private boolean obtainLogicalLock(String sqlTable, Connection connection) throws SQLException {
@@ -190,11 +196,16 @@ private boolean obtainNamedLock(Connection connection) throws SQLException {
       return false;
     }
 
+    @SuppressWarnings("all")
     @Override
-    void unlockMigrationTable(String sqlTable, Connection connection) throws SQLException {
-      String hash = Integer.toHexString(connection.getMetaData().getURL().hashCode());
-      try (Statement query = connection.createStatement()) {
-        query.execute("select release_lock('ebean_migration-" + hash + "')");
+    void unlockMigrationTable(String sqlTable, Connection connection) {
+      try {
+        String hash = Integer.toHexString(connection.getMetaData().getURL().hashCode());
+        try (Statement query = connection.createStatement()) {
+          query.execute("select release_lock('ebean_migration-" + hash + "')");
+        }
+      } catch (SQLException e) {
+        throw new MigrationException("Error releasing lock for ebean_migration", e);
       }
     }
   }

ebean-migration/src/main/java/io/ebean/migration/runner/MigrationTable.java

@@ -176,7 +176,7 @@ private void obtainLockWithWait() throws SQLException {
   /**
    * Release a lock on the migration table (MySql, MariaDB only).
    */
-  void unlockMigrationTable() throws SQLException {
+  void unlockMigrationTable() {
     platform.unlockMigrationTable(sqlTable, connection);
   }