kontext 0.1.0

data/.rspec

@@ -0,0 +1 @@
+--color --order random

data/Gemfile

@@ -0,0 +1,6 @@
+source :rubygems
+
+gemspec
+
+gem "rake"
+gem "awesome_print"

data/README.md

@@ -0,0 +1,48 @@
+Thread and Fiber specific global state
+======================================
+
+Know DataMapper's repository contexts? Need request-specific global state in an
+evented webserver? Wanna do complex AOP but care about concurrency? Kontext
+comes to your rescue. It provides you with nestable per-thread, per-fiber
+context stacks.
+
+```ruby
+require "kontext"
+
+context = Kontext.new
+
+context.with "foo" do
+  p context.last
+  # => "foo"
+
+  context.with "bar" do
+    p context.stack
+    # => ["foo", "bar"]
+  end
+
+  Fiber.new do
+    p context.last
+    # => nil
+
+    context.with "different context!" do
+      p context.last
+      # => "different context!"
+
+      Thread.start do
+        context.with "again different" do
+          p context.stack
+          # => ["again different"]
+        end
+      end
+
+      sleep 0.1
+    end
+  end.resume
+end
+```
+
+To do
+-----
+
+* Fix memory leak in `#store`
+* Make new threads and fibers inherit a copy of the previous context stack

data/Rakefile

@@ -0,0 +1,4 @@
+task :default => :spec
+
+require "rspec/core/rake_task"
+RSpec::Core::RakeTask.new :spec

data/kontext.gemspec

@@ -0,0 +1,20 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "kontext/version"
+
+Gem::Specification.new do |s|
+  s.name        = "kontext"
+  s.version     = Kontext::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Lars Gierth"]
+  s.email       = ["lars.gierth@gmail.com"]
+  s.homepage    = "https://rubygems.org/gems/kontext"
+  s.summary     = %q{Thread and Fiber specific global state}
+  s.description = s.summary
+
+  s.add_development_dependency "rspec"
+
+  s.files         = `git ls-files`.split("\n") - [".gitignore", ".travis.yml"]
+  s.test_files    = `git ls-files -- spec/*`.split("\n")
+  s.require_paths = ["lib"]
+end

data/lib/kontext/version.rb

@@ -0,0 +1,3 @@
+class Kontext
+  VERSION = "0.1.0"
+end

data/lib/kontext.rb

@@ -0,0 +1,67 @@
+require "kontext/version"
+
+begin
+  require "fiber"
+rescue LoadError => ex
+  raise(ex) if ex.message =~ "fiber"
+end
+
+class Kontext
+  RootFiber = defined?(::Fiber) ? Fiber.current : nil
+
+  def thread?
+    Thread.current != Thread.main
+  end
+
+  def fiber?
+    defined?(::Fiber) && Fiber.current != RootFiber
+  end
+
+  def store
+    if fiber?
+      # TODO over time this will leak massively
+      Thread.current[Fiber.current.object_id.to_s] ||= {}
+    elsif thread?
+      Thread.current
+    else
+      @store ||= {}
+    end
+  end
+
+  def stack
+    store[self] ||= []
+  end
+
+  def with(obj)
+    old_size = size
+    push(obj)
+
+    begin
+      result = yield(obj)
+    ensure
+      truncate(old_size)
+    end
+
+    result
+  end
+
+  def push(obj)
+    stack.push(obj)
+  end
+
+  def pop
+    stack.pop
+  end
+
+  def last
+    stack.last
+  end
+
+  def size
+    stack.length
+  end
+
+  def truncate(new_size)
+    pop until size == new_size
+  end
+end

data/spec/kontext_spec.rb

@@ -0,0 +1,169 @@
+require "awesome_print"
+require "kontext"
+
+describe Kontext do
+  describe Kontext::RootFiber do
+    it "is a reference to the root fiber" do
+      Kontext::RootFiber.should equal(Fiber.current)
+    end
+  end
+
+  let :kontext do
+    Kontext.new
+  end
+
+  describe "#thread?" do
+    it "returns true if the current thread is not the main thread" do
+      result = nil
+      Thread.new { result = kontext.thread? }
+      sleep 0.01
+
+      result.should be_true
+    end
+
+    it "returns false if we're in the main thread" do
+      kontext.thread?.should be_false
+    end
+  end
+
+  describe "#fiber?" do
+    it "returns true if the current fiber is not the root fiber" do
+      result = nil
+      Fiber.new { result = kontext.fiber? }.resume
+
+      result.should be_true
+    end
+
+    it "returns false if we're in the root fiber" do
+      kontext.fiber?.should be_false
+    end
+  end
+
+  describe "#store" do
+    describe "when in a fiber" do
+      before do
+        kontext.stub :fiber? => true, :thread? => false
+      end
+
+      it "returns a fiber specific store" do
+        key = Fiber.current.object_id.to_s
+        kontext.store.should equal(Thread.current[key])
+      end
+    end
+
+    describe "when in a thread" do
+      before do
+        kontext.stub :fiber? => false, :thread? => true
+      end
+
+      it "returns a thread specific hash store" do
+        kontext.store.should equal(Thread.current)
+      end
+    end
+
+    describe "when neither in thread nor in fiber" do
+      before do
+        kontext.stub :fiber? => false, :thread? => false
+      end
+
+      it "returns a hash" do
+        kontext.store.should be_a(Hash)
+        kontext.store.should be_empty
+      end
+    end
+  end
+
+  describe "#stack" do
+    it "returns the instance's stack" do
+      kontext.stack.should be_an(Array)
+      kontext.stack.should be_empty
+      kontext.stack.should equal(kontext.store[kontext])
+    end
+  end
+
+  describe "#with(obj)" do
+    let(:obj) { stub "obj" }
+
+    it "pushes the object to the stack" do
+      kontext.with obj do
+        kontext.last.should equal(obj)
+
+        kontext.with obj do
+          kontext.stack.should == [obj, obj]
+        end
+      end
+    end
+
+    it "yields the object" do
+      kontext.with obj do |o|
+        o.should equal(obj)
+      end
+    end
+
+    it "correctly updates the stack when the block raises" do
+      begin
+        kontext.with obj do
+          begin
+            kontext.with(obj) { raise }
+          ensure
+            kontext.stack.should == [obj]
+          end
+        end
+      rescue
+        kontext.stack.should be_empty
+      end
+    end
+
+    it "expects a block" do
+      proc { kontext.with obj }.should raise_error(LocalJumpError)
+    end
+  end
+
+  describe "#push(obj)" do
+    it "adds an object to the stack" do
+      kontext.push :foo
+      kontext.push :bar
+      kontext.last.should equal(:bar)
+    end
+  end
+
+  describe "#pop" do
+    it "removes an object from the top of the stack" do
+      kontext.pop.should be_nil
+
+      kontext.push :foo
+      kontext.pop.should equal(:foo)
+      kontext.size.should be_zero
+    end
+  end
+
+  describe "#last" do
+    it "returns the object at the top of the stack" do
+      kontext.last.should be_nil
+
+      kontext.push :foo
+      kontext.last.should equal(:foo)
+    end
+  end
+
+  describe "#size" do
+    it "is zero initially" do
+      kontext.size.should be_zero
+    end
+
+    it "increases by 1 for each nesting level" do
+      kontext.with :foo do
+        kontext.size.should equal(1)
+      end
+    end
+  end
+
+  describe "#truncate(new_size)" do
+    it "pops objects until there are $new_size object left" do
+      [:foo, :bar, :baz].each &kontext.method(:push)
+
+      kontext.truncate 1
+      kontext.last.should equal(:foo)
+    end
+  end
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification
+name: kontext
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Lars Gierth
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-20 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &85309510 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *85309510
+description: Thread and Fiber specific global state
+email:
+- lars.gierth@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .rspec
+- Gemfile
+- README.md
+- Rakefile
+- kontext.gemspec
+- lib/kontext.rb
+- lib/kontext/version.rb
+- spec/kontext_spec.rb
+homepage: https://rubygems.org/gems/kontext
+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.16
+signing_key: 
+specification_version: 3
+summary: Thread and Fiber specific global state
+test_files:
+- spec/kontext_spec.rb
+has_rdoc: