fork_buddy 0.0.1 → 0.0.2

data/example-client.rb

@@ -0,0 +1,5 @@
+require 'fork_buddy'
+
+fb = ForkBuddy.new(socket: '/tmp/fork_buddy.sock', rails_env: 'test')
+
+fb.acquire(:rspec_ready, rails_env: 'test')

data/example.rb

@@ -0,0 +1,39 @@
+require 'fork_buddy/server'
+
+fbs = ForkBuddy::Server.start(socket: '/tmp/fork_buddy.sock') do
+  stage :initial {
+    option :rails_env
+    stage :require_bundler {
+      stage :require_application {
+        stage :application_loaded {
+          stage :rspec_ready
+          stage :cucumber_ready
+        }
+      }
+    }
+  }
+end
+
+fbs.stage(:initial)
+
+require 'rubygems'
+require 'bundler'
+
+fbs.stage(:require_bundler)
+
+Bundler.require :default, fbs.option(:rails_env)
+
+fbs.stage(:require_application)
+
+ENV['RAILS_ENV'] = fbs.option(:rails_env)
+require 'application and stuff'
+
+fbs.stage(:application_booted)
+
+fbs.stage(:rspec_ready) do
+  require 'rspec/rails'
+end
+
+fbs.stage(:cucumber_ready) do
+  require 'cucumber/rails'
+end

data/lib/fork_buddy.rb

@@ -1,5 +1,106 @@
 require "fork_buddy/version"
+require 'socket'
 
-module ForkBuddy
-  # Your code goes here...
+$_stack = []
+$_children = []
+
+module Kernel
+  alias_method :__fork, :fork
+  def fork(&block)
+    $_children ||= []
+    r = __fork(&block)
+    $_children << r
+    r
+  end
+end
+
+class ForkBuddy
+
+  class Server
+    def initialize
+      $socket, server = UNIXSocket.pair
+      if __fork
+        @processes = {}
+        begin
+          while line = server.recv(2**12) do
+            process_command(line)
+          end
+        ensure
+          @processes.each { |k, data|
+            Process.kill("QUIT", data[:pid])
+          }
+        end
+      end
+    end
+
+    def process_command(line)
+      data = Marshal.load(line)
+      puts "*** #{data.inspect}"
+
+      case data[:command]
+      when :register
+        data.delete(:command)
+        @processes[data[:forkpoint]] = data
+      when :invalidate
+        forkpoint = data[:forkpoint]
+        @processes.each { |k,v|
+          if k == forkpoint || v[:stack].include?(forkpoint)
+            Process.kill("QUIT", v[:pid])
+            v[:prune] = true
+          end
+        }
+
+        restart_forkpoint = @processes.find{|k,v|k==forkpoint}[1][:stack].first
+        restart = @processes.find{|k,v|k==restart_forkpoint}
+        @processes.reject!{|k,v|v[:prune]}
+
+        Process.kill("USR1", restart[1][:pid])
+      end
+    end
+
+    def forkpoint(sym, &block)
+      loop do
+        if pid = __fork
+          data = {
+            command:   :register,
+            forkpoint: sym,
+            stack:     $_stack,
+            pid:       pid
+          }
+          $socket.send Marshal.dump(data), 0
+
+          # Spin until sent the USR1 signal, then run the next iteration of the loop.
+          Kernel.trap("USR1") { throw(:next) }
+          Kernel.trap("USR2") {
+            # Inject code somehow.
+          }
+          catch(:next) { loop { sleep 10 } }
+        else
+          forkpoint_child(block, sym)
+          exit(0) # end of execution reached
+        end
+      end
+    end
+
+    def forkpoint_child(block, sym)
+      Kernel.trap("QUIT") {
+        ($_children||[]).each { |child|
+          puts "*** REAPING: #{child}"
+          Process.kill("KILL", child) rescue nil
+        }
+        exit
+      }
+      $_stack.unshift sym
+      block.call
+    end
+
+    def invalidate(sym)
+      data = {
+        command:   :invalidate,
+        forkpoint: sym
+      }
+      $socket.send Marshal.dump(data), 0
+    end
+
+  end
 end

data/lib/fork_buddy/version.rb

@@ -1,3 +1,3 @@
 module ForkBuddy
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: fork_buddy
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: 
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Burke Libbey
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-08-19 00:00:00 Z
+date: 2011-08-20 00:00:00 Z
 dependencies: []
 
 description: Write a gem description
@@ -31,9 +31,13 @@ files:
 - .gitignore
 - Gemfile
 - Rakefile
+- example-client.rb
+- example.rb
 - fork_buddy.gemspec
 - lib/fork_buddy.rb
 - lib/fork_buddy/version.rb
+- spec/fork_buddy_spec.rb
+- spec/spec_helper.rb
 homepage: ""
 licenses: []
 
@@ -67,5 +71,6 @@ rubygems_version: 1.7.2
 signing_key: 
 specification_version: 3
 summary: Write a gem summary
-test_files: []
-
+test_files: 
+- spec/fork_buddy_spec.rb
+- spec/spec_helper.rb