com.mchange.sc.v1.sql.ResourceUtils.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mchange-commons-scala_2.11 Show documentation
mchange-commons-scala
The newest version!
/*
 * Distributed as part of mchange-commons-scala v0.4.0
 *
 * Copyright (C) 2015 Machinery For Change, Inc.
 *
 * Author: Steve Waldman 
 *
 * This library is free software; you can redistribute it and/or modify
 * it under the terms of EITHER:
 *
 *     1) The GNU Lesser General Public License (LGPL), version 2.1, as 
 *        published by the Free Software Foundation
 *
 * OR
 *
 *     2) The Eclipse Public License (EPL), version 1.0
 *
 * You may choose which license to accept if you wish to redistribute
 * or modify this work. You may offer derivatives of this work
 * under the license you have chosen, or you may provide the same
 * choice of license which you have been offered here.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received copies of both LGPL v2.1 and EPL v1.0
 * along with this software; see the files LICENSE-EPL and LICENSE-LGPL.
 * If not, the text of these licenses are currently available at
 *
 * LGPL v2.1: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html
 *  EPL v1.0: http://www.eclipse.org/org/documents/epl-v10.php 
 * 
 */

package com.mchange.sc.v1.sql;

import language.reflectiveCalls;

import java.sql.{Connection,DriverManager,PreparedStatement,Statement,ResultSet};
import javax.sql.DataSource;
import com.mchange.sc.v1.util.ClosableUtils._;

object ResourceUtils
{
  type ConnectionSource = { def getConnection() : Connection } // javax.sql.DataSource conforms, but when we don't need a full implementation

  def connectionSource( jdbcUrl : String, jdbcUserName : String, jdbcPassword : String ) : ConnectionSource =
  {
    if ( jdbcUserName == null )
      new { def getConnection() = DriverManager.getConnection( jdbcUrl ); }
    else
      new { def getConnection() = DriverManager.getConnection( jdbcUrl, jdbcUserName, jdbcPassword ); }
  }

  def connectionSource( jdbcUrl : String ) : ConnectionSource = connectionSource( jdbcUrl, null, null );

  def withConnection[T]( csrc : ConnectionSource )( op : (Connection) => T) : T =
  {
    var con : Connection = null;
    
    try
    {
      con = csrc.getConnection();
      op(con);
    }
    finally
    { attemptClose( con ); }
  }

  // to avoid the reflective method call when we have a real javax.sql.DataSource
  def withConnection[T]( csrc : DataSource )( op : (Connection) => T) : T =
  {
    var con : Connection = null;
    
    try
    {
      con = csrc.getConnection();
      op(con);
    }
    finally
    { attemptClose( con ); }
  }

  def withStatement[T]( con : Connection )( op : (Statement) => T) : T =
  {
    var stmt : Statement = null;
    
    try
    {
      stmt = con.createStatement();
      op(stmt);
    }
    finally
    { attemptClose( stmt ); }
  }

  def withPreparedStatement[T]( con : Connection, sql : String )( op : (PreparedStatement) => T ) : T =
  {
    var ps : PreparedStatement = null;
    
    try
    {
      ps = con.prepareStatement( sql );
      op( ps );
    }
    finally
    { attemptClose( ps ); }
  }

  def withResultSet[T]( stmt : Statement, query : String )( op : (ResultSet) => T ) : T =
  {
    var rs : ResultSet = null;

    try
    {
      rs = stmt.executeQuery( query );
      op( rs );
    }
    finally
    { attemptClose( rs ); }
  }

  def withResultSet[T]( ps : PreparedStatement )( op : (ResultSet) => T ) : T =
  {
    var rs : ResultSet = null;

    try
    {
      rs = ps.executeQuery();
      op( rs );
    }
    finally
    { attemptClose( rs ); }
  }

  def withWrappedPreparedStatement[T]( con : Connection, mgr : PsManager)( op : (WrappedPreparedStatement) => T ) : T =
  {
    var wps : WrappedPreparedStatement = null;

    try
    {
      wps = mgr.prepareStatement(con);
      op( wps );
    }
    finally
    { attemptClose( wps ); }
  }
}