alloc8 0.0.1

data/.gitignore

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

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in alloc8.gemspec
+gemspec

data/README.md

@@ -0,0 +1,18 @@
+Simple distributed resource allocation using Redis.
+
+<pre>
+require 'alloc8'
+
+HOST = "my-redis"
+PORT = 6379 # default Redis port
+
+# to create resources
+@allocator = Alloc8::Tor.new(HOST, PORT)
+@allocator.create("worker", "worker1.my.org")
+@allocator.create("worker", "worker2.my.org")
+
+# to use them
+Alloc8::Tor.with_resource("worker", HOST, PORT) do |worker|
+  # something
+end
+</pre>

data/Rakefile

@@ -0,0 +1,6 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+RSpec::Core::RakeTask.new(:spec)
+task :default => :spec

data/alloc8.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "alloc8/version"
+
+Gem::Specification.new do |s|
+  s.name        = "alloc8"
+  s.version     = Alloc8::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Ariel Salomon"]
+  s.email       = ["ariel@oscillatory.org"]
+  s.homepage    = ""
+  s.summary     = %q{Simple distributed resource allocator based on Redis}
+  s.description = %q{Simple distributed resource allocation using Redis.
+    Alloc8::Tor.with_resource do ...}
+
+  s.rubyforge_project = "alloc8"
+
+  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_runtime_dependency "redis", "~> 2.2.0"
+  s.add_development_dependency "rspec", "~> 2.0"
+end

data/lib/alloc8.rb

@@ -0,0 +1,52 @@
+require 'redis'
+
+module Alloc8
+  class Tor
+    # initialize connection to a redis
+    def initialize(host, port, db=nil)
+      @redis = Redis.new(:host => host, :port => port)
+      @redis.select(db) if db
+    end
+
+    # create a resource in a class
+    def create(resource_class, resource)
+      if @redis.sadd("resource:#{resource_class}", resource.to_s)
+        @redis.lpush "resource:#{resource_class}:avail", resource.to_s
+      end
+    end
+
+    # delete a resource
+    def purge(resource_class)
+      nkeys = @redis.del "resource:#{resource_class}:avail", "resource:#{resource_class}"
+      nkeys == 2 # true if purged all
+    end
+
+    # reset available items in resource class
+    def reset(resource_class)
+      @redis.del "resource:#{resource_class}:avail"
+      @redis.smembers("resource:#{resource_class}").each do |resource|
+        @redis.lpush "resource:#{resource_class}:avail", resource.to_s
+      end
+    end
+
+    # acquire next available resource
+    def acquire(resource_class, timeout = 0)
+      key, res = @redis.brpop "resource:#{resource_class}:avail", timeout
+      res
+    end
+
+    # return resource
+    def return(resource_class, resource)
+      raise Exception if !@redis.sismember "resource:#{resource_class}", resource.to_s
+      @redis.lpush "resource:#{resource_class}:avail", resource.to_s
+    end
+
+    # block to allocate a resource
+    def self.with_resource(resource_class, host, port, db=nil)
+       a = Alloc8::Tor.new host, port, db
+       res = a.acquire(resource_class)
+       yield res
+       a.return(resource_class, res)
+    end
+  end
+end

data/lib/alloc8/version.rb

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

data/spec/alloc8_spec.rb

@@ -0,0 +1,69 @@
+require 'alloc8'
+
+describe Alloc8 do
+  KIND = "ResourceClass"
+  HOST = "localhost"
+  PORT = 6379 # default redis port
+
+  before {
+    @allocator = Alloc8::Tor.new(HOST, PORT)
+    @allocator.purge(KIND)
+  }
+
+  context "having no resources" do
+    it "can create a resource" do
+      @allocator.create(KIND, "a").should == 1
+    end
+  end
+
+  context "having a single resource" do
+    before { @allocator.create KIND, "a" }
+
+    it "can't allocate it again" do
+      @allocator.create(KIND, "a").should == nil
+    end
+
+    it "can allocate other resources" do
+      @allocator.create(KIND, "b").should == 2
+      @allocator.create(KIND, "c").should == 3
+    end
+
+    it "can purge resource class" do
+      @allocator.purge(KIND).should == true
+    end
+
+    it "can acquire and return a resource" do
+      @allocator.acquire(KIND).should == "a"
+      @allocator.return(KIND, "a").should == 1
+    end
+
+    it "can acquire with a block" do
+      Alloc8::Tor.with_resource(KIND, HOST, PORT) do |res|
+        res.should == "a"
+      end
+    end
+
+    it "can't acquire a second resource" do
+      @allocator.acquire(KIND)
+      @allocator.acquire(KIND, 1).should == nil # with 1s timeout
+    end
+  end
+
+  context "having multiple resources" do
+    before { ["a", "b", "c"].each {|r| @allocator.create KIND, r} }
+
+    it "can purge resource class" do
+      @allocator.purge(KIND).should == true
+    end
+
+    it "can acquire and return multiple resources" do
+      @allocator.acquire(KIND).should == "a"
+      @allocator.acquire(KIND).should == "b"
+      @allocator.acquire(KIND).should == "c"
+      @allocator.return(KIND, "c").should == 1
+      @allocator.return(KIND, "b").should == 2
+      @allocator.return(KIND, "a").should == 3
+    end
+  end
+
+end

metadata

@@ -0,0 +1,107 @@
+--- !ruby/object:Gem::Specification 
+name: alloc8
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Ariel Salomon
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-20 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: redis
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 2
+        - 2
+        - 0
+        version: 2.2.0
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 2
+        - 0
+        version: "2.0"
+  type: :development
+  version_requirements: *id002
+description: |-
+  Simple distributed resource allocation using Redis.
+      Alloc8::Tor.with_resource do ...
+email: 
+- ariel@oscillatory.org
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- README.md
+- Rakefile
+- alloc8.gemspec
+- lib/alloc8.rb
+- lib/alloc8/version.rb
+- spec/alloc8_spec.rb
+has_rdoc: true
+homepage: ""
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: alloc8
+rubygems_version: 1.5.3
+signing_key: 
+specification_version: 3
+summary: Simple distributed resource allocator based on Redis
+test_files: 
+- spec/alloc8_spec.rb