scalikejdbc.async.internal.postgresql.PostgreSQLConnectionPoolImpl.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalikejdbc-async_2.11 Show documentation

scalikejdbc-async

There is a newer version: 0.11.0

Show newest version

/*
 * Copyright 2013 Kazuhiro Sera
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */
package scalikejdbc.async.internal.postgresql

import scalikejdbc._, async._, internal._
import com.github.mauricio.async.db.pool.ConnectionPool
import com.github.mauricio.async.db.pool.PoolConfiguration
import com.github.mauricio.async.db.postgresql.PostgreSQLConnection
import com.github.mauricio.async.db.postgresql.pool.PostgreSQLConnectionFactory

/**
 * PostgreSQL Connection Pool
 *
 * @param url jdbc url
 * @param user username
 * @param password password
 * @param settings extra settings
 */
private[scalikejdbc] class PostgreSQLConnectionPoolImpl(
  override val url: String,
  override val user: String,
  password: String,
  override val settings: AsyncConnectionPoolSettings = AsyncConnectionPoolSettings())
    extends AsyncConnectionPool(url, user, password, settings)
    with LogSupport {

  private[this] val factory = new PostgreSQLConnectionFactory(config)
  private[this] val pool = new ConnectionPool[PostgreSQLConnection](
    factory = factory,
    configuration = PoolConfiguration(
      maxObjects = settings.maxPoolSize,
      maxIdle = settings.maxIdleMillis,
      maxQueueSize = settings.maxQueueSize)
  )

  override def borrow(): AsyncConnection = new PoolableAsyncConnection(pool) with PostgreSQLConnectionImpl

  override def close(): Unit = pool.disconnect

  override def giveBack(conn: NonSharedAsyncConnection): Unit = conn match {
    case conn: NonSharedAsyncConnectionImpl => pool.giveBack(conn.underlying.asInstanceOf[PostgreSQLConnection])
    case _ => log.debug("You don't need to give back this connection to the pool.")
  }

}