testdata_generater_for_mysql 0.0.2 → 0.0.3

data/README.md

@@ -1,7 +1,8 @@
-# テスト用データジェネレータforMySQL
+# テスト用データジェネレータ for MySQL
 
 まぁよく「テストデータ」ってのを作るんですが、、、毎回マルチプルインサートな簡易スクリプトを作っては捨てしてきた自分からの卒業を目論んだリポジトリです。
 ドキュメントはありませんが、サンプルでほぼやれることは解説していますのでそちらを参考にしてください。
 
 サンプル
-https://github.com/pacojp/testdata_generater_for_mysql/blob/master/example/sample1.rb
+https://github.com/pacojp/testdata_generater_for_mysql/blob/master/example/sample1.rb
+

data/example/sample1.rb

@@ -73,16 +73,16 @@ loops = [
 # シングルクォーテーションでの囲みがかからなくなります
 #
 procs = {
-  :brand_id    => Proc.new{|v|v[:brand_id]},
-  :shop_id     => Proc.new{|v|v[:shop_id]},
-  :user_id     => Proc.new{|v|v['user_id']},
-  :name        => Proc.new{|v|"#{v[:brand_id]}_#{v[:shop_id]}_#{v['user_id']}'s name"},
-  :value1      => Proc.new{rand(10000)},
-  :value_nil   => Proc.new{nil},
-  :value_func  => Proc.new{"CONCAT('My', 'S', 'QL')".to_func}, #関数指定時はString#to_funcで
-  :value_true  => Proc.new{true}, # true は 1 false は 0
-  :value_time  => Proc.new{Time.mktime(2001,2,3,4,35,6)}, # Time,DateTime,Date
-  :created_at  => Proc.new{'NOW()'.to_func},
+  :brand_id    => proc{|v|v[:brand_id]},
+  :shop_id     => proc{|v|v[:shop_id]},
+  :user_id     => proc{|v|v['user_id']},
+  :name        => proc{|v|"#{v[:brand_id]}_#{v[:shop_id]}_#{v['user_id']}'s name"},
+  :value1      => proc{rand(10000)},
+  :value_nil   => proc{nil},
+  :value_func  => proc{"CONCAT('My', 'S', 'QL')".to_func}, #関数指定時はString#to_funcで
+  :value_true  => proc{true}, # true は 1 false は 0
+  :value_time  => proc{Time.mktime(2001,2,3,4,35,6)}, # Time,DateTime,Date
+  :created_at  => proc{'NOW()'.to_func},
 }
 #
 # 実際にテストデータを作成します
@@ -94,7 +94,6 @@ create_rows(
   procs
 )
 
-
 # 以下作成結果のサンプルを出力しています
 puts '=' * 60
 puts query("SELECT count(id) AS cnt FROM tests").first['cnt'].to_s + "rows created"

data/lib/testdata_generater_for_mysql/core.rb

@@ -1,7 +1,8 @@
 # -*- coding: utf-8 -*-
 
 require 'progressbar'
-require 'mysql2'
+require 'mysql2wrapper'
+require 'logger'
 
 #
 # TODO 以下の削除 サンプルの作成
@@ -34,27 +35,13 @@ require 'mysql2'
 #  )
 #
 
-class String
-  def to_func
-    @__is_function = true
-    self
-  end
-
-  def function?
-    if @__is_function
-      true
-    else
-      false
-    end
-  end
-end
-
 module TestdataGeneraterForMysql
 
   INSERT_PER_ROWS = 100
 
   def setup_mysql_client(hash)
-    @__client = Mysql2::Client.new(hash)
+    @__client = Mysql2wrapper::Client.new(hash,nil)
+    #@__client = Mysql2wrapper::Client.new(hash)
   end
 
   def insert_per_rows(v)
@@ -134,47 +121,8 @@ module TestdataGeneraterForMysql
 
   def do_insert
     return if @__insert_values.size == 0
-    @__inserted_rows += @__insert_values.size
-    query = <<"EOS"
-INSERT INTO `#{@__client.escape(@__table_name)}`
-(#{@__col_procs.keys.map{|o|"`#{@__client.escape(o.to_s)}`"}.join(',')})
-VALUES
-#{
-  @__insert_values.map do |row|
-  "(#{
-    row.map do |key,value|
-      case value
-      when nil
-        "NULL"
-      when TrueClass,FalseClass
-        if value
-          "'1'"
-        else
-          "'0'"
-        end
-      # TODO when datetime time
-      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?) && s.function?
-          "#{value.to_s}"
-        else
-          "'#{@__client.escape(value.to_s)}'"
-        end
-      end
-    end.join(',')
-  })"
-  end.join(',')
-}
-EOS
-    @__client.query(query)
-    if @__pbar
-      @__pbar.inc(@__insert_values.size)
-    end
+    @__client.insert(@__client.escape(@__table_name),@__insert_values)
+    @__pbar.inc(@__insert_values.size) if @__pbar
     @__insert_values = []
   end
 end

data/lib/testdata_generater_for_mysql/version.rb

@@ -1,3 +1,3 @@
 module TestdataGeneraterForMysql
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/test/test.rb

@@ -148,6 +148,7 @@ CREATE TABLE tests (
       :value1      => Proc.new{rand(10000)},
       :value2      => Proc.new{rand(10000)},
       :value_nil   => Proc.new{nil},
+      #:value_func  => Proc.new{"CONCAT('My', 'S', 'QL')".to_func},
       :value_func  => Proc.new{"CONCAT('My', 'S', 'QL')".to_func},
       :value_true  => Proc.new{true},
       :value_false => Proc.new{false},

data/testdata_generater_for_mysql.gemspec

@@ -11,7 +11,7 @@ Gem::Specification.new do |s|
   s.summary     = %q{oreore mysql test data generater}
   s.description = %q{oreore mysql test data generater}
 
-  s.add_dependency 'mysql2'
+  s.add_dependency 'mysql2wrapper','0.0.1' # https://github.com/pacojp/mysql2wrapper
   s.add_dependency 'progressbar'
 
   s.rubyforge_project = "testdata_generater_for_mysql"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: testdata_generater_for_mysql
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,24 +9,24 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-10-07 00:00:00.000000000 Z
+date: 2012-10-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: mysql2
+  name: mysql2wrapper
   requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
+    - - '='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.0.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
+    - - '='
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.0.1
 - !ruby/object:Gem::Dependency
   name: progressbar
   requirement: !ruby/object:Gem::Requirement