hash_ring 0.0.1

data/ext/hash_ring/extconf.rb

@@ -0,0 +1,9 @@
+require 'mkmf'
+
+extension_name = 'hash_ring'
+
+dir_config(extension_name)
+
+have_library("hashring")
+
+create_makefile("hash_ring/hash_ring")

data/ext/hash_ring/hash_ring.c

@@ -0,0 +1,109 @@
+#include "ruby.h"
+#include "hash_ring.h"
+
+VALUE cHashRing;
+
+static VALUE hash_ring_dealloc(void * ring)
+{
+  hash_ring_free(ring);
+}
+
+static VALUE hash_ring_alloc(VALUE klass)
+{
+  hash_ring_t * ring;
+  VALUE obj = Data_Make_Struct(klass, hash_ring_t, NULL, hash_ring_dealloc, ring);
+  return obj;
+}
+
+static VALUE method_initialize(VALUE self, VALUE num_replicas, VALUE hash_function)
+{
+  Check_Type(num_replicas,  T_FIXNUM);
+  Check_Type(hash_function, T_FIXNUM);
+
+  hash_ring_t * ring;
+  hash_ring_t * new_ring;
+
+  int nr_int = FIX2INT(num_replicas);
+  int hf_int = FIX2INT(hash_function);
+
+  Data_Get_Struct(self, hash_ring_t, ring);
+
+  new_ring = hash_ring_create(nr_int, hf_int);
+
+  MEMCPY(ring, new_ring, hash_ring_t, 1);
+
+  hash_ring_free(new_ring);
+
+  return self;
+}
+
+static VALUE method_add_node(VALUE self, VALUE name)
+{
+  Check_Type(name, T_STRING);
+  hash_ring_t *ring;
+
+  Data_Get_Struct(self, hash_ring_t, ring);
+
+  char * cname = StringValuePtr(name);
+  hash_ring_add_node(ring, cname, strlen(cname));
+
+  return self;
+}
+
+static VALUE method_remove_node(VALUE self, VALUE name)
+{
+  Check_Type(name, T_STRING);
+
+  hash_ring_t * ring;
+
+  char * cname = StringValuePtr(name);
+
+  Data_Get_Struct(self, hash_ring_t, ring);
+
+  hash_ring_remove_node(ring, cname, strlen(cname));
+
+  return self;
+}
+
+static VALUE method_print_ring(VALUE self)
+{
+  hash_ring_t * ring;
+  Data_Get_Struct(self, hash_ring_t, ring);
+
+  hash_ring_print(ring);
+  return self;
+}
+
+static VALUE method_find_node(VALUE self, VALUE key){
+  Check_Type(key, T_STRING);
+
+  char * ckey = StringValuePtr(key);
+
+  hash_ring_t * ring;
+  Data_Get_Struct(self, hash_ring_t, ring);
+
+  hash_ring_node_t *node = hash_ring_find_node(ring, ckey, strlen(ckey));
+
+  VALUE cs;
+
+  if(node == NULL){
+    rb_raise(rb_eRuntimeError, "No nodes found");
+  } else {
+    cs = rb_str_new(node->name, node->nameLen);
+  }
+
+  return cs;
+}
+
+void Init_hash_ring()
+{
+  cHashRing = rb_define_class("HashRing", rb_cObject);
+
+  rb_define_alloc_func(cHashRing, hash_ring_alloc);
+
+  rb_define_method(cHashRing, "init",         method_initialize,   2);
+  rb_define_method(cHashRing, "addNode",      method_add_node,     1);
+  rb_define_method(cHashRing, "removeNode",   method_remove_node,  1);
+  rb_define_method(cHashRing, "findNode",     method_find_node,    1);
+  rb_define_method(cHashRing, "printRing",    method_print_ring,   0);
+}

data/lib/hash_ring.rb

@@ -0,0 +1,37 @@
+require 'hash_ring/hash_ring'
+
+class HashRing
+  VERSION = "0.0.1"
+
+  attr_accessor :nodes, :num_replicas, :hash_function
+
+  def initialize(num_replicas, hash_function)
+    @nodes = []
+    @num_replicas = num_replicas
+    @hash_function = hash_function
+    init(num_replicas, hash_function)
+  end
+
+  def add_node(name)
+    @nodes << name
+    @nodes.uniq!
+    self.addNode(name)
+  end
+
+  def remove_node(name)
+    @nodes.reject!{|n| n == name}
+    self.removeNode(name)
+  end
+
+  def find_node(name)
+    self.findNode(name)
+  end
+
+  def print
+    self.printRing
+  end
+
+  def to_s
+    "<HashRing:#{self.object_id} @num_replicas=#{@num_replicas} @hash_function=#{@hash_function} @nodes=#{@nodes}>"
+  end
+end

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: hash_ring
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- Michael R. Bernstein
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-12 00:00:00.000000000Z
+dependencies: []
+description: Wraps https://github.com/mikeyk/hash-ring
+email: michaelrbernstein@gmail.com
+executables: []
+extensions:
+- ext/hash_ring/extconf.rb
+extra_rdoc_files: []
+files:
+- lib/hash_ring.rb
+- ext/hash_ring/hash_ring.c
+- ext/hash_ring/extconf.rb
+homepage: http://github.com/mrb/hash_ring
+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: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: A wrapper for a fast C hash ring library
+test_files: []