import scala.concurrent.duration.Duration
import scala.concurrent.{Await, ExecutionContext}

import slick.dbio.DBIO

import org.flywaydb.core.api.migration.{BaseJavaMigration, Context}


abstract class SlickMigration extends BaseJavaMigration {
  implicit val executionContext: ExecutionContext = ExecutionContext.global

  def action: DBIO[Unit]

  override def migrate(context: Context): Unit = {
    val db = new UnmanagedDatabase(context.getConnection)
    Await.result(db.run(action), Duration.Inf)
  }
}