rubberband_flamethrower 0.3.1 → 0.4.0

data/VERSION

@@ -1 +1 @@
-0.3.1
+0.4.0

data/bin/flamethrower

@@ -12,6 +12,8 @@ command = args.shift.strip rescue 'help'
 case command
 when "fire"
   RubberbandFlamethrower.fire(*args)
+when "auto"
+  RubberbandFlamethrower.auto(*args)
 else
   RubberbandFlamethrower.help
 end

data/lib/rubberband_flamethrower.rb

@@ -7,12 +7,28 @@ module RubberbandFlamethrower
     require File.dirname(__FILE__)+"/rubberband_flamethrower/flamethrower.rb"
     flamethrower = Flamethrower.new
     time = Benchmark.measure do
-      flamethrower.fire(how_many, starting_id, server_url, index, type)
+      flamethrower.fire(how_many, starting_id, server_url, index, type, 1)
     end
+    puts "\nFinished Inserting #{how_many} documents into Elastic Search."
     puts "  user       system     total    real"
     puts time
   end
 
+  def self.auto(how_many_batches=3, per_batch=5000, starting_id=1, server_url="http://localhost:9200", index="twitter", type="tweet")
+    require "active_support/core_ext"
+    require 'httparty'
+    require "benchmark"
+    require File.dirname(__FILE__)+"/rubberband_flamethrower/data_generator.rb"
+    require File.dirname(__FILE__)+"/rubberband_flamethrower/flamethrower.rb"
+    flamethrower = Flamethrower.new
+    Benchmark.bm(8) do |x|
+      how_many_batches.to_i.times do |i|
+        x.report("set #{i+1}:")   { flamethrower.fire(per_batch, starting_id, server_url, index, type) }
+        starting_id = starting_id.to_i + per_batch.to_i
+      end
+    end
+  end
+
   def self.help
     puts "Flamethrower Commands Available:\n\nflamethrower #display this help message\nflamethrower fire #sends a batch to the elastic search server"
   end

data/lib/rubberband_flamethrower/flamethrower.rb

@@ -1,9 +1,10 @@
 # This class is designed to insert objects created with the DataGenerator class into Elastic Search
 module RubberbandFlamethrower
   class Flamethrower
-    def fire(how_many, starting_id, server_url, index, type)
+    def fire(how_many, starting_id, server_url, index, type, printing_level=0)
 
-      $stdout.sync = true
+      
+      ($stdout.sync = true) if printing_level == 1
 
       # a unique ID must be provided for each document stored in Elastic Search
       id = starting_id.to_i
@@ -15,17 +16,16 @@ module RubberbandFlamethrower
 
         # generate a piece of random data to insert that approxiamates a tweet
         insert_data = data.generate_random_insert_data
-        #puts insert_data
+        (puts insert_data) if printing_level == 2
 
         # insert the random data into local elastic search index "twitter" as type "tweet" with set id
         response = HTTParty.put("#{server_url}/#{index}/#{type}/#{id}", body: insert_data) 
-        #puts response.body
-        print "."
+        (puts response.body) if printing_level == 2
+        (print ".") if printing_level == 1
 
         #increment the insert id
         id = id + 1
       end
-      puts "\n\nFinished Inserting #{how_many} documents into Elastic Search."
     end
     
   end

data/rubberband_flamethrower.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "rubberband_flamethrower"
-  s.version = "0.3.1"
+  s.version = "0.4.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Michael Orr"]
-  s.date = "2013-04-08"
+  s.date = "2013-04-09"
   s.description = "Use to quickly fill up some indicies in Elastic Search and to retrieve statistics about insertion rates"
   s.email = "michael@cloudspace.com"
   s.executables = ["flamethrower"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rubberband_flamethrower
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.4.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-04-08 00:00:00.000000000 Z
+date: 2013-04-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
-  requirement: &13986480 !ruby/object:Gem::Requirement
+  requirement: &17477820 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,10 +21,10 @@ dependencies:
         version: 0.10.2
   type: :runtime
   prerelease: false
-  version_requirements: *13986480
+  version_requirements: *17477820
 - !ruby/object:Gem::Dependency
   name: activesupport
-  requirement: &13985980 !ruby/object:Gem::Requirement
+  requirement: &17477280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -32,10 +32,10 @@ dependencies:
         version: 3.2.13
   type: :runtime
   prerelease: false
-  version_requirements: *13985980
+  version_requirements: *17477280
 - !ruby/object:Gem::Dependency
   name: rdoc
-  requirement: &13985500 !ruby/object:Gem::Requirement
+  requirement: &17476800 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -43,10 +43,10 @@ dependencies:
         version: '3.12'
   type: :development
   prerelease: false
-  version_requirements: *13985500
+  version_requirements: *17476800
 - !ruby/object:Gem::Dependency
   name: bundler
-  requirement: &13985000 !ruby/object:Gem::Requirement
+  requirement: &17476280 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -54,10 +54,10 @@ dependencies:
         version: 1.3.0
   type: :development
   prerelease: false
-  version_requirements: *13985000
+  version_requirements: *17476280
 - !ruby/object:Gem::Dependency
   name: jeweler
-  requirement: &13984520 !ruby/object:Gem::Requirement
+  requirement: &17475800 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -65,7 +65,7 @@ dependencies:
         version: 1.8.4
   type: :development
   prerelease: false
-  version_requirements: *13984520
+  version_requirements: *17475800
 description: Use to quickly fill up some indicies in Elastic Search and to retrieve
   statistics about insertion rates
 email: michael@cloudspace.com
@@ -117,7 +117,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -629124849433174154
+      hash: 3868063792966642724
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements: