Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
package tv.spotx.scala.dbutils
import java.sql.{Connection, DriverManager}
import java.util.Properties
import org.apache.commons.pool2.impl.{DefaultPooledObject, GenericObjectPool}
import org.apache.commons.pool2.{BasePooledObjectFactory, PooledObject}
import org.apache.logging.log4j.scala.Logging
// Used for MySQL Connection
class ConnectionFactory(jdbc: String) extends BasePooledObjectFactory[Connection] {
def create(): Connection = {
DriverManager.getConnection(jdbc)
}
def wrap(connection: Connection): PooledObject[Connection] = {
new DefaultPooledObject[Connection](connection)
}
}
case class ConnectionPool(jdbc: String) extends Logging{
logger.info(s"Generating connection pool for: ${jdbc}")
val connectionPool = new GenericObjectPool[Connection](new ConnectionFactory(jdbc))
def getConnection: Connection = {
connectionPool.borrowObject
}
def returnConnection(connection: Connection): Unit = {
connectionPool.returnObject(connection)
}
}
case class MySQLConnection(
host: String,
port: Option[Int] = Option(3306), // scalastyle:off magic.number
database: String,
table: String,
username: String,
password: String,
options: Map[String, String] = Map[String,String]()
) {
override def toString: String = {
s"jdbc:mysql://$host" +
(port match { case Some(p) => s":$p" case None => "" }) +
s"/$database?user=$username&password=$password" +
options.map(x => s"&${x._1}=${x._2}").mkString("")
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.