com.github.tototoshi.play2.flyway.UrlParser.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of play-flyway_2.11 Show documentation

play-flyway

There is a newer version: 1.2.2

package com.github.tototoshi.play2.flyway

import play.api.{ Mode, Play }

/**
 * Most of the code is taken from package play.api.db.DB.
 */
trait UrlParser {
  val PostgresFullUrl = "^postgres://([a-zA-Z0-9_]+):([^@]+)@([^/]+)/([^\\s]+)$".r
  val MysqlFullUrl = "^mysql://([a-zA-Z0-9_]+):([^@]+)@([^/]+)/([^\\s]+)$".r
  val MysqlCustomProperties = ".*\\?(.*)".r
  val H2DefaultUrl = "^jdbc:h2:mem:.+".r

  def parseUrl(url: String): Tuple3[String, Option[String], Option[String]] = {

    url match {
      case PostgresFullUrl(username, password, host, dbname) =>
        ("jdbc:postgresql://%s/%s".format(host, dbname), Some(username), Some(password))
      case url @ MysqlFullUrl(username, password, host, dbname) =>
        val defaultProperties = """?useUnicode=yes&characterEncoding=UTF-8&connectionCollation=utf8_general_ci"""
        val addDefaultPropertiesIfNeeded = MysqlCustomProperties.findFirstMatchIn(url).map(_ => "").getOrElse(defaultProperties)
        ("jdbc:mysql://%s/%s".format(host, dbname + addDefaultPropertiesIfNeeded), Some(username), Some(password))
      case url @ H2DefaultUrl() if !url.contains("DB_CLOSE_DELAY") =>
        val jdbcUrl = if (Play.maybeApplication.exists(_.mode == Mode.Dev)) {
          url + ";DB_CLOSE_DELAY=-1"
        } else {
          url
        }
        (jdbcUrl, None, None)
      case s: String =>
        (s, None, None)
    }

  }
}