bunny_burrow 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4dfcb9dc41f611dddb4036a1498092df18673cea
-  data.tar.gz: b4f87e37e1589109a31d6de589b24d7879c5ca54
+  metadata.gz: f9c676f62c43bcb59cd98e4df385c514d32753ff
+  data.tar.gz: e744b1a510e83e6ece4753af0b1fd7a4d01abef0
 SHA512:
-  metadata.gz: 57630b4667e989aa11bea9596ea359c49e9fe1d137699f720df706046f6ceb6a3c9e22260bbd0e4dd26483fb2381cd3aa7369483c8e463298db2660410cb01e8
-  data.tar.gz: 44c128dc362cd389cfe6f70e5828839f3db2a53a7ff4c916c9a7302fc6c11b9e52ae3c3a7d6bb1f6f70eabf8390341ab2fc8dc05b81db941c4d066aa4cf894a2
+  metadata.gz: 838c97dbe6236e5e55fbbd5a5a72bda3ff27d1ab20ce2e9d54d71ad147dbf3d744ec30b77e549df57f8712f23959d1f3a9d068d9916591993d027be8360f7c86
+  data.tar.gz: 8d2af0701684e95d516f8d1a1468e3f9d33255d298de1546133388b242ea22d7157c6003e8c5db0d6f076c764860fdb82ca7d982de37d82279d03328e1284747

data/lib/bunny_burrow/base.rb

@@ -0,0 +1,75 @@
+require 'bunny'
+require 'json'
+require 'thread'
+
+module BunnyBurrow
+  STATUS_OK = 'ok'
+  STATUS_CLIENT_ERROR = 'client_error'
+  STATUS_SERVER_ERROR = 'server_error'
+
+  class Base
+    attr_accessor :rabbitmq_url, :rabbitmq_exchange, :logger, :log_prefix
+    attr_writer :timeout, :log_request, :log_response
+
+    def initialize
+      yield self if block_given?
+    end
+
+    def timeout
+      @timeout ||= 60
+    end
+
+    def log_request?
+      @log_request ||= false
+    end
+
+    def log_response?
+      @log_response ||= false
+    end
+
+    def shutdown
+      log 'Shutting down'
+      channel.close
+      connection.close
+    end
+
+    private
+
+    def connection
+      unless @connection
+        @connection = Bunny.new(rabbitmq_url)
+        @connection.start
+      end
+
+      @connection
+    end
+
+    def channel
+      @channel ||= connection.create_channel
+    end
+
+    def default_exchange
+      @default_exchange ||= channel.default_exchange
+    end
+
+    def topic_exchange
+      @topic_exchange ||= channel.topic(rabbitmq_exchange, durable: true)
+    end
+
+    def lock
+      @lock ||= Mutex.new
+    end
+
+    def condition
+      @condition ||= ConditionVariable.new
+    end
+
+    def log(message, level: :info)
+      return unless logger
+      prefix = log_prefix || 'BunnyBurrow'
+      message = "#{prefix}: #{message}"
+      logger.send(level, message)
+    end
+  end
+end
+

data/lib/bunny_burrow/client.rb

@@ -0,0 +1,46 @@
+require_relative 'base'
+
+module BunnyBurrow
+  class Client < Base
+    def publish(payload, routing_key)
+      result = nil
+
+      details = {
+        routing_key: routing_key,
+        reply_to: reply_to
+      }
+      details[:request] = payload if log_request?
+      log "Publishing #{details}"
+
+      options = {
+        routing_key: routing_key,
+        reply_to: reply_to.name,
+        persistence: false
+      }
+
+      topic_exchange.publish(payload.to_json, options)
+
+      Timeout.timeout(timeout) do
+        reply_to.subscribe do |_, _, payload|
+          details[:response] = payload if log_response?
+          log "Receiving #{details}"
+          result = payload
+          lock.synchronize { condition.signal }
+        end
+
+        lock.synchronize { condition.wait(lock) }
+      end
+
+      result
+    end
+
+    private
+
+    def reply_to
+      # when creating a queue, a blank name indicates we want the AMPQ broker
+      # to generate a unique name for us. Also note that this queue will be on
+      # the default exchange
+      @reply_to ||= channel.queue('', exclusive: true, auto_delete: true)
+    end
+  end
+end

data/lib/bunny_burrow/server.rb

@@ -0,0 +1,79 @@
+require_relative 'base'
+
+module BunnyBurrow
+  class Server < Base
+    def self.create_response
+      {
+        status: STATUS_OK,
+        error_message: nil,
+        data: {}
+      }
+    end
+
+    def subscribe(routing_key, &block)
+      queue = channel.queue('', exclusive: true, auto_delete: true)
+      queue.bind(topic_exchange, routing_key: routing_key)
+
+      details = {
+        routing_key: routing_key,
+        queue: queue.name,
+        exchange: topic_exchange.name
+      }
+
+      log "Subscribing #{details}"
+      queue.subscribe(manual_ack: true) do |delivery_info, properties, payload|
+        begin
+          details = {
+            delivery_info: delivery_info,
+            properties: properties
+          }
+
+          details[:request] = payload if log_request?
+          log "Receiving #{details}"
+
+          response = block.call(payload)
+
+          details[:response] = response if log_response?
+
+          log "Replying #{details}"
+          default_exchange.publish(response.to_json, :routing_key => properties.reply_to, persistence: false)
+
+          log "Acknowledging #{details}"
+          channel.ack delivery_info.delivery_tag
+        rescue => e
+          log e.message, level: :error
+          response = {
+            status: STATUS_SERVER_ERROR,
+            error_message: e.message
+          }
+          default_exchange.publish(response.to_json, :routing_key => properties.reply_to, persistence: false)
+        end
+      end
+    rescue => e
+      log e.message, level: :error
+    end
+
+    def wait
+      process_lock.synchronize { process_condition.wait(process_lock) }
+    end
+
+    def stop_waiting
+      process_lock.synchronize { process_condition.signal }
+    end
+
+    def shutdown
+      stop_waiting
+      super
+    end
+
+    private
+
+    def process_lock
+      @process_lock ||= Mutex.new
+    end
+
+    def process_condition
+      @process_condition ||= ConditionVariable.new
+    end
+  end
+end

data/lib/bunny_burrow/version.rb

@@ -0,0 +1,3 @@
+module BunnyBurrow
+  VERSION = '1.1.0'
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bunny_burrow
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Vericity
@@ -129,6 +129,10 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/bunny_burrow.rb
+- lib/bunny_burrow/base.rb
+- lib/bunny_burrow/client.rb
+- lib/bunny_burrow/server.rb
+- lib/bunny_burrow/version.rb
 homepage: https://github.com/johann-koebbe/bunny_burrow
 licenses:
 - MIT