jdbc-wrapper 0.2

data/README

@@ -0,0 +1,13 @@
+Minimum Requirements
+====================
+
+* Java 1.5
+* JRuby 1.1 (tested with RC2)
+* JDBC Drivers
+
+How to Install
+==============
+
+* gem install jdbc-wrapper
+* get your jdbc driver jars on your classpath
+  (the easier way to do this is drop them in $JRUBY_HOME/lib)

data/Rakefile

@@ -0,0 +1,25 @@
+desc 'run all unit tests'
+task :test do 
+  require 'rake/runtest'
+  Rake.run_tests 'test/test_*.rb'
+end
+
+require 'rubygems' 
+Gem::manage_gems 
+require 'rake/gempackagetask' 
+
+gemspec = Gem::Specification.new do |s| 
+  s.name = "jdbc-wrapper"
+  s.version = "0.2" 
+  s.author = "Larry Myers" 
+  s.email = "larry@larrymyers.com" 
+  s.platform = Gem::Platform::RUBY 
+  s.summary = "A basic JDBC wrapper for JRuby" 
+  s.files = FileList["{examples,lib,test}/**/*","Rakefile","README"].to_a 
+  s.require_path = "lib" 
+  s.has_rdoc = false 
+end 
+
+Rake::GemPackageTask.new(gemspec) do |pkg| 
+  pkg.need_tar = true 
+end

data/examples/derby.rb

@@ -0,0 +1,21 @@
+$:.unshift File.join(File.dirname(__FILE__),'..','lib')
+
+require 'fileutils'
+require 'jdbc'
+
+db_path = File.join(File.dirname(__FILE__),'..','derby-example')
+
+JDBC::DB.start(:derby,nil,nil,nil,nil,db_path) do |db|
+  db.query("CREATE TABLE records " + 
+    "(name VARCHAR(80), created_on TIMESTAMP)")
+    
+  db.query("INSERT INTO records (name, created_on) " + 
+    "VALUES ('foo', '#{Time.now.strftime("%Y-%m-%d %H:%M:%S")}')")
+  
+  db.query("SELECT * FROM records").each_hash do |row|
+    puts row
+  end
+end
+
+FileUtils.rm_r(db_path, :force => true) if File.exist?(db_path)
+FileUtils.rm("derby.log") if File.exist?("derby.log")

data/examples/h2.rb

@@ -0,0 +1,15 @@
+$:.unshift File.join(File.dirname(__FILE__),'..','lib')
+
+require 'jdbc'
+
+JDBC::DB.start(:h2,nil,nil,nil,nil,'foo') do |db|
+  db.query("CREATE TABLE records " + 
+    "(name VARCHAR(80), created_on TIMESTAMP)")
+    
+  db.query("INSERT INTO records (name, created_on) " + 
+    "VALUES ('foo', '#{Time.now.strftime("%Y-%m-%d %H:%M:%S")}')")
+  
+  db.query("SELECT * FROM records").each_hash do |row|
+    row.each { |k,v| puts "#{k} = #{v}" }
+  end
+end

data/examples/mysql.rb

@@ -0,0 +1,15 @@
+$:.unshift File.join(File.dirname(__FILE__),'..','lib')
+
+require 'jdbc'
+
+JDBC::DB.start(:mysql,"localhost","3306","root","root","tasks_dev") do |db|
+  db.query("select * from jobs limit 1").each do |row|
+    row.each { |v| puts v }
+    puts
+  end
+
+  db.query("select * from jobs limit 1").each_hash do |row|
+    row.each { |k,v| puts "#{k} = #{v}, #{v.class}" }
+    puts
+  end
+end

data/lib/jdbc/adapters/derby.rb

@@ -0,0 +1,29 @@
+module JDBC
+  module Adapters
+    class Derby
+      attr_reader :host, :port, :user, :password, :schema
+      
+      def initialize(host, port, user, password, schema)
+        @host = host
+        @port = port
+        @user = user
+        @password = password
+        @schema = schema
+      end
+      
+      def connection_string
+        str = "jdbc:derby:#{@schema}"
+      
+        if ! File.exist?(@schema)
+          str << ";create=true"
+        end
+      
+        return str
+      end
+      
+      def class_name
+        "org.apache.derby.jdbc.EmbeddedDriver"
+      end
+    end
+  end
+end

data/lib/jdbc/adapters/h2.rb

@@ -0,0 +1,23 @@
+module JDBC
+  module Adapters
+    class H2
+      attr_reader :host, :port, :user, :password, :schema
+      
+      def initialize(host, port, user, password, schema)
+        @host = host
+        @port = port
+        @user = user
+        @password = password
+        @schema = schema
+      end
+      
+      def connection_string
+        "jdbc:h2:mem:#{@schema}"
+      end
+      
+      def class_name
+        "org.h2.Driver"
+      end
+    end
+  end
+end

data/lib/jdbc/adapters/hsqldb.rb

@@ -0,0 +1,23 @@
+module JDBC
+  module Adapters
+    class Hsqldb
+      attr_reader :host, :port, :user, :password, :schema
+      
+      def initialize(host, port, user, password, schema)
+        @host = host
+        @port = port
+        @user = user
+        @password = password
+        @schema = schema
+      end
+      
+      def connection_string
+        "jdbc:hsqldb:mem:#{@schema}"
+      end
+      
+      def class_name
+        "org.hsqldb.jdbcDriver"
+      end
+    end
+  end
+end

data/lib/jdbc/adapters/mysql.rb

@@ -0,0 +1,24 @@
+module JDBC
+  module Adapters
+    class Mysql
+      attr_reader :host, :port, :user, :password, :schema
+      
+      def initialize(host, port, user, password, schema)
+        @host = host
+        @port = port
+        @user = user
+        @password = password
+        @schema = schema
+      end
+      
+      def connection_string
+        "jdbc:mysql://#{@host}:#{@port}/#{@schema}" +
+                                  "?user=#{@user}&password=#{@password}"
+      end
+      
+      def class_name
+        "com.mysql.jdbc.Driver"
+      end
+    end
+  end
+end

data/lib/jdbc/adapters/postgresql.rb

@@ -0,0 +1,24 @@
+module JDBC
+  module Adapters
+    class Postgresql
+      attr_reader :host, :port, :user, :password, :schema
+      
+      def initialize(host, port, user, password, schema)
+        @host = host
+        @port = port
+        @user = user
+        @password = password
+        @schema = schema
+      end
+      
+      def connection_string
+        "jdbc:postgresql://#{@host}:#{@port}/#{@schema}" +
+                                  "?user=#{@user}&password=#{@password}"
+      end
+      
+      def class_name
+        "org.postgresql.Driver"
+      end
+    end
+  end
+end

data/lib/jdbc/adapters.rb

@@ -0,0 +1,7 @@
+$:.unshift File.join(File.dirname(__FILE__),'.')
+
+require 'adapters/derby'
+require 'adapters/h2'
+require 'adapters/hsqldb'
+require 'adapters/mysql'
+require 'adapters/postgresql'

data/lib/jdbc/db.rb

@@ -0,0 +1,85 @@
+$:.unshift File.join(File.dirname(__FILE__),'.')
+
+require 'java'
+require 'time'
+require 'fileutils'
+
+require 'adapters'
+
+module JDBC
+  class DB
+    def initialize(engine, host, port, user, password, schema)
+      adapter = get_adapter(engine, host, port, user, password, schema)
+      
+      begin
+        java.lang.Class.forName(adapter.class_name).newInstance()
+    
+        @conn = JavaSql::DriverManager.getConnection(
+                                  adapter.connection_string)
+      rescue JavaSql::SQLException => e
+        raise RuntimeError.new(e.message)
+      end
+    end
+    
+    def self.start(db_class, host, port, user, password, database)
+      db = nil
+      
+      begin
+          db = DB.new(db_class, host, port, user, password, database)
+      
+          yield(db)
+      rescue JavaSql::SQLException => e
+        raise RuntimeError.new(e.message)
+      ensure
+        db.close unless db.nil?
+      end
+    end
+  
+    def query(sql)
+      stmt = nil
+      result = nil
+      
+      begin
+        stmt = @conn.createStatement
+        
+        res = stmt.execute(sql)
+        
+        if res == false
+          return stmt.getUpdateCount
+        end
+        
+        return Result.new(stmt.getResultSet, stmt)
+      rescue JavaSql::SQLException => e
+        stmt.close unless stmt.nil?
+        raise RuntimeError.new(e.message)
+      end
+    end
+    
+    def prepare(sql)
+      return PreparedStatement.new(@conn.prepareStatement(sql))
+    end
+    
+    def close
+      @conn.close unless @conn.nil?
+    end
+  
+    private
+    
+    def get_adapter(engine, host, port, user, password, schema)
+      case engine
+      when :derby
+        return Adapters::Derby.new(host, port, user, password, schema)
+      when :h2
+        return Adapters::H2.new(host, port, user, password, schema)
+      when :hsqldb
+        return Adapters::Hsqldb.new(host, port, user, password, schema)
+      when :mysql
+        return Adapters::Mysql.new(host, port, user, password, schema)
+      when :postgresql
+        return Adapters::Postgresql.new(host, port, user, password, schema)
+      end
+      
+      raise InvalidException("#{engine} is not supported.")
+    end
+  end
+end

data/lib/jdbc/prepared_statement.rb

@@ -0,0 +1,31 @@
+module JDBC
+  class PreparedStatement
+    def initialize(stmt)
+      @stmt = stmt
+      @meta_data = stmt.getParameterMetaData
+    end
+    
+    def execute(*args)
+      if args.length != @meta_data.getParameterCount
+        raise RuntimeError.new("Got #{args.length} params, " + 
+                      "expected #{@meta_data.getParameterCount}.")
+      end
+      
+      @stmt.clearParameters
+      
+      args.each_with_index do |arg, i|
+        @stmt.setObject(i+1, arg, @meta_data.getParameterType(i+1))
+      end
+      
+      if @stmt.execute
+        return Result.new(@stmt.getResultSet, @stmt)
+      end
+      
+      return @stmt.getUpdateCount
+    end
+    
+    def close
+      @stmt.close
+    end
+  end
+end

data/lib/jdbc/result.rb

@@ -0,0 +1,106 @@
+module JDBC
+  class Result
+    def initialize(resultSet, statement)
+      @rs = resultSet
+      @stmt = statement
+      @columns = get_rs_meta_data
+    end
+    
+    def fetch
+      if @rs.next
+        result = []
+      
+        @columns.each do |column| 
+          result << fetch_and_cast(@rs, column)
+        end
+        
+        return result
+      end
+      
+      close
+      
+      return nil
+    end
+    
+    def fetch_hash
+      if @rs.next
+        result = {}
+      
+        @columns.each do |column| 
+          result[column[:name]] = fetch_and_cast(@rs, column)
+        end
+        
+        return result
+      end
+      
+      close
+      
+      return nil
+    end
+    
+    def each
+      while(result = fetch)
+        yield(result)
+      end
+    end
+  
+    def each_hash
+      while(result = fetch_hash)
+        yield(result)
+      end
+    end
+  
+    def close
+      @rs.close unless @rs.nil?
+      @stmt.close unless @stmt.nil?
+    end
+  
+    private
+    
+    def get_rs_meta_data
+      meta_data = @rs.getMetaData
+    
+      columns = []
+    
+      meta_data.getColumnCount.times do |i|
+        columns << {
+          :name => meta_data.getColumnName(i+1).downcase,
+          :type => meta_data.getColumnType(i+1)
+        }
+      end
+      
+      return columns
+    end
+    
+    def fetch_and_cast(rs, column)
+      if column[:type] == JavaSql::Types::NULL
+        return nil
+      end
+    
+      if column[:type] == JavaSql::Types::INTEGER  ||
+         column[:type] == JavaSql::Types::SMALLINT ||
+         column[:type] == JavaSql::Types::TINYINT  ||
+         column[:type] == JavaSql::Types::BIGINT 
+        return rs.getInt(column[:name])
+      end
+    
+      if column[:type] == JavaSql::Types::DECIMAL ||
+         column[:type] == JavaSql::Types::FLOAT
+        return rs.getDouble(column[:name])
+      end
+    
+      if column[:type] == JavaSql::Types::DATE ||
+         column[:type] == JavaSql::Types::TIME ||
+         column[:type] == JavaSql::Types::TIMESTAMP
+      
+        val = rs.getString(column[:name])
+      
+        return nil if val.nil?
+      
+        return Time.parse(val)
+      end
+    
+      return rs.getString(column[:name])
+    end
+  end
+end

data/lib/jdbc.rb

@@ -0,0 +1,11 @@
+$:.unshift File.join(File.dirname(__FILE__),'.')
+
+require 'java'
+
+module JavaSql
+  include_package 'java.sql'
+end
+
+require 'jdbc/db'
+require 'jdbc/result'
+require 'jdbc/prepared_statement'

data/test/test_db.rb

@@ -0,0 +1,38 @@
+$:.unshift File.join(File.dirname(__FILE__),'..','lib')
+
+require 'test/unit'
+require 'jdbc'
+
+class TestDB < Test::Unit::TestCase 
+  def setup    
+    @db = JDBC::DB.new(:h2,nil,nil,nil,nil,'test')
+  end
+  
+  def teardown
+    @db.close
+  end
+  
+  def test_create_table
+    assert_equal(0, create_records_table)
+  end
+  
+  def test_insert_and_select
+    create_records_table
+    
+    count = @db.query("INSERT INTO records (name) VALUES ('foo')")
+    
+    assert_equal(1, count)
+    
+    result = @db.query("SELECT * FROM records").fetch
+    
+    assert_equal('foo', result[0])
+    
+    result = @db.query("SELECT * FROM records").fetch_hash
+    
+    assert_equal('foo', result['name'])
+  end
+  
+  def create_records_table
+    @db.query("CREATE TABLE records (name VARCHAR(80))")
+  end
+end

data/test/test_prepared_statement.rb

@@ -0,0 +1,46 @@
+$:.unshift File.join(File.dirname(__FILE__),'..','lib')
+
+require 'test/unit'
+require 'jdbc'
+
+class TestDB < Test::Unit::TestCase 
+  def setup    
+    @db = JDBC::DB.new(:h2,nil,nil,nil,nil,'test')
+  end
+  
+  def teardown
+    @db.close
+  end
+  
+  def test_insert_and_select
+    create_and_populate_records_table
+    
+    stmt = @db.prepare("INSERT INTO records (name,id) VALUES (?,?)")
+    
+    assert_equal(1, stmt.execute('bar', 2))
+    
+    stmt = @db.prepare("SELECT * FROM records where id = ?")
+    
+    row = stmt.execute(2).fetch
+    
+    assert_equal('bar', row[0])
+    assert_equal(2, row[1])
+  end
+  
+  def test_runtime_error_on_invalid_arg_count
+    create_and_populate_records_table
+    
+    stmt = @db.prepare("SELECT * FROM records WHERE name = ?")
+    
+    begin
+      stmt.execute('arg1','bad_arg2')
+      flunk
+    rescue RuntimeError => e
+      assert_equal("Got 2 params, expected 1.", e.message)
+    end
+  end
+  
+  def create_and_populate_records_table
+    @db.query("CREATE TABLE records (name VARCHAR(80), id INT)")
+  end
+end

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+extensions: []
+homepage: 
+executables: []
+version: !ruby/object:Gem::Version
+  version: !str 0.2
+post_install_message: 
+date: 2008-03-13 04:00:00 +00:00
+files: 
+- examples/derby.rb
+- examples/h2.rb
+- examples/mysql.rb
+- lib/jdbc
+- lib/jdbc.rb
+- lib/jdbc/adapters
+- lib/jdbc/adapters.rb
+- lib/jdbc/db.rb
+- lib/jdbc/prepared_statement.rb
+- lib/jdbc/result.rb
+- lib/jdbc/adapters/derby.rb
+- lib/jdbc/adapters/h2.rb
+- lib/jdbc/adapters/hsqldb.rb
+- lib/jdbc/adapters/mysql.rb
+- lib/jdbc/adapters/postgresql.rb
+- test/test_db.rb
+- test/test_prepared_statement.rb
+- Rakefile
+- README
+rubygems_version: 1.0.1
+rdoc_options: []
+signing_key: 
+cert_chain: []
+name: jdbc-wrapper
+has_rdoc: false
+platform: ruby
+summary: A basic JDBC wrapper for JRuby
+default_executable: 
+bindir: bin
+required_rubygems_version: !ruby/object:Gem::Requirement
+  version: 
+  requirements: 
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: !str 0
+required_ruby_version: !ruby/object:Gem::Requirement
+  version: 
+  requirements: 
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: !str 0
+require_paths: 
+- lib
+specification_version: 2
+test_files: []
+dependencies: []
+description: 
+email: larry@larrymyers.com
+authors: 
+- Larry Myers
+extra_rdoc_files: []
+requirements: []
+rubyforge_project: 
+autorequire: