asir_beanstalk 1.1.1

data/.gitignore

@@ -0,0 +1,18 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+example/*.out

data/.rspec

@@ -0,0 +1 @@
+--color -f d

data/Changelog

@@ -0,0 +1,4 @@
+2012-12-04  Kurt A. Stephens  <ks.git@kurtstephens.com>
+
+	* v1.1.1: Initial version: Pulled from asir v1.1.0.
+

data/Gemfile

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

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2012 Kurt Stephens
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,29 @@
+# asir_beanstalk
+
+Beanstalk ASIR transport.
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'asir_beanstalk'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install asir_beanstalk
+
+## Usage
+
+See example/.
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,34 @@
+require "bundler/gem_tasks"
+
+gem 'rspec'
+require 'rspec/core/rake_task'
+
+desc "Run specs"
+RSpec::Core::RakeTask.new(:spec) do |t|
+  t.pattern = "./spec/**/*_spec.rb" # don't need this, it's default.
+  # Put spec opts in a file named .rspec in root
+end
+
+desc "Generate code coverage"
+RSpec::Core::RakeTask.new(:coverage) do |t|
+  t.pattern = "./spec/**/*_spec.rb" # don't need this, it's default.
+  t.rcov = true
+  t.rcov_opts = ['--exclude', 'spec']
+end
+
+######################################################################
+
+desc "Default => :test"
+task :default => :test
+
+desc "Run all tests"
+task :test => [ :spec ]
+
+desc "Run examples."
+task :example do
+  ENV["ASIR_EXAMPLE_SILENT"]="1"
+  Dir["example/ex[0-9]*.rb"].each do | rb |
+    sh %Q{ruby -I example -I lib #{rb}}
+  end
+  ENV.delete("ASIR_EXAMPLE_SILENT")
+end

data/asir_beanstalk.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+# -*- ruby -*-
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'asir_beanstalk/version'
+
+Gem::Specification.new do |gem|
+  gem.name          = "asir_beanstalk"
+  gem.version       = AsirBeanstalk::VERSION
+  gem.authors       = ["Kurt Stephens"]
+  gem.email         = ["ks.github@kurtstephens.com"]
+  gem.description   = %q{Beanstalkd transport for ASIR}
+  gem.summary       = %q{Beanstalkd transport for ASIR}
+  gem.homepage      = "http://github.com/kstephens/abstracting_services_in_ruby"
+
+  gem.files         = `git ls-files`.split($/)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
+  gem.require_paths = ["lib"]
+
+  gem.add_dependency "asir", "~> 1.1.0"
+
+  gem.add_development_dependency 'rake', '>= 0.9.0'
+  gem.add_development_dependency 'rspec', '~> 2.12.0'
+  gem.add_development_dependency 'simplecov', '>= 0.1'
+end

data/example/ex01.rb

@@ -0,0 +1,34 @@
+# !SLIDE :capture_code_output true
+# Asynchronous beanstalkd service
+
+require 'example_helper'
+require 'asir/transport/beanstalk'
+require 'asir/coder/zlib'
+begin
+  Email.asir.transport = t =
+    ASIR::Transport::Beanstalk.new(:address => '127.0.0.1', :port => 30904)
+  t.encoder =
+    ASIR::Coder::Chain.new(:encoders =>
+                           [ ASIR::Coder::Marshal.new,
+                            ASIR::Coder::Zlib.new, ])
+  t.start_conduit!; sleep 1
+  pr Email.asir.send_email(:pdf_invoice,
+                             :to => "user@email.com", :customer => @customer)
+  sleep 2
+  server_process do
+    t.prepare_server!
+    t.run_server!
+  end
+rescue Object => err
+  $stderr.puts "#{err.inspect}\n#{err.backtrace * "\n"}"
+ensure
+  t.close; sleep 3; server_kill; sleep 2
+  t.stop_conduit!
+end
+
+# !SLIDE END
+# EXPECT: : client process
+# EXPECT: : server process
+# EXPECT: : Email.send_mail :pdf_invoice
+# EXPECT: : pr: nil
+

data/example/example_helper.rb

@@ -0,0 +1,77 @@
+# Sample client support
+#
+require 'rubygems'
+case RUBY_PLATFORM
+when /java/i
+  gem 'spoon'; require 'spoon'
+end
+
+$: << File.expand_path("../../lib", __FILE__)
+gem 'asir'
+
+require 'asir'
+require 'asir/coder/chain'
+require 'asir/coder/marshal'
+ASIR::Log.enabled = true unless ENV['ASIR_EXAMPLE_SILENT']
+require 'sample_service'
+
+require 'pp'
+
+@customer = 123
+
+class ::Object
+
+def pr result
+  $stdout.puts "*** #{$$}: pr: #{PP.pp(result, '')}"
+end
+
+def server_process &blk
+  # $stderr.puts "  at #{__FILE__}:#{__LINE__}"
+  case RUBY_PLATFORM
+  when /java/i
+    # JRuby cannot fork.
+    # So we must prevent spawn a new jruby and
+    # instruct it to only run the server blk, and not
+    # the subsequent client code.
+    # In other words, we cannot rely on how Process.fork
+    # terminates within the block.
+    if ENV['ASIR_JRUBY_SPAWNED']
+      $stderr.puts "  spawned server at #{__FILE__}:#{__LINE__}"
+      puts "*** #{$$}: server process"; $stdout.flush
+      yield
+      Process.exit!(0)
+      # dont do client, client is our parent process.
+    else
+      $stderr.puts "  spawning at #{__FILE__}:#{__LINE__}"
+      ENV['ASIR_JRUBY_SPAWNED'] = "1"
+      cmd = "ruby -I #{File.dirname(__FILE__)} -I #{File.expand_path('../../lib', __FILE__)} #{$0} #{ARGV * ' '}"
+      $stderr.puts "  cmd = #{cmd}"
+      $server_pid = Spoon.spawnp(cmd)
+      ENV.delete('ASIR_JRUBY_SPAWNED')
+      $stderr.puts "  spawned #{$server_pid} at #{__FILE__}:#{__LINE__}"
+    end
+  else
+    # $stderr.puts "  at #{__FILE__}:#{__LINE__}"
+    $server_pid = Process.fork do
+      puts "*** #{$$}: server process"; $stdout.flush
+      yield
+    end
+  end
+  sleep 1 # wait for server to be ready.
+  return false # do client.
+end
+
+def server_kill
+  if $server_pid
+    Process.kill 9, $server_pid
+    Process.waitpid($server_pid)
+  end
+rescue Errno::ESRCH
+ensure
+  $server_pid = nil
+end
+
+end
+
+puts "*** #{$$}: client process"; $stdout.flush
+

data/example/sample_service.rb

@@ -0,0 +1,24 @@
+require 'asir'
+# Added .asir support.
+module Email
+  include ASIR::Client # Email.asir
+  def send_email template_name, options
+    $stderr.puts "*** #{$$}: Email.send_mail #{template_name.inspect} #{options.inspect}"
+    :ok
+  end
+  def do_raise msg
+    raise msg
+  end
+  extend self
+end
+
+class MyClass
+  include ASIR::Client
+  def initialize x
+    @x = x
+  end
+  def method_missing sel, *args
+    @x.send(sel, *args)
+  end
+end
+

data/lib/asir/transport/beanstalk.rb

@@ -0,0 +1,184 @@
+require 'asir/transport/tcp_socket'
+
+module ASIR
+  class Transport
+    # !SLIDE
+    # Beanstalk Transport
+    class Beanstalk < TcpSocket
+      LINE_TERMINATOR = "\r\n".freeze
+
+      attr_accessor :tube, :priority, :delay, :ttr
+
+      def initialize *args
+        @port ||= 11300
+        @tube ||= 'asir'
+        @priority ||= 0
+        @delay ||= 0
+        @ttr ||= 600
+        super
+      end
+
+      # !SLIDE
+      # Sends the encoded Message payload String.
+      def _send_message message, message_payload
+        stream.with_stream! do | s |
+          begin
+            match = 
+              _beanstalk(s, 
+                         "put #{message[:beanstalk_priority] || @priority} #{message[:beanstalk_delay] || @delay} #{message[:beanstalk_ttr] || @ttr} #{message_payload.size}\r\n",
+                         /\AINSERTED (\d+)\r\n\Z/,
+                         message_payload)
+            job_id = message[:beanstalk_job_id] = match[1].to_i
+            _log { "beanstalk_job_id = #{job_id.inspect}" } if @verbose >= 2
+          rescue ::Exception => exc
+            message[:beanstalk_error] = exc
+            close
+            raise exc
+          end
+        end
+      end
+
+      RESERVE = "reserve\r\n".freeze
+
+      # !SLIDE
+      # Receives the encoded Message payload String.
+      def _receive_message channel, additional_data
+        channel.with_stream! do | stream |
+          begin
+            match = 
+              _beanstalk(stream,
+                         RESERVE,
+                         /\ARESERVED (\d+) (\d+)\r\n\Z/)
+            additional_data[:beanstalk_job_id] = match[1].to_i
+            additional_data[:beanstalk_message_size] = 
+              size = match[2].to_i
+            message_payload = stream.read(size)
+            _read_line_and_expect! stream, /\A\r\n\Z/
+            # Pass the original stream used to #_send_result below.
+            [ message_payload, stream ]
+          rescue ::Exception => exc
+            _log { [ :_receive_message, :exception, exc ] }
+            additional_data[:beanstalk_error] = exc
+            channel.close
+            raise exc
+          end
+        end
+      end
+
+      # !SLIDE
+      # Sends the encoded Result payload String.
+      def _send_result message, result, result_payload, channel, stream
+        #
+        # There is a possibility here the following could happen:
+        #
+        #   _receive_message
+        #     channel == #<Channel:1>   
+        #     channel.stream == #<TCPSocket:1234>
+        #   end
+        #   ...
+        #   ERROR OCCURES:
+        #      channel.stream.close
+        #      channel.stream = nil
+        #   ...
+        #   _send_result 
+        #     channel == #<Channel:1>
+        #     channel.stream == #<TCPSocket:5678> # NEW CONNECTION
+        #     stream.write "delete #{job_id}"
+        #   ...
+        #
+        # Therefore: _receiver_message passes the original message stream to us.
+        # We insure that the same stream is still the active one and use it.
+        channel.with_stream! do | maybe_other_stream |
+          _log [ :_send_result, "stream lost" ] if maybe_other_stream != stream
+          job_id = message[:beanstalk_job_id] or raise "no beanstalk_job_id"
+          _beanstalk(stream,
+                     "delete #{job_id}\r\n",
+                     /\ADELETED\r\n\Z/)
+        end
+      end
+
+      # !SLIDE
+      # Receives the encoded Result payload String.
+      def _receive_result message, opaque_result
+        nil
+      end
+
+      # !SLIDE
+      # Sets beanstalk_delay if message.delay was specified.
+      def relative_message_delay! message, now = nil
+        if delay = super
+          message[:beanstalk_delay] = delay.to_i
+        end
+        delay
+      end
+
+      # !SLIDE
+      # Beanstalk protocol support
+
+      # Send "something ...\r\n".
+      # Expect /\ASOMETHING (\d+)...\r\n".
+      def _beanstalk stream, message, expect, payload = nil
+        _log { [ :_beanstalk, :message, message ] } if @verbose >= 3
+        stream.write message
+        if payload
+          stream.write payload
+          stream.write LINE_TERMINATOR
+        end
+        stream.flush
+        if match = _read_line_and_expect!(stream, expect)
+          _log { [ :_beanstalk, :result, match[0] ] } if @verbose >= 3
+        end
+        match
+      end
+
+      def _after_connect! stream
+        if @tube
+          _beanstalk(stream,
+                     "use #{@tube}\r\n",
+                     /\AUSING #{@tube}\r\n\Z/)
+        end
+      end
+
+      # !SLIDE
+      # Beanstalk Server
+      def _server!
+        _log { "_server! #{uri}" } if @verbose >= 1
+        @server = connect!(:try_max => nil,
+                           :try_sleep => 1,
+                           :try_sleep_increment => 0.1,
+                           :try_sleep_max => 10) do | stream |
+          if @tube
+            _beanstalk(stream, 
+                       "watch #{@tube}\r\n",
+                       /\AWATCHING (\d+)\r\n\Z/)
+          end
+        end
+        self
+      end
+
+      def _server_accept_connection! server
+        prepare_server! unless @server
+        [ @server, @server ]
+      end
+
+      def _server_close_connection! in_stream, out_stream
+        # NOTHING
+      end
+
+      def stream_eof? stream
+        # Note: stream.eof? on a beanstalkd connection,
+        # will cause blocking read *forever* because
+        # beanstalk connections are long lived.
+        false
+      end
+
+      def _start_conduit!
+        addr = address ? "-l #{address} " : ""
+        cmd = "beanstalkd #{addr}-p #{port}"
+        exec(cmd)
+      end
+    end
+    # !SLIDE END
+  end # class
+end # module
+

data/lib/asir_beanstalk.rb

@@ -0,0 +1,8 @@
+require "asir_beanstalk/version"
+
+module AsirBeanstalk
+end
+
+gem 'asir'
+require 'asir'
+require 'asir/transport/beanstalk'

data/lib/asir_beanstalk/version.rb

@@ -0,0 +1,3 @@
+module AsirBeanstalk
+  VERSION = "1.1.1"
+end

data/spec/example_spec.rb

@@ -0,0 +1,88 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+$:.unshift File.expand_path('../../example', __FILE__)
+
+describe "ASIR Example" do
+  attr_accessor :file, :expects
+
+  before(:each) do
+    @expects = [ ]
+  end
+
+  after(:each) do
+    @file.should_not == nil
+    File.open(@file) do | fh |
+      until fh.eof?
+        line = fh.readline
+        line.chomp!
+        case
+        when line.sub!(/^\s*#\s*EXPECT\/:\s*/, '')
+          expect Regexp.new(line)
+        when line.sub!(/^\s*#\s*EXPECT!\/:\s*/, '')
+          expect Regexp.new(line), :'!~'
+        when line.sub!(/^\s*#\s*EXPECT:\s*/, '')
+          expect Regexp.new(Regexp.escape(line))
+        when line.sub!(/^\s*#\s*EXPECT!:\s*/, '')
+          expect Regexp.new(Regexp.escape(line)), :'!~'
+        end
+      end
+    end
+    @output, @exit_code = run_file!(@file)
+    @exit_code.should == 0
+    @expects.empty?.should_not == true
+    @expects.each do | rx, mode |
+      $stderr.puts "    Checking #{mode} #{rx.inspect}" if ENV['SPEC_VERBOSE']
+      case mode
+      when :'=~'
+        @output.should =~ rx
+      when :'!~'
+        @output.should_not =~ rx
+      else
+        raise ArgumentError
+      end
+    end
+  end
+
+  def run_file! file, output = StringIO.new('')
+    progname_save, stdout_save, stderr_save = $0, $stdout, $stderr
+    exc = system_exit = nil; exit_code = 0
+    begin
+      if true
+        cmd = "ASIR_EXAMPLE_SILENT=1 ruby -I example -I lib #{file}"
+        $stderr.puts "\n   Running #{cmd}:" if ENV['SPEC_VERBOSE']
+        output = `#{cmd} 2>&1 | tee #{file}.out`
+      else
+        $stderr.puts "\n   Loading #{file}:" if ENV['SPEC_VERBOSE']
+        $stdout.puts "*** #{$$}: client process"; $stdout.flush
+        $stdout = $stderr = output
+        $0 = file
+        Kernel.load(file, true)
+        output = output.string if StringIO === output
+      end
+    rescue ::SystemExit => system_exit
+      exit_code = 1 # ???
+    rescue ::Exception => exc
+      exit_code = -1
+    end
+    [ output, exit_code ]
+  ensure
+    $0, $stdout, $stderr = progname_save, stdout_save, stderr_save
+    $stderr.write output  if ENV['SPEC_VERBOSE']
+    if exc
+      stderr_save.puts "ERROR: #{file}: #{exc.inspect}\n#{exc.backtrace * "\n"}"
+      raise exc
+    end
+  end
+
+  def expect rx, mode = :'=~'
+    @expects << [ rx, mode ]
+  end
+
+  Dir['example/**/ex[0-9]*.rb'].sort.each do | file |
+    title = File.open(file) { | fh | fh.read(4096) }
+    title = title =~ /#\s+!SLIDE[^\n]*\n\s*#\s*([^\n]+)/ && $1
+    it "#{file} - #{title}" do
+      @file = file
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,37 @@
+require 'rubygems'
+$:.unshift File.expand_path('../../lib', __FILE__)
+
+gem 'asir'
+require 'asir'
+
+module ASIR
+  module Test
+    class TestError < ::Exception; end
+    class TestObject
+      include ASIR::Client
+      attr_accessor :spec, :arg, :cls, :msg, :transport, :message
+      def initialize spec
+        @spec = spec
+      end
+
+      def get_data!
+        @transport = ASIR::Transport.current
+        @message = @transport && @transport.message
+      end
+
+      def return_argument arg
+        get_data!
+        @arg = arg
+        arg
+      end
+
+      def raise_exception! cls, msg
+        get_data!
+        @cls = cls
+        @msg = msg
+        raise cls, msg
+      end
+    end
+  end
+end
+

metadata

@@ -0,0 +1,133 @@
+--- !ruby/object:Gem::Specification
+name: asir_beanstalk
+version: !ruby/object:Gem::Version
+  version: 1.1.1
+  prerelease: 
+platform: ruby
+authors:
+- Kurt Stephens
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-12-04 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: asir
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.9.0
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.12.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.12.0
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0.1'
+description: Beanstalkd transport for ASIR
+email:
+- ks.github@kurtstephens.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- Changelog
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- asir_beanstalk.gemspec
+- example/ex01.rb
+- example/example_helper.rb
+- example/sample_service.rb
+- lib/asir/transport/beanstalk.rb
+- lib/asir_beanstalk.rb
+- lib/asir_beanstalk/version.rb
+- spec/example_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/kstephens/abstracting_services_in_ruby
+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'
+      segments:
+      - 0
+      hash: -1937949383851329736
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: -1937949383851329736
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Beanstalkd transport for ASIR
+test_files:
+- spec/example_spec.rb
+- spec/spec_helper.rb