hashafras 0.0.1

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*
+*.swp

data/Gemfile

@@ -0,0 +1,3 @@
+source "http://rubygems.org"
+
+gemspec

data/Rakefile

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

data/hashafras.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "hashafras/version"
+
+Gem::Specification.new do |s|
+  s.name        = "hashafras"
+  s.version     = Hashafras::VERSION
+  s.authors     = ["Tim Ariyeh"]
+  s.email       = ["tim.ariyeh@gmail.com"]
+  s.homepage    = "https://github.com/timariyeh/hashafras"
+  s.summary     = %q{Stupid-simple consistent hashing for Ruby}
+  s.description = %q{}
+
+  s.rubyforge_project = "hashafras"
+
+  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"]
+
+  s.add_development_dependency "rspec"
+end

data/lib/hashafras/ring.rb

@@ -0,0 +1,64 @@
+require 'zlib'
+
+module Hashafras
+  class Ring
+    DEFAULTS = {:replicas => 20}
+
+    attr_accessor :options
+
+    def initialize(options = {})
+      @options = DEFAULTS.merge(options)
+    end
+
+    def add_member(name, host)
+      options[:replicas].times do |t|
+        key = hash("#{name}_#{t}")
+        members[key] = host
+        member_positions << key
+        member_positions.sort!
+      end
+    end
+
+    def remove_member(name)
+      options[:replicas].times do |t|
+        key = hash("#{name}_#{t}")
+        members.delete(key)
+        member_positions.delete(key)
+        member_positions.compact!
+      end
+    end
+    
+    def members
+      @members ||= {}
+    end
+
+    def member_positions
+      @member_positions ||= []
+    end
+
+    def hash(key)
+      Zlib.crc32("#{key}")
+    end
+
+    def find_host_for_key(key)
+      return nil if members.empty?
+      return members.first if members.size == 1
+
+      hash_value = hash(key)
+      return members[hash_value] if members[hash_value]
+      
+      return find_nearest_member(hash_value)
+    end
+
+    def find_nearest_member(key)
+      member = nil
+      member_positions.each do |m_position|
+        if m_position > key
+          member = members[m_position]
+          break
+        end
+      end
+      member ||= members.first.last
+    end
+  end
+end

data/lib/hashafras/version.rb

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

data/lib/hashafras.rb

@@ -0,0 +1,2 @@
+require "hashafras/version"
+require "hashafras/ring"

data/spec/.rspec

@@ -0,0 +1,2 @@
+--color
+--format nested

data/spec/ring_spec.rb

@@ -0,0 +1,80 @@
+require 'bundler'
+Bundler.require
+require 'benchmark'
+describe Hashafras::Ring do
+  context "has only one member" do
+    let(:ring){
+      Hashafras::Ring.new.tap do |o|
+        o.add_member("s1", "s1:80")
+      end
+    }
+    it "should always return the same member" do
+      ring.find_host_for_key("foo").should == "s1:80"
+      ring.find_host_for_key("bar").should == "s1:80"
+      ring.find_host_for_key("baz").should == "s1:80"
+    end
+  end
+
+  context "has many members" do
+    def members
+      @members ||= 10.times.inject([]) {|memo,obj| memo.push(:name => "s#{obj}", :host => "s#{obj}:80") }
+    end
+
+    def iterations
+      1000
+    end
+    let(:ring){
+      Hashafras::Ring.new.tap do |o|
+        members.each do |s|
+          o.add_member(s[:name], s[:host])
+        end
+      end
+    }
+    it "should evenly distribute keys" do
+      results = {}
+      iterations.times do |key|
+        host = ring.find_host_for_key(key)
+        results[host] ||= 0
+        results[host] += 1
+      end
+      members.count.should == results.keys.count
+      max = results.max.last.to_f
+      min = results.min.last.to_f
+
+      (min / max).should > 0.8
+    end
+
+    context "when topology changes" do
+      def results 
+        results = {}
+        iterations.times do |key|
+          host = ring.find_host_for_key(key)
+          results[key] = host
+        end
+        results
+      end
+
+      def result_diff(result1, result2)
+        result1.reject {|k,v| result2[k] == v}
+      end
+
+      it "should be minimally disruptive to the keyspace when nodes are added" do
+        original_results = results
+        ring.add_member("new_host", "new_host:80")
+        updated_results = results
+
+        diff = result_diff(original_results, updated_results)
+        diff.count.should <= (iterations / (members.count))
+      end
+
+      it "should be minimally disruptive to the keyspace when nodes are removed" do
+        original_results = results
+        ring.remove_member("new_host")
+        updated_results = results
+
+        diff = result_diff(original_results, updated_results)
+        diff.count.should <= (iterations / (members.count))
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,65 @@
+--- !ruby/object:Gem::Specification
+name: hashafras
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Tim Ariyeh
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-10-05 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &8301720 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *8301720
+description: ''
+email:
+- tim.ariyeh@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- Rakefile
+- hashafras.gemspec
+- lib/hashafras.rb
+- lib/hashafras/ring.rb
+- lib/hashafras/version.rb
+- spec/.rspec
+- spec/ring_spec.rb
+homepage: https://github.com/timariyeh/hashafras
+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: hashafras
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Stupid-simple consistent hashing for Ruby
+test_files: []