rebar 0.2.0

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Paul Rosania
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,71 @@
+# Rebar
+
+![Rebar](http://upload.wikimedia.org/wikipedia/commons/thumb/6/62/Trebar.jpg/220px-Trebar.jpg)
+
+A simple stress testing tool for Redis.
+
+* Are sorted sets fast enough with 1,000,000 items?
+* How big a dataset will fit in a 512 MB Redis instance?
+* Should I use a thousand hashes with a thousand keys, or a million strings?
+
+Rebar makes it easy to fill a Redis instance with sample data, so you can test
+performance under different database designs. Check memory usage and database
+creation time, or prepare a test dataset for a benchmarking exercise.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'rebar'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install rebar
+
+## Usage
+
+More documentation to follow. For now, here's some sample code.
+
+    require 'rubygems'
+    require 'rebar'
+    require 'redis'
+
+    Rebar.build(Redis.new) do
+      # Build 10 sets with 100 members each
+      sets(10) do
+        members 100
+      end
+
+      # Build 10 sorted sets with 100 members each
+      sorted_sets(10) do
+        members 100
+      end
+
+      # Build 10 hashes with 100 keys each
+      hashes(10) do
+        keys 100
+      end
+
+      # Build 10 lists with 100 elements each
+      lists(10) do
+        elements 100
+      end
+
+      # Build 10 random key-value pairs
+      strings(10)
+    end
+
+By default, Rebar cleans the `rebar` namespace before it builds a dataset. To
+disable this behavior, pass `:clean => false` when you call `Rebar.build`.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Added some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,2 @@
+#!/usr/bin/env rake
+require "bundler/gem_tasks"

data/lib/rebar.rb

@@ -0,0 +1,12 @@
+require "rebar/version"
+require "rebar/builder"
+require "rebar/random"
+require "rebar/dsl"
+
+module Rebar
+  def self.build(connection, options = {}, &block)
+    builder = Rebar::Builder.new(connection)
+    builder.clean! unless options[:clean] == false
+    builder.build(&block)
+  end
+end

data/lib/rebar/builder.rb

@@ -0,0 +1,19 @@
+module Rebar
+  class Builder
+    def initialize(connection)
+      @connection = connection
+    end
+
+    attr_reader :connection
+
+    def build(&block)
+      Rebar::DSL::Database.new(connection).instance_eval(&block)
+    end
+
+    def clean!
+      connection.keys('rebar:*').each do |key|
+        connection.del(key)
+      end
+    end
+  end
+end

data/lib/rebar/dsl.rb

@@ -0,0 +1,5 @@
+require "rebar/dsl/database"
+require "rebar/dsl/hash"
+require "rebar/dsl/list"
+require "rebar/dsl/set"
+require "rebar/dsl/sorted_set"

data/lib/rebar/dsl/database.rb

@@ -0,0 +1,48 @@
+module Rebar
+  module DSL
+    class Database
+      def initialize(connection)
+        @connection = connection
+        @max_ids = ::Hash.new(0)
+      end
+
+      attr_reader :connection
+
+      def strings(count)
+        count.times do
+          connection.set("rebar:string:#{next_id(:string)}", Rebar::Random.string)
+        end
+      end
+
+      def hashes(count, &block)
+        count.times do
+          Rebar::DSL::Hash.new(connection, "rebar:hash:#{next_id(:hash)}").instance_eval(&block)
+        end
+      end
+
+      def lists(count, &block)
+        count.times do
+          Rebar::DSL::List.new(connection, "rebar:list:#{next_id(:list)}").instance_eval(&block)
+        end
+      end
+
+      def sets(count, &block)
+        count.times do
+          Rebar::DSL::Set.new(connection, "rebar:set:#{next_id(:set)}").instance_eval(&block)
+        end
+      end
+
+      def sorted_sets(count, &block)
+        count.times do
+          Rebar::DSL::SortedSet.new(connection, "rebar:sorted_set:#{next_id(:sorted_set)}").instance_eval(&block)
+        end
+      end
+
+      private
+
+      def next_id(type)
+        @max_ids[type] += 1
+      end
+    end
+  end
+end

data/lib/rebar/dsl/hash.rb

@@ -0,0 +1,19 @@
+module Rebar
+  module DSL
+    class Hash
+      def initialize(connection, key)
+        @connection = connection
+        @key        = key
+      end
+
+      attr_reader :connection
+      attr_reader :key
+
+      def keys(count)
+        count.times do
+          connection.hset(key, Rebar::Random.string, Rebar::Random.string)
+        end
+      end
+    end
+  end
+end

data/lib/rebar/dsl/list.rb

@@ -0,0 +1,19 @@
+module Rebar
+  module DSL
+    class List
+      def initialize(connection, key)
+        @connection = connection
+        @key        = key
+      end
+
+      attr_reader :connection
+      attr_reader :key
+
+      def elements(count)
+        count.times do
+          connection.rpush(key, Rebar::Random.string)
+        end
+      end
+    end
+  end
+end

data/lib/rebar/dsl/set.rb

@@ -0,0 +1,19 @@
+module Rebar
+  module DSL
+    class Set
+      def initialize(connection, key)
+        @connection = connection
+        @key        = key
+      end
+
+      attr_reader :connection
+      attr_reader :key
+
+      def members(count)
+        count.times do
+          connection.sadd(key, Rebar::Random.string)
+        end
+      end
+    end
+  end
+end

data/lib/rebar/dsl/sorted_set.rb

@@ -0,0 +1,19 @@
+module Rebar
+  module DSL
+    class SortedSet
+      def initialize(connection, key)
+        @connection = connection
+        @key        = key
+      end
+
+      attr_reader :connection
+      attr_reader :key
+
+      def members(count)
+        count.times do
+          connection.zadd(key, Rebar::Random.score, Rebar::Random.string)
+        end
+      end
+    end
+  end
+end

data/lib/rebar/random.rb

@@ -0,0 +1,13 @@
+require 'digest'
+
+module Rebar
+  module Random
+    def self.score
+      rand * 1_000_000_000
+    end
+
+    def self.string
+      Digest::SHA256.hexdigest(Time.now.to_s + rand.to_s)
+    end
+  end
+end

data/lib/rebar/version.rb

@@ -0,0 +1,3 @@
+module Rebar
+  VERSION = "0.2.0"
+end

data/rebar.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/rebar/version', __FILE__)
+
+Gem::Specification.new do |s|
+  s.platform    = Gem::Platform::RUBY
+  s.name        = 'rebar'
+  s.version     = Rebar::VERSION
+  s.summary     = 'A simple stress testing tool for Redis'
+  s.description = ''
+  s.license     = 'MIT'
+
+  s.required_ruby_version = '>= 1.9.2'
+
+  s.author        = 'Paul Rosania'
+  s.email         = 'paul.rosania@gmail.com'
+  s.homepage      = "https://github.com/paulrosania/rebar"
+
+  s.executables   = `git ls-files -- bin/*`.split("\n").map { |f| File.basename(f) }
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.require_path  = 'lib'
+
+  s.add_runtime_dependency('redis',           '>= 2.2')
+  s.add_runtime_dependency('redis-namespace', '>= 1.1')
+end

metadata

@@ -0,0 +1,83 @@
+--- !ruby/object:Gem::Specification
+name: rebar
+version: !ruby/object:Gem::Version
+  version: 0.2.0
+  prerelease: 
+platform: ruby
+authors:
+- Paul Rosania
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-14 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: &70350297747800 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '2.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70350297747800
+- !ruby/object:Gem::Dependency
+  name: redis-namespace
+  requirement: &70350297746520 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70350297746520
+description: ''
+email: paul.rosania@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- lib/rebar.rb
+- lib/rebar/builder.rb
+- lib/rebar/dsl.rb
+- lib/rebar/dsl/database.rb
+- lib/rebar/dsl/hash.rb
+- lib/rebar/dsl/list.rb
+- lib/rebar/dsl/set.rb
+- lib/rebar/dsl/sorted_set.rb
+- lib/rebar/random.rb
+- lib/rebar/version.rb
+- rebar.gemspec
+homepage: https://github.com/paulrosania/rebar
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.11
+signing_key: 
+specification_version: 3
+summary: A simple stress testing tool for Redis
+test_files: []