diff --git a/src/main/java/io/ebean/dbmigration/DbMigration.java b/src/main/java/io/ebean/dbmigration/DbMigration.java index 505664739..0680e06af 100644 --- a/src/main/java/io/ebean/dbmigration/DbMigration.java +++ b/src/main/java/io/ebean/dbmigration/DbMigration.java @@ -6,6 +6,8 @@ import io.ebean.config.ServerConfig; import io.ebean.config.dbplatform.DatabasePlatform; import java.io.IOException; +import java.util.Iterator; +import java.util.ServiceLoader; /** * Generates DDL migration scripts based on changes to the current model. @@ -39,7 +41,12 @@ public interface DbMigration { * Create a DbMigration implementation to use. */ static DbMigration create() { - return DbMigrationFactory.create(); + + Iterator loader = ServiceLoader.load(DbMigration.class).iterator(); + if (loader.hasNext()) { + return loader.next(); + } + throw new IllegalStateException("No service implementation found for DbMigration?"); } /** diff --git a/src/main/java/io/ebean/dbmigration/DbMigrationFactory.java b/src/main/java/io/ebean/dbmigration/DbMigrationFactory.java deleted file mode 100644 index 4256af1f4..000000000 --- a/src/main/java/io/ebean/dbmigration/DbMigrationFactory.java +++ /dev/null @@ -1,16 +0,0 @@ -package io.ebean.dbmigration; - -import java.util.Iterator; -import java.util.ServiceLoader; - -class DbMigrationFactory { - - static DbMigration create() { - - Iterator loader = ServiceLoader.load(DbMigration.class).iterator(); - if (loader.hasNext()) { - return loader.next(); - } - throw new IllegalStateException("No service implementation found for DbMigration?"); - } -}