mysql2wrapper 0.0.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+coverage
+InstalledFiles
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+
+# YARD artifacts
+.yardoc
+_yardoc
+doc/

data/Gemfile

@@ -0,0 +1,5 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in batchbase.gemspec
+gemspec
+

data/Gemfile.lock

@@ -0,0 +1,20 @@
+PATH
+  remote: .
+  specs:
+    mysql2wrapper (0.0.1)
+      batchbase (= 0.0.2)
+      mysql2 (= 0.3.11)
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    batchbase (0.0.2)
+      sys-proctable (= 0.9.1)
+    mysql2 (0.3.11)
+    sys-proctable (0.9.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  mysql2wrapper!

data/README.md

@@ -0,0 +1,6 @@
+mysql2wrapper
+=============
+
+very simple mysql2 wrapper.
+it's ore ore library!
+

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/config/database.yml

@@ -0,0 +1,4 @@
+test:
+  host: 127.0.0.1
+  database: mysql2wrapper_test
+  username: root

data/config/database_multiple.yml

@@ -0,0 +1,9 @@
+test:
+  master:
+    host: 127.0.0.1
+    database: mysql2wrapper_test_master
+    username: root
+  slave:
+    host: 127.0.0.1
+    database: mysql2wrapper_test_slave
+    username: root

data/example/sample.rb

@@ -0,0 +1,77 @@
+require 'rubygems'
+require 'mysql2wrapper'
+require 'logger'
+
+#
+# HACKME サンプルをちょっとはマシに書く、、、、
+#
+
+
+logger = Logger.new(STDOUT)
+# フォーマッターを変える
+#logger.formatter = Kanamei::LogFormatter.formatter
+db_yml_path = File::dirname(__FILE__) + '/../config/database.yml'
+logger.info db_yml_path
+
+# 第3引数は複数DB接続が無いならば指定不要
+db_config = Mysql2wrapper::Client.config_from_yml(db_yml_path,'test')
+
+client = Mysql2wrapper::Client.new(db_config,logger)
+# ログ出力しないなら第二引数をnilで
+#client = Mysql2wrapper::Client.new(db_config,nil)
+
+query = '
+DROP TABLE IF EXISTS `hoges`'
+client.query query
+query = '
+CREATE TABLE IF NOT EXISTS `hoges` (
+  `id` int(11) NOT NULL auto_increment,
+  `value1` int(11) NOT NULL,
+  `created_at` datetime NOT NULL,
+  `updated_at` timestamp NOT NULL default CURRENT_TIMESTAMP on update CURRENT_TIMESTAMP,
+  PRIMARY KEY  (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;'
+client.query query
+
+logger.info "hoge has #{client.count 'hoges'}rows"
+client.query "INSERT INTO hoges (value1,created_at)VALUES(#{Time.now.to_i},NOW())"
+logger.info "hoge has #{client.count 'hoges'}rows"
+
+begin
+  # トランザクションはブロックを渡す（ARと同じく）
+  client.transaction do
+    logger.info "hoge has #{client.count 'hoges'}rows"
+   client.query "INSERT INTO hoges (value1,created_at)VALUES(#{Time.now.to_i},NOW())"
+    client.query query
+    logger.info "hoge has #{client.count 'hoges'}rows"
+    raise 'error' # call ROLLBACK
+  end
+rescue
+end
+logger.info "hoge has #{client.count 'hoges'}rows"
+
+# ハッシュを引数にしたインサートの発行
+@i = 0
+hash = {
+  :value1=>proc{@i+=1}, # procを引数にもできる
+  :created_at=>'now()'.to_func
+}
+client.insert('hoges',hash)
+logger.info "affected_rows:#{client.affected_rows}"
+logger.info "hoge has #{client.count 'hoges'}rows"
+
+# 配列を引数にしたインサートの発行
+ar = []
+20.times{ar << hash}
+client.insert('hoges',ar)
+logger.info "affected_rows:#{client.affected_rows}" # Client#affected_rowsでマルチプルインサートのトータルインサート数が取れる
+logger.info "hoge has #{client.count 'hoges'}rows"
+
+# ハッシュを引数にしたアップデート
+client.update 'hoges',{:value1=>3},'id = 1'
+logger.info "hoge has #{client.count 'hoges'}rows"
+# 全行アップデートをしたなら第三引数でMysql2wrapper::Client::UpdateAllClassを引数にしないとダメ
+client.update 'hoges',{:value1=>3},Mysql2wrapper::Client::UpdateAllClass
+# clientインスタンスにショートカットあり
+client.update 'hoges',{:value1=>4},client.update_all_flag
+logger.info "hoge has #{client.count 'hoges'}rows"

data/lib/mysql2wrapper.rb

@@ -0,0 +1,16 @@
+require "mysql2wrapper/version"
+require "mysql2wrapper/client"
+
+class String
+  def to_func
+    @__function_sql = true
+    self
+  end
+
+  def function_sql?
+    @__function_sql == true
+  end
+end
+
+module Mysql2wrapper
+end

data/lib/mysql2wrapper/client.rb

@@ -0,0 +1,203 @@
+# -*- coding: utf-8 -*-
+
+require 'yaml'
+require 'mysql2'
+
+class Mysql2wrapper::Client
+  attr_accessor :config, :client, :logger
+  attr_reader :affected_rows
+
+  # 全行更新用のフラグ表現クラス
+  class UpdateAllClass;end
+
+  UPDATE_ALL          = UpdateAllClass
+  QUERY_BASE_COLOR    = 35
+  QUERY_SPECIAL_COLOR = 31
+
+  MULTIPLE_INSERT_DEFAULT = 100
+
+  def initialize(config,_logger=Logger.new(STDOUT))
+    self.logger = _logger
+    if self.logger
+      self.logger.info "mysql2 client created with #{config.inspect}"
+    end
+    self.client = Mysql2::Client.new(config)
+    self.config = config
+    # サーバが古いので一応問題あるけど以下の方向で
+    # http://kennyqi.com/archives/61.html
+    self.class.query(self.client,"SET NAMES 'utf8'",self.logger)
+    self.class.query(self.client,"SET SQL_AUTO_IS_NULL=0",self.logger)
+    self.class.query(self.client,"SET SQL_MODE=STRICT_ALL_TABLES",self.logger)
+  end
+
+  def self.query(client,sql_query,logger=nil,color=QUERY_BASE_COLOR)
+    s = Time.now
+    ret = client.query(sql_query)
+    e = Time.now
+    if logger
+      # TODO too long sql shorten
+      if sql_query.size > 1400
+        sql_query = sql_query[0..600] + "\n<< trimed >>\n" + sql_query[(sql_query.size - 600)..(sql_query.size - 1)]
+      end
+      logger.info "[QUERY] "" \e[#{color}m (#{((e-s)*1000).round(2)}ms/#{client.affected_rows}rows) #{sql_query}\e[0m"
+    end
+    ret
+  end
+
+  def stop_logging
+    self.logger = nil
+  end
+
+  def dump
+    "#{self.config.inspect}\n#{self.client.pretty_inspect}"
+  end
+
+  def query(str,color=QUERY_BASE_COLOR)
+    res = self.class.query(self.client,str,self.logger,color)
+    @affected_rows = self.client.affected_rows
+    res
+  end
+
+  def transaction(&proc)
+    raise ArgumentError, "No block was given" unless block_given?
+    #query "SET AUTOCOMMIT=0;",QUERY_SPECIAL_COLOR
+    query "BEGIN",QUERY_SPECIAL_COLOR
+    begin
+      yield
+      query "COMMIT",QUERY_SPECIAL_COLOR
+    rescue => e
+      query "ROLLBACK",QUERY_SPECIAL_COLOR
+      raise e
+    end
+  end
+
+  def self.make_config_key_symbol(config)
+    new_config = {}
+    config.each do |key,value|
+      new_config[key.to_sym] = value
+    end
+    config = new_config
+  end
+
+  def count(table_name,key_name='*')
+    query("SELECT COUNT(#{escape(key_name)}) AS cnt FROM #{escape(table_name)}").first['cnt']
+  end
+
+  def close
+    self.client.close if self.client
+  end
+
+  def escape(str)
+    self.client.escape(str)
+  end
+
+  def sanitize(str)
+    escape(str)
+  end
+
+  #
+  # where は 'WHERE'を付けずに指定
+  # 全行updateを使用するシチュエーションはほぼ0に
+  # 近いと思われるので、簡単には実行できなくしてます
+  # 実際に実行する際はwhereにUpdateAllを引数とします
+  # client.update 'test01',{:v_int1=>3},Mysql2wrapper::Client::UPDATE_ALL
+  #
+  def update(table_name,hash,where)
+    case where
+    when String
+      where = "WHERE #{where}"
+    when UpdateAllClass.class
+      where = ''
+    else
+      raise ArgumentError, "where must be String or UpdateAll"
+    end
+    query = "UPDATE `#{escape(table_name)}` SET #{
+      hash.map do |key,value|
+        "`#{escape(key.to_s)}` = #{proceed_value(value)}"
+      end.join(',')
+    }" + where
+    self.query(query)
+  end
+
+  def update_all_flag
+    self.class::UPDATE_ALL
+  end
+
+  def proceed_value(value)
+    case value
+    when Proc
+      "'#{escape(value.call.to_s)}'"
+    when nil
+      "NULL"
+    when TrueClass,FalseClass
+      if value
+        "'1'"
+      else
+        "'0'"
+      end
+    when Time,DateTime
+      "'#{value.strftime("%Y-%m-%d %H:%M:%S")}'"
+    when Date
+      "'#{value.strftime("%Y-%m-%d")}'"
+    else
+      s = value
+      s = s.to_s unless s.kind_of?(String)
+      if s.respond_to?(:function_sql?) && s.function_sql?
+        "#{value.to_s}"
+      else
+        "'#{escape(value.to_s)}'"
+      end
+    end
+  end
+
+  def insert(table_name,data,multiple_insert_by=MULTIPLE_INSERT_DEFAULT)
+    @affected_rows = 0 # 一応リセットしとくか、、、
+    affected_rows_total = 0
+    query = "INSERT INTO `#{escape(table_name)}`"
+    _datas = data.clone
+    case _datas
+    when Array
+      ;
+    when Hash
+      _datas = [_datas]
+    else
+      raise ArgumentError, "data must be Array or Hash"
+    end
+
+    return nil if _datas.size == 0
+
+    # TODO affected_rows by multiple
+    _datas.each_slice(multiple_insert_by).each do |rows|
+      query = <<"EOS"
+INSERT INTO `#{escape(table_name)}`
+(#{rows.first.keys.map{|o|"`#{escape(o.to_s)}`"}.join(',')})
+VALUES
+#{
+  rows.map do |row|
+  "(#{
+    row.map do |key,value|
+      proceed_value(value)
+    end.join(',')
+  })"
+  end.join(',')
+}
+EOS
+      self.query(query.chomp)
+      affected_rows_total += self.client.affected_rows
+    end
+    @affected_rows = affected_rows_total
+  end
+
+  #
+  # db_server_nameはDB名そのもの（複数DB対応）
+  #
+  def self.config_from_yml(yml_path,environment,db_server_name=nil)
+    db_config = YAML.load_file(yml_path)[environment]
+    if db_server_name
+      db_config = self.make_config_key_symbol(db_config[db_server_name])
+    else
+      db_config = self.make_config_key_symbol(db_config)
+    end
+    db_config
+  end
+end

data/lib/mysql2wrapper/version.rb

@@ -0,0 +1,3 @@
+module Mysql2wrapper
+  VERSION = "0.0.1"
+end

data/mysql2wrapper.gemspec

@@ -0,0 +1,24 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "mysql2wrapper/version"
+
+Gem::Specification.new do |s|
+  s.name        = "mysql2wrapper"
+  s.version     = Mysql2wrapper::VERSION
+  s.authors     = ["pacojp"]
+  s.email       = ["paco.jp@gmail.com"]
+  s.homepage    = ""
+  s.summary     = %q{oreore mysql2 wrapper class}
+  s.description = %q{oreore mysql2 wrapper class}
+  s.rubyforge_project = "mysql2wrapper"
+  s.add_dependency 'mysql2','0.3.11'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # specify any dependencies here; for example:
+  # s.add_development_dependency "rspec"
+  # s.add_runtime_dependency "rest-client"
+end

data/test/by_hand.rb

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+
+#
+# trial space
+#
+
+$: << File.dirname(__FILE__)
+
+require 'rubygems'
+require 'test_helper'
+require 'logger'
+
+require 'mysql2wrapper'
+
+db_yml_path = File::dirname(__FILE__) + '/../config/database.yml'
+puts db_yml_path
+# 第3引数は複数DB接続が無いならば指定不要
+db_config = Mysql2wrapper::Client.config_from_yml(db_yml_path,'test')
+
+client = Mysql2wrapper::Client.new(db_config)
+query = 'DROP TABLE IF EXISTS `hoges`'
+client.query(query)
+query = '
+CREATE TABLE IF NOT EXISTS `hoges` (
+  `id` int(11) NOT NULL auto_increment,
+  `v_int1` int(11) NOT NULL,
+  `v_int2` int(11) NOT NULL,
+  `v_int3` int(11),
+  `v_str1` varchar(10) NOT NULL,
+  `v_str2` varchar(10) NOT NULL,
+  `v_str3` varchar(10) NOT NULL,
+  `v_datetime1` datetime NOT NULL,
+  `v_datetime2` datetime NOT NULL,
+  `created_at` datetime NOT NULL,
+  `updated_at` timestamp NOT NULL default CURRENT_TIMESTAMP on update CURRENT_TIMESTAMP,
+  PRIMARY KEY  (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;
+'
+client.query query
+
+@i = 0
+increment = proc{@i+=1}
+
+# hashでインサート
+hash = {
+  :v_int1=>increment,
+  :v_int2=>rand(10000),
+  :v_int3=>nil,
+  :v_str1=>'test',
+  :v_str2=>'hoho',
+  :v_str3=>"CONCAT('My','S','QL')".to_func,
+  :v_datetime1=>Time.mktime(2011,12,3,4,5,6),
+  :v_datetime2=>DateTime.new(2011,12,3,4,5,6),
+  :created_at=>'NOW()'.to_func
+}
+client.insert('hoges',hash)
+
+# hashの配列でマルチプルインサート
+array = []
+101.times{array << hash}
+client.insert('hoges',array)
+puts "hoge has #{client.count 'hoges'}rows"
+
+client.transaction do
+  client.insert('hoges',hash)
+end
+puts "hoge has #{client.count 'hoges'}rows"
+
+begin
+  client.transaction do
+    client.insert('hoges',hash)
+    raise 'roooooooooollback!'
+  end
+rescue
+end
+puts "hoge has #{client.count 'hoges'}rows"
+
+client.update 'hoges',{:v_int1=>2},'id = 1'
+client.update 'hoges',{:v_int1=>2},client.class::UPDATE_ALL
+

data/test/test.rb

@@ -0,0 +1,382 @@
+# -*- coding: utf-8 -*-
+
+$: << File.dirname(__FILE__)
+require 'test_helper'
+require 'test/unit'
+require 'Fileutils'
+require 'logger'
+
+#
+# localhostに以下のデータベースを作成しrootのパスワード無しでアクセスできるようにしておいてください
+#
+# mysql2wrapper_test
+# mysql2wrapper_test_master
+# mysql2wrapper_test_slave
+#
+
+class TestMysql2wrapper < Test::Unit::TestCase
+
+  def setup
+    @i = 0
+    client = get_client
+
+    query = 'DROP TABLE IF EXISTS `test01`'
+    client.query query
+    query = 'DROP TABLE IF EXISTS `test02`'
+    client.query query
+
+    query = simple_table_create_query('test01')
+    client.query query
+
+    query = '
+CREATE TABLE IF NOT EXISTS `test02` (
+  `id` int(11) NOT NULL auto_increment,
+  `v_int1` int(11) NOT NULL,
+  `v_int2` int(11) NOT NULL,
+  `v_int3` int(11),
+  `v_int4` int(11),
+  `v_bool1` tinyint(1) NOT NULL,
+  `v_bool2` tinyint(1) NOT NULL,
+  `v_bool3` tinyint(1),
+  `v_bool4` tinyint(1),
+  `v_str1` varchar(10) NOT NULL,
+  `v_str2` varchar(10) NOT NULL,
+  `v_str3` varchar(10),
+  `v_str4` varchar(10),
+  `v_date1` date NOT NULL,
+  `v_date2` date,
+  `v_date3` date,
+  `v_datetime1` datetime NOT NULL,
+  `v_datetime2` datetime,
+  `v_datetime3` datetime,
+  `v_time1` datetime NOT NULL,
+  `v_time2` datetime,
+  `v_time3` datetime,
+  `created_at` datetime NOT NULL,
+  `updated_at` timestamp NOT NULL default CURRENT_TIMESTAMP on update CURRENT_TIMESTAMP,
+  PRIMARY KEY  (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;
+    '
+    client.query query
+
+    client.close
+  end
+
+  def get_client
+    db_yml_path = File::dirname(__FILE__) + '/../config/database.yml'
+    db_config = Mysql2wrapper::Client.config_from_yml(db_yml_path,'test')
+    client = Mysql2wrapper::Client.new(db_config,nil)
+    client
+  end
+
+  def simple_table_create_query(table)
+    "
+CREATE TABLE IF NOT EXISTS `#{table}` (
+  `id` int(11) NOT NULL auto_increment,
+  `v_int1` int(11) NOT NULL,
+  `v_int2` int(11),
+  `v_str1` varchar(10) ,
+  `created_at` datetime NOT NULL,
+  `updated_at` timestamp NOT NULL default CURRENT_TIMESTAMP on update CURRENT_TIMESTAMP,
+  PRIMARY KEY  (`id`)
+) ENGINE=InnoDB DEFAULT CHARSET=utf8;
+    "
+  end
+
+  def hash_for_simple_table
+    {
+      :v_int1     => proc{ @i+=1 },
+      :created_at => 'NOW()'.to_func,
+    }
+  end
+
+  def insert(client)
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(#{Time.now.to_i},NOW())"
+    client.query query
+  end
+
+  def count(client=get_client)
+    client.count('test01','id')
+  end
+
+  def test_to_func
+    assert_equal false,''.function_sql?
+    assert_equal true,''.to_func.function_sql?
+    st = ""
+    assert_equal false,st.function_sql?
+    st = st.to_func
+    assert_equal true,st.function_sql?
+    assert_equal true,proc{"".to_func}.call.function_sql?
+    assert_equal true,Proc.new{"".to_func}.call.function_sql?
+  end
+
+  def test_count
+    client = get_client
+    assert_equal 0,client.count('test01','id')
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(#{Time.now.to_i},NOW())"
+    client.query query
+    assert_equal 1,client.count('test01','id')
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(#{Time.now.to_i},NOW())"
+    client.query query
+    assert_equal 2,client.count('test01','id')
+    client.close
+  end
+
+  def test_select
+    client = get_client
+    assert_equal 0,client.count('test01','id')
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(123,NOW())"
+    client.query query
+    assert_equal 1,client.count('test01','id')
+    res = client.query "SELECT * FROM test01 WHERE id = 1"
+    assert_equal 123, res.first['v_int1']
+    assert_equal 1, client.affected_rows
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(123,NOW())"
+    client.query query
+    res = client.query "SELECT * FROM test01 WHERE id = 1"
+    assert_equal 1, client.affected_rows
+    res = client.query "SELECT * FROM test01"
+    assert_equal 2, client.affected_rows
+    client.close
+  end
+
+  def test_update
+    client = get_client
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(1,NOW())"
+    client.query query
+    client.query query
+    client.query query
+
+    res = client.query "SELECT * FROM test01 WHERE id = 1"
+    assert_equal 1, res.first['v_int1']
+    assert_equal nil, res.first['v_int2']
+    res = client.query "SELECT * FROM test01 WHERE id = 2"
+    assert_equal 1, res.first['v_int1']
+    assert_equal nil, res.first['v_int2']
+
+    client.update 'test01',{:v_int1=>2},'id = 1'
+    assert_equal 1, client.affected_rows
+
+    res = client.query "SELECT * FROM test01 WHERE id = 1"
+    assert_equal 2, res.first['v_int1']
+    assert_equal nil, res.first['v_int2']
+    res = client.query "SELECT * FROM test01 WHERE id = 2"
+    assert_equal 1, res.first['v_int1']
+    assert_equal nil, res.first['v_int2']
+
+    client.update 'test01',{:v_int1=>3}, client.update_all_flag
+
+    assert_equal 3, client.affected_rows
+    res = client.query "SELECT * FROM test01 WHERE id = 1"
+    assert_equal 3, res.first['v_int1']
+    assert_equal nil, res.first['v_int2']
+    res = client.query "SELECT * FROM test01 WHERE id = 2"
+    assert_equal 3, res.first['v_int1']
+    assert_equal nil, res.first['v_int2']
+    client.update 'test01',{:v_int1=>3},Mysql2wrapper::Client::UPDATE_ALL
+    assert_equal 0, client.affected_rows
+    client.update 'test01',{:v_int1=>4},client.update_all_flag
+    assert_equal 3, client.affected_rows
+
+    client.close
+  end
+
+  def test_update_all_limitter
+    client = get_client
+    query = "INSERT INTO test01 (v_int1,created_at)VALUES(1,NOW())"
+    client.query query
+
+    assert_raise(ArgumentError){
+      client.update 'test01',{:v_int1=>3},nil
+    }
+
+    assert_raise(Mysql2::Error){
+      client.update 'test01',{:v_int1=>3},''
+    }
+  end
+
+  def test_escape
+    client = get_client
+    assert_equal 0,client.count('test01','id')
+    query = "INSERT INTO test01 (v_int1,v_str1,created_at)VALUES(#{Time.now.to_i},'#{client.escape("'te'st'st'")}',NOW())"
+    client.query query
+    res = client.query "SELECT * FROM test01"
+    assert_equal 1,res.size
+    assert_equal "'te'st'st'",res.first['v_str1']
+    client.close
+  end
+
+  def test_tranasction_simple
+    client = get_client
+    insert(client)
+    assert_equal 1,count(client)
+    begin
+      client.transaction do
+        insert(client)
+        assert_equal 2,count(client)
+        raise 'roooooooooollback!!!!'
+      end
+    rescue
+    end
+    assert_equal 1,count(client)
+    insert(client)
+    assert_equal 2,count(client)
+    client.close
+  end
+
+  def test_tranasction_2_clients
+    client1 = get_client
+    client2 = get_client
+    insert(client1)
+    assert_equal 1,count(client1)
+    assert_equal 1,count(client2)
+    begin
+      # コネクションはトランザクション単位
+      client1.transaction do
+        insert(client1)
+        assert_equal 2,count(client1)
+        assert_equal 1,count(client2)
+        raise 'roooooooooollback!!!!'
+      end
+    rescue
+    end
+    assert_equal 1,count(client1)
+    assert_equal 1,count(client2)
+
+    client1.transaction do
+      insert(client2) #  raiseしないなら
+    end
+    assert_equal 2,count(client1)
+    assert_equal 2,count(client2)
+
+    begin
+      # コネクションはトランザクション単位
+      # client1のtransactionブロック内でclient2のインサートを発行しても
+      # もちろんロールバックされない
+      client1.transaction do
+        insert(client1)
+        assert_equal 3,count(client1)
+        assert_equal 2,count(client2)
+        insert(client2)
+        assert_equal 3,count(client1) # ここは3
+        assert_equal 3,count(client2)
+        raise 'roooooooooollback!!!!'
+      end
+    rescue
+    end
+    assert_equal 3,count(client1)
+    assert_equal 3,count(client2)
+    client1.close
+    client2.close
+  end
+
+  def test_insert
+    client = get_client
+    @i = 0
+    hash = {
+      :v_int1 => proc{ @i+=1 },
+      :v_int2 => 246,
+      :v_int3 => nil,
+      :v_str1=>"te'st日本語",
+      :v_str2=>"CONCAT('My','S','QL')".to_func,
+      :v_str3 => nil,
+      :v_bool1 => true,
+      :v_bool2 => false,
+      :v_bool3 => nil,
+      :v_date1 => Date.new(2000,1,2),
+      :v_date2 => nil,
+      :v_datetime1 => DateTime.new(2000,1,2,3,4,5),
+      :v_datetime2 => nil,
+      :v_time1 => Time.mktime(2000,1,2,3,4,5),
+      :v_time2 => nil,
+      :created_at => 'NOW()'.to_func,
+    }
+    client.insert('test02',hash)
+    assert_equal 1, client.affected_rows
+    client.insert('test02',hash)
+    res = client.query ("SELECT * FROM test02")
+    assert_equal 2, res.size
+    row = res.to_a.last
+    assert_equal 2,       row['v_int1']
+    assert_equal 246,     row['v_int2']
+    assert_equal nil,     row['v_int3']
+    assert_equal nil,     row['v_int4']
+    assert_equal 'te\'st日本語', row['v_str1']
+    assert_equal 'MySQL', row['v_str2']
+    assert_equal nil,     row['v_str3']
+    assert_equal nil,     row['v_str4']
+    assert_equal 1,       row['v_bool1']
+    assert_equal 0,       row['v_bool2']
+    assert_equal nil,     row['v_bool3']
+    assert_equal nil,     row['v_bool4']
+    assert_equal Date.new(2000,1,2),row['v_date1']
+    assert_equal nil,     row['v_date2']
+    assert_equal nil,     row['v_date3']
+    assert_equal Time.mktime(2000,1,2,3,4,5),row['v_datetime1']
+    assert_equal nil,     row['v_datetime2']
+    assert_equal nil,     row['v_datetime3']
+    assert_equal Time.mktime(2000,1,2,3,4,5),row['v_time1']
+    assert_equal nil,     row['v_time2']
+    assert_equal nil,     row['v_time3']
+  end
+
+  def test_multiple_insert
+    client = get_client
+    @i = 0
+    hash = hash_for_simple_table
+    client.insert('test01',hash)
+    res = client.query ("SELECT * FROM test01")
+    assert_equal 1, res.size
+    assert_equal 1, client.affected_rows
+    client.insert('test01',hash)
+    res = client.query ("SELECT * FROM test01")
+    assert_equal 2, res.size
+    assert_equal 2, res.to_a.last['v_int1']
+    ar = []
+    111.times{ar << hash}
+    client.insert('test01',ar)
+    assert_equal 111, client.affected_rows
+    res = client.query ("SELECT * FROM test01")
+    assert_equal 113, res.size
+    assert_equal 113, res.to_a.last['v_int1']
+  end
+
+  def test_sql_mode_is_strict_all_tables
+    assert_raise(Mysql2::Error) do
+      client = get_client
+      query = "INSERT INTO test01 (v_int1)values(1)"
+      client.query(query)
+    end
+  end
+
+  def test_multiple_database
+    db_yml_path   = File::dirname(__FILE__) + '/../config/database_multiple.yml'
+    config_master = Mysql2wrapper::Client.config_from_yml(db_yml_path,'test','master')
+    config_slave  = Mysql2wrapper::Client.config_from_yml(db_yml_path,'test','slave')
+    client_master = Mysql2wrapper::Client.new(config_master,nil)
+    client_slave  = Mysql2wrapper::Client.new(config_slave,nil)
+
+    client_master.query 'DROP TABLE IF EXISTS tbl_master'
+    client_master.query simple_table_create_query('tbl_master')
+    client_slave.query  'DROP TABLE IF EXISTS tbl_slave'
+    client_slave.query  simple_table_create_query('tbl_slave')
+
+    assert_equal 0,client_master.count('tbl_master')
+    assert_equal 0,client_slave.count('tbl_slave')
+
+    client_master.insert('tbl_master',hash_for_simple_table)
+
+    assert_equal 1,client_master.count('tbl_master')
+    assert_equal 0,client_slave.count('tbl_slave')
+
+    client_master.insert('tbl_master',hash_for_simple_table)
+
+    assert_equal 2,client_master.count('tbl_master')
+    assert_equal 0,client_slave.count('tbl_slave')
+
+    client_slave.insert('tbl_slave',hash_for_simple_table)
+
+    assert_equal 2,client_master.count('tbl_master')
+    assert_equal 1,client_slave.count('tbl_slave')
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,5 @@
+$: << File.dirname(__FILE__) + '/../lib'
+#require File.dirname(__FILE__) + '/../lib/image_convert_lite'
+
+require 'mysql2wrapper'
+require 'mysql2wrapper/version'

metadata

@@ -0,0 +1,80 @@
+--- !ruby/object:Gem::Specification
+name: mysql2wrapper
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- pacojp
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mysql2
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.3.11
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.3.11
+description: oreore mysql2 wrapper class
+email:
+- paco.jp@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- config/database.yml
+- config/database_multiple.yml
+- example/sample.rb
+- lib/mysql2wrapper.rb
+- lib/mysql2wrapper/client.rb
+- lib/mysql2wrapper/version.rb
+- mysql2wrapper.gemspec
+- test/by_hand.rb
+- test/test.rb
+- test/test_helper.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: mysql2wrapper
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: oreore mysql2 wrapper class
+test_files:
+- test/by_hand.rb
+- test/test.rb
+- test/test_helper.rb
+has_rdoc: