stomping_ground-celluloid 0.1

data/lib/stomping_ground.rb

@@ -0,0 +1,6 @@
+require 'json'
+require 'celluloid/io'
+require 'fileutils'
+
+require 'stomping_ground/server'
+require 'stomping_ground/connection'

data/lib/stomping_ground/connection.rb

@@ -0,0 +1,114 @@
+module StompingGround
+  class Connection
+    attr_accessor :server, :socket, :frame, :frame_info
+
+    def initialize(server, socket)
+      @server = server
+      @socket = socket
+    end
+
+    def handle
+      loop do
+        break if @socket.closed?
+
+        data = @socket.readpartial 4096
+
+        append_data(data)
+
+        each_frame do |frame|
+          @frame = frame
+          @frame_info = parse(frame)
+          case frame_info[:command]
+          when "CONNECT"
+            connect_frame
+          when "SUBSCRIBE"
+            subscribe_frame
+          when "DISCONNECT"
+            disconnect_frame
+          when "SEND"
+            send_frame
+          end
+        end
+      end
+    rescue EOFError
+      @socket.close
+    end
+
+    def connect_frame
+      write "CONNECTED\n"
+      write "version:1.1\n"
+      write "heart-beat:0,0\n" if frame_info[:'heart-beat']
+      write "\n"
+      write "\0"
+    end
+
+    def subscribe_frame
+      if @server.queue_name.nil? || frame_info[:destination] == @server.queue_name
+        message = @server.message_body || "hello"
+        write "MESSAGE\n"
+        write "subscription:#{frame_info[:id]}\n"
+        write "message-id:007\n"
+        write "destination:#{frame_info[:destination]}\n"
+        write "content-type:text/plain\n"
+        write "content-length:#{message.length}\n"
+        write "\n"
+        write "#{message}\0"
+      end
+    end
+
+    def disconnect_frame
+      if frame_info[:'receipt-id']
+        write "RECEIPT\n"
+        write "receipt-id:#{frame_info[:'receipt-id']}\n"
+        write "\0"
+      end
+      @socket.close
+    end
+
+    def send_frame
+      filename = sent_message_filename_for(frame)
+      if filename
+        dirname = File.dirname(filename)
+        FileUtils.mkdir_p(dirname) if !File.exists?(dirname)
+        File.open(filename, "w") { |file| file.write(frame) }
+      end
+    end
+
+    def write(data)
+      @socket.write data
+    end
+
+    private
+
+    def sent_message_filename_for(frame)
+      return nil unless @server.sent_message_dir
+      md5 = OpenSSL::Digest::MD5.new(frame)
+      "#{@server.sent_message_dir}/#{md5}.msg"
+    end
+
+    def parse(frame)
+      frame_info = {}
+      data_array = frame.split("\n").reject { |line| line =~ /^\s+$/ }
+      frame_info[:command] = data_array.shift
+      data_array.each do |info|
+        if info.include?(':')
+          frame_info[info.split(':')[0].to_sym] = info.split(':')[1]
+        end
+      end
+      frame_info
+    end
+
+    def append_data data
+      (@data ||= '') << data
+    end
+
+    def each_frame(&block)
+      while eoframe = @data.index("\0")
+        block.call(@data.slice!(0, eoframe + 1))
+      end
+    end
+
+  end
+
+end
+

data/lib/stomping_ground/server.rb

@@ -0,0 +1,39 @@
+module StompingGround
+  class Server
+    include Celluloid::IO
+
+    attr_accessor :message_body
+    attr_accessor :queue_name
+    attr_accessor :sent_message_dir
+
+    def initialize(host, port, options = {})
+      @host = host
+      @port = port
+
+      @message_body     = options[:message]
+      @queue_name       = options[:queue_name]
+      @sent_message_dir = options[:sent_message_dir]
+
+      @server = TCPServer.new(@host, @port)
+
+      async.run
+    end
+
+    def self.start(*opts)
+      supervise(*opts)
+    end
+
+    def run
+      loop { async.handle_connection @server.accept }
+    end
+
+    def handle_connection(socket)
+      Connection.new(self, socket).handle
+    end
+
+    def finalize
+      @server.close if @server
+    end
+
+  end
+end

data/lib/stomping_ground/version.rb

@@ -0,0 +1,5 @@
+module StompingGround
+  MAJOR=0
+  MINOR=1
+  VERSION="#{MAJOR}.#{MINOR}"
+end

metadata

@@ -0,0 +1,150 @@
+--- !ruby/object:Gem::Specification
+name: stomping_ground-celluloid
+version: !ruby/object:Gem::Version
+  version: '0.1'
+  prerelease: 
+platform: ruby
+authors:
+- Francisco Trindade
+- Geoffrey Giesemann
+- Emilien Taque
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-01-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: onstomp
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: celluloid-io
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- francisco.trindade@rea-group.com
+- geoffrey.giesemann@rea-group.com
+- e.taque@alphalink.fr
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/stomping_ground/connection.rb
+- lib/stomping_ground/server.rb
+- lib/stomping_ground/version.rb
+- lib/stomping_ground.rb
+homepage: http://github.com/etaque/stomping_ground
+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.24
+signing_key: 
+specification_version: 3
+summary: Mock Stomp server to be used when testing stomp consumers, Celluloid::IO
+  version.
+test_files: []