huck 0.2.5 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5d0b4c0bc632a390c11a3ee1b4546bdb8afd95f2
-  data.tar.gz: c1d5ca0f8d56c7b33dbab32ed7c962add05bf142
+  metadata.gz: ca3bf22a8ff5affb95d7bb88e7d40aa58e8077ab
+  data.tar.gz: ed30d1fedd87e7cc62b4942a7247323c57a77265
 SHA512:
-  metadata.gz: 948b55d8484a13477cc4af97677e5e94ad5dcaf1270108a3b6d23ba145f1db9545abc5cfd9200b7d0114c2163f1585471a4a15e7a7c5b43a3a0b485e6cf83527
-  data.tar.gz: 1bf55fb6c949b806e7687ed3e65c433a7fc98650d34f8c535a86d1fef3df0ed9b203d3f1d8dbf8ea90eccc322f430da3303863c121bd16c28ed4f3dc7ce341ac
+  metadata.gz: fc300e2459538853941313c9352fed1ba7b661530dce80da41c9eb18a36991b1581918e4f206934603f979b5819d170a70dce7095fa8b8bc4bb8f5ab565091b7
+  data.tar.gz: 80592af74c45e1eb246f20b1b99acfdedf363e8ef9adef532c3b918628523b2b4512805f9f0ca22c27950a2b50f9ec199e2ac06ac00f3867051b7d8233c80901

data/lib/huck/receiver.rb

@@ -19,17 +19,13 @@ module Huck
       name = Huck::getarg kwargs, :name, nil
       config = Huck::getarg kwargs, :config, nil
 
-      if name.nil?
-        if Huck::try_load 'aws-sdk'
-          name = 'sqs'
-        else
-          raise RuntimeError, 'unable to load any receivers'
-        end
-      end
+      name = 'sqs' if name.nil?
 
       case name
       when 'sqs'
         r = Receivers::SQSReceiver.new
+      when 'rabbitmq'
+        r = Receivers::RabbitMQReceiver.new
       else
         raise RuntimeError, "bad receiver: #{name}"
       end

data/lib/huck/receivers/rabbitmq.rb

@@ -0,0 +1,49 @@
+module Huck
+
+  module Receivers
+
+    # A receiver that talks to RabbitMQ
+    class RabbitMQReceiver < Receiver
+
+      # Includes all required modules for the RabbitMQ receiver
+      def initialize
+        Huck::must_load 'bunny'
+      end
+
+      # Ensures that configuration is set properly before trying to use the
+      # connection data to talk to RabbitMQ
+      def verify_config
+        if !@config.has_key? 'rabbitmq'
+          raise RuntimeError, 'missing rabbitmq config'
+        end
+        if !@config['rabbitmq'].has_key? 'queue_name'
+          raise RuntimeError, 'missing rabbitmq receiver config: queue_name'
+        end
+      end
+
+      # A long-running poller process which reads messages out of the remote
+      # queue and yields them to higher-order logic.
+      def receive
+        verify_config
+
+        options = Hash.new
+        [:host, :port, :user, :pass, :vhost].each do |arg|
+          if @config['rabbitmq'].has_key? arg.to_s
+            options[arg] = @config['rabbitmq'][arg.to_s]
+          end
+        end
+
+        conn = Bunny.new options
+        conn.start
+
+        ch = conn.create_channel
+        queue = ch.queue(@config['rabbitmq']['queue_name'])
+
+        queue.subscribe :block => true do |_, _, msg|
+          yield msg
+        end
+      end
+
+    end
+  end
+end

data/lib/huck/sender.rb

@@ -19,17 +19,13 @@ module Huck
       name = Huck::getarg kwargs, :name, nil
       config = Huck::getarg kwargs, :config, nil
 
-      if name.nil?
-        if Huck::try_load 'aws-sdk'
-          name = 'sqs'
-        else
-          raise RuntimeError, 'unable to load any senders'
-        end
-      end
+      name = 'sqs' if name.nil?
 
       case name
       when 'sqs'
         s = Senders::SQSSender.new
+      when 'rabbitmq'
+        s = Senders::RabbitMQSender.new
       else
         raise RuntimeError, "bad sender: #{name}"
       end

data/lib/huck/senders/rabbitmq.rb

@@ -0,0 +1,52 @@
+module Huck
+
+  module Senders
+
+    # A sender that talks to RabbitMQ
+    class RabbitMQSender < Sender
+
+      # Includes all required modules for the RabbitMQ sender
+      def initialize
+        Huck::must_load 'bunny'
+      end
+
+      # Ensures that configuration is set properly before trying to use the
+      # connection data to talk to rabbitmq
+      def verify_config
+        if !@config.has_key? 'rabbitmq'
+          raise RuntimeError, 'missing rabbitmq sender config'
+        end
+        if !@config['rabbitmq'].has_key? 'queue_name'
+          raise RuntimeError, 'missing rabbitmq sender config: queue_name'
+        end
+      end
+
+      # Send an arbitrary text message to the queue for processing
+      #
+      # == Parameters:
+      # msg::
+      #   The message to process
+      #
+      def send msg
+        verify_config
+
+        options = Hash.new
+        [:host, :port, :user, :pass, :vhost].each do |arg|
+          if @config['rabbitmq'].has_key? arg.to_s
+            options[arg] = @config['rabbitmq'][arg.to_s]
+          end
+        end
+
+        conn = Bunny.new options
+        conn.start
+
+        ch = conn.create_channel
+        queue = ch.queue(@config['rabbitmq']['queue_name'])
+        xfer = ch.default_exchange
+
+        xfer.publish msg, :routing_key => @config['rabbitmq']['queue_name']
+      end
+
+    end
+  end
+end

data/lib/huck/version.rb

@@ -1,3 +1,3 @@
 module Huck
-  Version = '0.2.5'
+  Version = '0.3.0'
 end

data/lib/huck.rb

@@ -11,8 +11,10 @@ require 'huck/generators/yaml'
 require 'huck/generators/json'
 require 'huck/sender'
 require 'huck/senders/sqs'
+require 'huck/senders/rabbitmq'
 require 'huck/receiver'
 require 'huck/receivers/sqs'
+require 'huck/receivers/rabbitmq'
 require 'huck/handler'
 require 'huck/handlers/echo'
 require 'huck/handlers/exec'

data/spec/huck/util_spec.rb

@@ -34,4 +34,19 @@ describe 'Module Loading' do
   it 'should raise a RuntimeError when module cannot be loaded' do
     expect{Huck::must_load 'nonexistent'}.to raise_error(RuntimeError)
   end
+
+end
+
+describe 'Data Serialization' do
+  input = {'a' => 'b'}
+
+  it 'should serialize to json' do
+    output = JSON.dump input
+    expect(Huck::serialize(input, :format => 'json')).to eq(output)
+  end
+
+  it 'should serialize to yaml' do
+    output = YAML.dump input
+    expect(Huck::serialize(input, :format => 'yaml')).to eq(output)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: huck
 version: !ruby/object:Gem::Version
-  version: 0.2.5
+  version: 0.3.0
 platform: ruby
 authors:
 - Ryan Uber
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-07-15 00:00:00.000000000 Z
+date: 2014-07-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -59,23 +59,25 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
-- bin/huck
-- lib/huck/generator.rb
-- lib/huck/generators/basic.rb
-- lib/huck/generators/facter.rb
-- lib/huck/generators/json.rb
 - lib/huck/generators/ohai.rb
+- lib/huck/generators/facter.rb
+- lib/huck/generators/basic.rb
 - lib/huck/generators/yaml.rb
-- lib/huck/handler.rb
-- lib/huck/handlers/echo.rb
-- lib/huck/handlers/exec.rb
+- lib/huck/generators/json.rb
+- lib/huck/senders/sqs.rb
+- lib/huck/senders/rabbitmq.rb
 - lib/huck/receiver.rb
-- lib/huck/receivers/sqs.rb
 - lib/huck/sender.rb
-- lib/huck/senders/sqs.rb
 - lib/huck/util.rb
+- lib/huck/generator.rb
+- lib/huck/receivers/sqs.rb
+- lib/huck/receivers/rabbitmq.rb
+- lib/huck/handlers/exec.rb
+- lib/huck/handlers/echo.rb
 - lib/huck/version.rb
+- lib/huck/handler.rb
 - lib/huck.rb
+- bin/huck
 - spec/huck/util_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/ryanuber/huck
@@ -98,7 +100,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.0.14
+rubygems_version: 2.1.11
 signing_key: 
 specification_version: 4
 summary: Information sharing framework