baza 0.0.17 → 0.0.18

data/lib/baza/drivers/mysql_java/column.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::MysqlJava::Column < Baza::Driver::Mysql::Column
+end

data/lib/baza/drivers/mysql_java/columns.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::MysqlJava::Columns < Baza::Driver::Mysql::Columns
+end

data/lib/baza/drivers/mysql_java/index.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::MysqlJava::Index < Baza::Driver::Mysql::Index
+end

data/lib/baza/drivers/mysql_java/indexes.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::MysqlJava::Indexes < Baza::Driver::Mysql::Indexes
+end

data/lib/baza/drivers/mysql_java/table.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::MysqlJava::Table < Baza::Driver::Mysql::Table
+end

data/lib/baza/drivers/mysql_java/tables.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::MysqlJava::Tables < Baza::Driver::Mysql::Tables
+end

data/lib/baza/drivers/sqlite3.rb

@@ -0,0 +1,81 @@
+#This class handels SQLite3-specific behaviour.
+class Baza::Driver::Sqlite3 < Baza::BaseSqlDriver
+  path = "#{File.dirname(__FILE__)}/sqlite3"
+
+  autoload :Table, "#{path}/table"
+  autoload :Tables, "#{path}/tables"
+  autoload :Column, "#{path}/column"
+  autoload :Columns, "#{path}/columns"
+  autoload :Index, "#{path}/index"
+  autoload :Indexes, "#{path}/indexes"
+  autoload :Result, "#{path}/result"
+  autoload :Sqlspecs, "#{path}/sqlspecs"
+  autoload :UnbufferedResult, "#{path}/unbuffered_result"
+
+  attr_reader :mutex_statement_reader
+
+  #Helper to enable automatic registering of database using Baza::Db.from_object
+  def self.from_object(args)
+    if args[:object].class.name == "SQLite3::Database"
+      return {
+        type: :success,
+        args: {
+          type: :sqlite3,
+          conn: args[:object]
+        }
+      }
+    end
+  end
+
+  #Constructor. This should not be called manually.
+  def initialize(baza_db)
+    super
+
+    @path = @baza.opts[:path] if @baza.opts[:path]
+    @mutex_statement_reader = Mutex.new
+
+    if @baza.opts[:conn]
+      @conn = @baza.opts[:conn]
+    else
+      raise "No path was given." unless @path
+      require 'sqlite3' unless ::Object.const_defined?(:SQLite3)
+
+      @conn = ::SQLite3::Database.open(@path)
+      @conn.type_translation = false # Type translation is always done in the C ext for SQLite3
+    end
+  end
+
+  #Executes a query against the driver.
+  def query(sql)
+    @mutex_statement_reader.synchronize do
+      return Baza::Driver::Sqlite3::Result.new(self, @conn.prepare(sql))
+    end
+  end
+
+  def query_ubuf(sql)
+    return Baza::Driver::Sqlite3::UnbufferedResult.new(self, @conn.prepare(sql))
+  end
+
+  #Escapes a string to be safe to used in a query.
+  def escape(string)
+    #This code is taken directly from the documentation so we dont have to rely on the SQLite3::Database class. This way it can also be used with JRuby and IronRuby...
+    #http://sqlite-ruby.rubyforge.org/classes/SQLite/Database.html
+    return string.to_s.gsub(/'/, "''")
+  end
+
+  #Returns the last inserted ID.
+  def last_id
+    return @conn.last_insert_row_id if @conn.respond_to?(:last_insert_row_id)
+    return query("SELECT last_insert_rowid() AS id").fetch[:id].to_i
+  end
+
+  #Closes the connection to the database.
+  def close
+    @mutex_statement_reader.synchronize { @conn.close }
+  end
+
+  #Starts a transaction, yields the database and commits.
+  def transaction
+    @conn.transaction { yield @baza }
+  end
+end

data/lib/baza/drivers/sqlite3/result.rb

@@ -0,0 +1,64 @@
+# This class handels the result when running MRI (or others).
+class Baza::Driver::Sqlite3::Result < Baza::ResultBase
+  # Constructor. This should not be called manually.
+  def initialize(driver, statement)
+    @statement = statement
+
+    begin
+      @statement.execute
+      @type_translation = driver.baza.opts[:type_translation]
+      @types = statement.types if @type_translation == true
+      @columns = statement.columns.map { |column| column.to_sym }
+      read_results
+      @index = -1
+    ensure
+      @statement.try(:close)
+    end
+  end
+
+  # Returns a single result.
+  def fetch
+    row = @results[@index += 1]
+
+    if row
+      if @types
+        row.map!.with_index { |value, index| translate_type(value, @types[index]) } if @types
+      elsif @type_translation == :string
+        row.map! { |value| value.to_s }
+      end
+
+      return Hash[*@columns.zip(row).flatten]
+    end
+  end
+
+  # Loops over every result yielding them.
+  def each
+    while data = fetch
+      yield data
+    end
+  end
+
+private
+
+  def read_results
+    @results = []
+
+    loop do
+      row = @statement.step
+      break if @statement.done?
+      @results << row
+    end
+  end
+
+  def translate_type(value, type)
+    if value
+      if type == 'datetime'
+        return Time.parse(value)
+      elsif type == 'date'
+        return Date.parse(value)
+      else
+        return value
+      end
+    end
+  end
+end

data/{include/drivers/sqlite3/sqlite3_sqlspecs.rb → lib/baza/drivers/sqlite3/sqlspecs.rb}

@@ -2,4 +2,4 @@ class Baza::Driver::Sqlite3::Sqlspecs < Baza::Sqlspecs
   def strftime(val, col_str)
     return "STRFTIME('#{val}', SUBSTR(#{col_str}, 0, 20))"
   end
-end
+end

data/lib/baza/drivers/sqlite3/unbuffered_result.rb

@@ -0,0 +1,33 @@
+class Baza::Driver::Sqlite3::UnbufferedResult
+  def initialize(driver, statement)
+    @statement = statement
+    @statement.execute
+    @columns = statement.columns.map { |column| column.to_sym }
+  end
+
+  def fetch
+    return nil if @closed
+
+    array = @statement.step
+
+    if @statement.done?
+      close
+      return nil
+    end
+
+    return Hash[*@columns.zip(array).flatten] if array
+  end
+
+  def each
+    while data = fetch
+      yield data
+    end
+  end
+
+private
+
+  def close
+    @statement.close
+    @closed = true
+  end
+end

data/lib/baza/drivers/sqlite3_java.rb

@@ -0,0 +1,75 @@
+#This class handels SQLite3-specific behaviour.
+class Baza::Driver::Sqlite3Java < Baza::JdbcDriver
+  path = "#{File.dirname(__FILE__)}/sqlite3_java"
+
+  autoload :Table, "#{path}/table"
+  autoload :Tables, "#{path}/tables"
+  autoload :Column, "#{path}/column"
+  autoload :Columns, "#{path}/columns"
+  autoload :Index, "#{path}/index"
+  autoload :Indexes, "#{path}/indexes"
+  autoload :Result, "#{path}/result"
+  autoload :Sqlspecs, "#{path}/sqlspecs"
+  autoload :UnbufferedResult, "#{path}/unbuffered_result"
+
+  attr_reader :mutex_statement_reader
+
+  #Helper to enable automatic registering of database using Baza::Db.from_object
+  def self.from_object(args)
+    if args[:object].class.name == "Java::OrgSqlite::SQLiteConnection"
+      return {
+        type: :success,
+        args: {
+          type: :sqlite3_java,
+          conn: args[:object]
+        }
+      }
+    end
+  end
+
+  #Constructor. This should not be called manually.
+  def initialize(baza_db)
+    super
+
+    @path = @baza.opts[:path] if @baza.opts[:path]
+    @preload_results = true
+
+    if @baza.opts[:conn]
+      @conn = @baza.opts[:conn]
+    else
+      org.sqlite.JDBC
+      reconnect
+    end
+  end
+
+  def reconnect
+    raise "No path was given." unless @path
+
+    @stmt = nil
+    @conn = java.sql.DriverManager::getConnection("jdbc:sqlite:#{@path}")
+  end
+
+  #Escapes a string to be safe to used in a query.
+  def escape(string)
+    #This code is taken directly from the documentation so we dont have to rely on the SQLite3::Database class. This way it can also be used with JRuby and IronRuby...
+    #http://sqlite-ruby.rubyforge.org/classes/SQLite/Database.html
+    return string.to_s.gsub(/'/, "''")
+  end
+
+  #Returns the last inserted ID.
+  def last_id
+    return query("SELECT LAST_INSERT_ROWID() AS id").fetch[:id].to_i
+  end
+
+  def transaction
+    query_no_result_set("BEGIN TRANSACTION")
+
+    begin
+      yield @baza
+      query_no_result_set("COMMIT")
+    rescue
+      query_no_result_set("ROLLBACK")
+      raise
+    end
+  end
+end

data/lib/baza/drivers/sqlite3_java/column.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::Sqlite3Java::Column < Baza::Driver::Sqlite3::Column
+end

data/lib/baza/drivers/sqlite3_java/columns.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::Sqlite3Java::Columns < Baza::Driver::Sqlite3::Columns
+end

data/lib/baza/drivers/sqlite3_java/index.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::Sqlite3Java::Index < Baza::Driver::Sqlite3::Index
+end

data/lib/baza/drivers/sqlite3_java/indexes.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::Sqlite3Java::Indexes < Baza::Driver::Sqlite3::Indexes
+end

data/lib/baza/drivers/sqlite3_java/table.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::Sqlite3Java::Table < Baza::Driver::Sqlite3::Table
+end

data/lib/baza/drivers/sqlite3_java/tables.rb

@@ -0,0 +1,2 @@
+class Baza::Driver::Sqlite3Java::Tables < Baza::Driver::Sqlite3::Tables
+end

data/lib/baza/drivers/sqlite3_java/unbuffered_result.rb

@@ -0,0 +1,33 @@
+class Baza::Driver::Sqlite3Java::UnbufferedResult < Baza::ResultBase
+  def initialize(driver, result_set)
+    @result_set = result_set
+
+    if @result_set
+      metadata = @result_set.meta_data
+      @columns_count = metadata.column_count
+
+      @columns = []
+      1.upto(@columns_count) do |count|
+        @columns << metadata.column_name(count).to_sym
+      end
+    end
+  end
+
+  def fetch
+    result = @result_set.next if @result_set
+    return nil unless result
+
+    hash = {}
+    @columns_count.times do |count|
+      hash[@columns[count]] = @result_set.string(count + 1)
+    end
+
+    return hash
+  end
+
+  def each
+    while data = fetch
+      yield data
+    end
+  end
+end

data/lib/baza/drivers/sqlite3_rhodes.rb

@@ -0,0 +1,78 @@
+#This class handels SQLite3-specific behaviour.
+class Baza::Driver::Sqlite3Rhodes < Baza::BaseSqlDriver
+  path = "#{File.dirname(__FILE__)}/sqlite3_rhodes"
+
+  autoload :Table, "#{path}/table"
+  autoload :Tables, "#{path}/tables"
+  autoload :Column, "#{path}/column"
+  autoload :Columns, "#{path}/columns"
+  autoload :Index, "#{path}/index"
+  autoload :Indexes, "#{path}/indexes"
+  autoload :Result, "#{path}/result"
+  autoload :Sqlspecs, "#{path}/sqlspecs"
+  autoload :UnbufferedResult, "#{path}/unbuffered_result"
+
+  attr_reader :mutex_statement_reader
+
+  # Helper to enable automatic registering of database using Baza::Db.from_object
+  def self.from_object(args)
+    if args[:object].class.name == "SQLite3::Database"
+      return {
+        type: :success,
+        args: {
+          type: :sqlite3,
+          conn: args[:object]
+        }
+      }
+    end
+  end
+
+  # Constructor. This should not be called manually.
+  def initialize(baza_db)
+    super
+
+    @path = @baza.opts[:path] if @baza.opts[:path]
+    @mutex_statement_reader = Mutex.new
+
+    if @baza.opts[:conn]
+      @conn = @baza.opts[:conn]
+    else
+      raise "No path was given." unless @path
+      @conn = ::SQLite3::Database.new(@path, @path)
+    end
+  end
+
+  # Executes a query against the driver.
+  def query(sql)
+    return Baza::Driver::Sqlite3::Result.new(self, @conn.execute(sql, sql))
+  end
+
+  def query_ubuf(sql)
+    return Baza::Driver::Sqlite3::UnbufferedResult.new(self, @conn.prepare(sql))
+  end
+
+  # Escapes a string to be safe to used in a query.
+  def escape(string)
+    #This code is taken directly from the documentation so we dont have to rely on the SQLite3::Database class. This way it can also be used with JRuby and IronRuby...
+    #http://sqlite-ruby.rubyforge.org/classes/SQLite/Database.html
+    return string.to_s.gsub(/'/, "''")
+  end
+
+  # Returns the last inserted ID.
+  def last_id
+    return @conn.last_insert_row_id if @conn.respond_to?(:last_insert_row_id)
+    return query("SELECT last_insert_rowid() AS id").fetch[:id].to_i
+  end
+
+  # Closes the connection to the database.
+  def close
+    @mutex_statement_reader.synchronize do
+      @conn.close
+    end
+  end
+
+  # Starts a transaction, yields the database and commits.
+  def transaction
+    @conn.transaction { yield @baza }
+  end
+end