baza 0.0.17 → 0.0.18

data/lib/baza/drivers/mysql2.rb

@@ -0,0 +1,259 @@
+class Baza::Driver::Mysql2 < Baza::BaseSqlDriver
+  path = "#{File.dirname(__FILE__)}/mysql2"
+
+  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 :UnbufferedResult, "#{path}/unbuffered_result"
+  autoload :Sqlspecs, "#{path}/sqlspecs"
+
+  attr_reader :conn, :conns
+
+  #Helper to enable automatic registering of database using Baza::Db.from_object
+  def self.from_object(args)
+    if args[:object].class.name == "Mysql2::Client"
+      return {
+        type: :success,
+        args: {
+          type: :mysql2,
+          conn: args[:object],
+          query_args: {
+            symbolize_keys: true
+          }
+        }
+      }
+    end
+
+    return nil
+  end
+
+  def initialize(baza)
+    super
+
+    @opts = @baza.opts
+
+    require "monitor"
+    @mutex = Monitor.new
+
+    if @opts[:encoding]
+      @encoding = @opts[:encoding]
+    else
+      @encoding = "utf8"
+    end
+
+    if @baza.opts.key?(:port)
+      @port = @baza.opts[:port].to_i
+    else
+      @port = 3306
+    end
+
+    reconnect
+  end
+
+  #Cleans the wref-map holding the tables.
+  def clean
+    tables.clean if tables
+  end
+
+  #Respawns the connection to the MySQL-database.
+  def reconnect
+    @mutex.synchronize do
+      args = {
+        host: @baza.opts[:host],
+        username: @baza.opts[:user],
+        password: @baza.opts[:pass],
+        database: @baza.opts[:db],
+        port: @port,
+        symbolize_keys: true,
+        cache_rows: false
+      }
+
+      #Symbolize keys should also be given here, else table-data wont be symbolized for some reason - knj.
+      @query_args = {symbolize_keys: true,}
+      @query_args[:cast] = false unless @baza.opts[:type_translation]
+      @query_args.merge!(@baza.opts[:query_args]) if @baza.opts[:query_args]
+
+      pos_args = [:as, :async, :cast_booleans, :database_timezone, :application_timezone, :cache_rows, :connect_flags, :cast]
+      pos_args.each do |key|
+        args[key] = @baza.opts[key] if @baza.opts.key?(key)
+      end
+
+      args[:as] = :array
+
+      tries = 0
+      begin
+        tries += 1
+        if @baza.opts[:conn]
+          @conn = @baza.opts[:conn]
+        else
+          require "mysql2"
+          @conn = Mysql2::Client.new(args)
+        end
+      rescue => e
+        if tries <= 3
+          if e.message == "Can't connect to local MySQL server through socket '/var/run/mysqld/mysqld.sock' (111)"
+            sleep 1
+            tries += 1
+            retry
+          end
+        end
+
+        raise e
+      end
+
+      query("SET NAMES '#{esc(@encoding)}'") if @encoding
+    end
+  end
+
+  #Executes a query and returns the result.
+  def query(str)
+    str = str.to_s
+    str = str.force_encoding("UTF-8") if @encoding == "utf8" && str.respond_to?(:force_encoding)
+    tries = 0
+
+    begin
+      tries += 1
+      @mutex.synchronize do
+        return Baza::Driver::Mysql2::Result.new(self, @conn.query(str, @query_args))
+      end
+    rescue => e
+      if tries <= 3
+        if e.message == "MySQL server has gone away" || e.message == "closed MySQL connection" or e.message == "Can't connect to local MySQL server through socket"
+          sleep 0.5
+          reconnect
+          retry
+        elsif e.message.include?("No operations allowed after connection closed") or e.message == "This connection is still waiting for a result, try again once you have the result" or e.message == "Lock wait timeout exceeded; try restarting transaction"
+          reconnect
+          retry
+        end
+      end
+
+      raise e
+    end
+  end
+
+  #Executes an unbuffered query and returns the result that can be used to access the data.
+  def query_ubuf(str)
+    @mutex.synchronize do
+      return Baza::Driver::Mysql2::Result.new(self, @conn.query(str, @query_args.merge(stream: true)))
+    end
+  end
+
+  #Escapes a string to be safe to use in a query.
+  def escape(string)
+    return @conn.escape(string.to_s)
+  end
+
+  #Returns the last inserted ID for the connection.
+  def last_id
+    @mutex.synchronize { return @conn.last_id.to_i }
+  end
+
+  #Closes the connection threadsafe.
+  def close
+    @mutex.synchronize { @conn.close }
+  end
+
+  #Destroyes the connection.
+  def destroy
+    @conn = nil
+    @baza = nil
+    @mutex = nil
+    @encoding = nil
+    @query_args = nil
+    @port = nil
+  end
+
+  #Inserts multiple rows in a table. Can return the inserted IDs if asked to in arguments.
+  def insert_multi(tablename, arr_hashes, args = nil)
+    sql = "INSERT INTO `#{tablename}` ("
+
+    first = true
+    if args && args[:keys]
+      keys = args[:keys]
+    elsif arr_hashes.first.is_a?(Hash)
+      keys = arr_hashes.first.keys
+    else
+      raise "Could not figure out keys."
+    end
+
+    keys.each do |col_name|
+      sql << "," unless first
+      first = false if first
+      sql << "`#{self.esc_col(col_name)}`"
+    end
+
+    sql << ") VALUES ("
+
+    first = true
+    arr_hashes.each do |hash|
+      if first
+        first = false
+      else
+        sql << "),("
+      end
+
+      first_key = true
+      if hash.is_a?(Array)
+        hash.each do |val|
+          if first_key
+            first_key = false
+          else
+            sql << ","
+          end
+
+          sql << @baza.sqlval(val)
+        end
+      else
+        hash.each do |key, val|
+          if first_key
+            first_key = false
+          else
+            sql << ","
+          end
+
+          sql << @baza.sqlval(val)
+        end
+      end
+    end
+
+    sql << ")"
+
+    return sql if args && args[:return_sql]
+
+    query(sql)
+
+    if args && args[:return_id]
+      first_id = self.last_id
+      raise "Invalid ID: #{first_id}" if first_id.to_i <= 0
+      ids = [first_id]
+      1.upto(arr_hashes.length - 1) do |count|
+        ids << first_id + count
+      end
+
+      ids_length = ids.length
+      arr_hashes_length = arr_hashes.length
+      raise "Invalid length (#{ids_length}, #{arr_hashes_length})." if ids_length != arr_hashes_length
+
+      return ids
+    else
+      return nil
+    end
+  end
+
+  def transaction
+    @baza.q("START TRANSACTION")
+
+    begin
+      yield @baza
+      @baza.q("COMMIT")
+    rescue
+      @baza.q("ROLLBACK")
+      raise
+    end
+  end
+end

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

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

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

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

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

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

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

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

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

@@ -0,0 +1,26 @@
+# This class controls the result for the MySQL2 driver.
+class Baza::Driver::Mysql2::Result < Baza::ResultBase
+  # Constructor. This should not be called manually.
+  def initialize(driver, result)
+    @result = result
+    @type_translation = driver.baza.opts[:type_translation]
+  end
+
+  # Returns a single result.
+  def fetch
+    begin
+      return to_enum.next
+    rescue StopIteration
+      return false
+    end
+  end
+
+  # Loops over every single result yielding it.
+  def each
+    @result.each(as: :hash, symbolize_keys: true) do |row|
+      next unless row # This sometimes happens when streaming results...
+      row = Hash[row.map { |k, v| [k, v.to_s] }] if @type_translation == :string
+      yield row
+    end
+  end
+end

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

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

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

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

data/lib/baza/drivers/mysql_java.rb

@@ -0,0 +1,178 @@
+class Baza::Driver::MysqlJava < Baza::JdbcDriver
+  path = "#{File.dirname(__FILE__)}/mysql_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 :UnbufferedResult, "#{path}/unbuffered_result"
+  autoload :Sqlspecs, "#{path}/sqlspecs"
+
+  attr_reader :conn, :conns
+
+  #Helper to enable automatic registering of database using Baza::Db.from_object
+  def self.from_object(args)
+    if args[:object].class.name == "Java::ComMysqlJdbc::JDBC4Connection"
+      return {
+        type: :success,
+        args: {
+          type: :mysql_java,
+          conn: args[:object]
+        }
+      }
+    end
+
+    return nil
+  end
+
+  def initialize(baza)
+    super
+
+    @opts = @baza.opts
+
+    if @opts[:encoding]
+      @encoding = @opts[:encoding]
+    else
+      @encoding = 'utf8'
+    end
+
+    if @baza.opts.key?(:port)
+      @port = @baza.opts[:port].to_i
+    else
+      @port = 3306
+    end
+
+    @java_rs_data = {}
+    reconnect
+  end
+
+  #Respawns the connection to the MySQL-database.
+  def reconnect
+    @mutex.synchronize do
+      if @baza.opts[:conn]
+        @jdbc_loaded = true
+        @conn = @baza.opts[:conn]
+      else
+        com.mysql.jdbc.Driver
+        @conn = java.sql::DriverManager.getConnection("jdbc:mysql://#{@baza.opts[:host]}:#{@port}/#{@baza.opts[:db]}?user=#{@baza.opts[:user]}&password=#{@baza.opts[:pass]}&populateInsertRowWithDefaultValues=true&zeroDateTimeBehavior=round&characterEncoding=#{@encoding}&holdResultsOpenOverStatementClose=true")
+      end
+
+      query_no_result_set("SET SQL_MODE = ''")
+      query_no_result_set("SET NAMES '#{self.esc(@encoding)}'") if @encoding
+    end
+  end
+
+  #Returns the last inserted ID for the connection.
+  def last_id
+    data = query("SELECT LAST_INSERT_ID() AS id").fetch
+    return data[:id].to_i if data[:id]
+    raise "Could not figure out last inserted ID."
+  end
+
+  #Closes the connection threadsafe.
+  def close
+    @mutex.synchronize { @conn.close }
+  end
+
+  #Destroyes the connection.
+  def destroy
+    @conn = nil
+    @baza = nil
+    @mutex = nil
+    @encoding = nil
+    @query_args = nil
+    @port = nil
+  end
+
+  #Inserts multiple rows in a table. Can return the inserted IDs if asked to in arguments.
+  def insert_multi(tablename, arr_hashes, args = nil)
+    sql = "INSERT INTO `#{tablename}` ("
+
+    first = true
+    if args && args[:keys]
+      keys = args[:keys]
+    elsif arr_hashes.first.is_a?(Hash)
+      keys = arr_hashes.first.keys
+    else
+      raise "Could not figure out keys."
+    end
+
+    keys.each do |col_name|
+      sql << "," unless first
+      first = false if first
+      sql << "`#{self.esc_col(col_name)}`"
+    end
+
+    sql << ") VALUES ("
+
+    first = true
+    arr_hashes.each do |hash|
+      if first
+        first = false
+      else
+        sql << "),("
+      end
+
+      first_key = true
+      if hash.is_a?(Array)
+        hash.each do |val|
+          if first_key
+            first_key = false
+          else
+            sql << ","
+          end
+
+          sql << @baza.sqlval(val)
+        end
+      else
+        hash.each do |key, val|
+          if first_key
+            first_key = false
+          else
+            sql << ","
+          end
+
+          sql << @baza.sqlval(val)
+        end
+      end
+    end
+
+    sql << ")"
+
+    return sql if args && args[:return_sql]
+
+    query_no_result_set(sql)
+
+    if args && args[:return_id]
+      first_id = self.last_id
+      raise "Invalid ID: #{first_id}" if first_id.to_i <= 0
+      ids = [first_id]
+      1.upto(arr_hashes.length - 1) do |count|
+        ids << first_id + count
+      end
+
+      ids_length = ids.length
+      arr_hashes_length = arr_hashes.length
+      raise "Invalid length (#{ids_length}, #{arr_hashes_length})." if ids_length != arr_hashes_length
+
+      return ids
+    else
+      return nil
+    end
+  end
+
+  def transaction
+    query_no_result_set("START TRANSACTION")
+
+    begin
+      yield @baza
+      query_no_result_set("COMMIT")
+    rescue
+      query_no_result_set("ROLLBACK")
+      raise
+    end
+  end
+end