spectre-rabbitmq 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: ff5d268247815c24c9ef18aa240d48880d939103f18219daedd70fe27babc866
+  data.tar.gz: 2c41a4cf22e0091c7d15e094c6bd226fa802f879e826fbf0ad675fa806cc0a92
+SHA512:
+  metadata.gz: e41362194b709e48652fc34de990258b9019f81b2b69e05c21559fec0998be5c92d6649dd0cfaf1e9f96c8a809f35e940cf1c90b7adb99914a597bc4bbe29c7f
+  data.tar.gz: c85e00ecf1670ba4dd90079083ba8c4ec3763873a688e4f2b3ad204a577ca76bc0810fcdf6b8f7096d2cce647728498447f605d35ebd4834395d9a97db7fa0ec

data/lib/spectre/rabbitmq.rb

@@ -0,0 +1,272 @@
+require 'spectre'
+require 'spectre/logger'
+require 'ostruct'
+require 'bunny'
+
+
+module Spectre
+  module RabbitMQ
+    class ActionParamsBase
+      attr_reader :config
+
+      def initialize config, logger
+        @logger = logger
+        @config = config.deep_clone
+        @config['routing_keys'] = []
+      end
+
+      def exchange name, type: 'topic', durable: false, auto_delete: false
+        @config['exchange'] = {
+          'name' => name,
+          'durable' => durable,
+          'type' => type,
+          'auto_delete' => auto_delete,
+        }
+      end
+
+      def topic name, durable: false, auto_delete: false
+        exchange(name, type: 'topic', durable: durable, auto_delete: auto_delete)
+      end
+
+      def routing_keys *names
+        @config['routing_keys'] = names
+      end
+
+      def routing_key name
+        @config['routing_keys'] << name
+      end
+    end
+
+    class ConsumeActionParams < ActionParamsBase
+      def initialize config, logger
+        super config, logger
+
+        @config['queue'] = {
+          'name' => nil,
+          'durable' => false,
+          'auto_delete' => false,
+        }
+
+        @config['messages'] = 1
+      end
+
+      def queue name, durable: false, auto_delete: false, exclusive: false
+        @config['queue'] = {
+          'name' => name,
+          'durable' => durable,
+          'auto_delete' => auto_delete,
+          'exclusive' => exclusive,
+        }
+      end
+
+      def timeout seconds
+        @config['timeout'] = seconds
+      end
+
+      def messages count
+        @config['messages'] = count
+      end
+    end
+
+    class PublishActionParams < ActionParamsBase
+      def payload data
+        @config['payload'] = data
+      end
+
+      def correlation_id id
+        @config['correlation_id'] = id
+      end
+
+      def reply_to receiver
+        @config['reply_to'] = receiver
+      end
+
+      alias :body :payload
+    end
+
+    class RabbitMQAction < Spectre::DslClass
+      attr_reader :conn, :action, :threads, :messages
+
+      def initialize config, logger
+        @logger = logger
+        @config = config
+        @conn = nil
+
+        @threads = []
+        @messages = []
+
+        @config['ssl'] = false
+      end
+
+      def action name
+        raise "invalid action '#{name}'" unless ['publish', 'subscribe'].include? name
+
+        @config['action'] = name
+      end
+
+      def host hostname
+        @config['host'] = hostname
+      end
+
+      def use_ssl!
+        @config['ssl'] = true
+      end
+
+      def username user
+        @config['username'] = user
+      end
+
+      def password pass
+        @config['password'] = pass
+      end
+
+      def virtual_host vhost
+        @config['virtual_host'] = vhost
+      end
+
+      def consume &block
+        params = ConsumeActionParams.new(@config, @logger)
+        params.instance_eval(&block)
+
+        connect()
+
+        channel = @conn.create_channel
+
+        exchange = declare_exchange(channel, params)
+
+        queue = channel.queue(
+          params.config['queue']['name'],
+          durable: params.config['queue']['durable'],
+          auto_delete: params.config['queue']['auto_delete'],
+          exclusive: params.config['queue']['exclusive'],
+        )
+
+        @logger.info("declare queue name=#{queue.name} durable=#{params.config['queue']['durable']} auto_delete=#{params.config['queue']['auto_delete']} exclusive=#{params.config['queue']['exclusive']}")
+
+        params.config['routing_keys'].each do |routing_key|
+          queue.bind(exchange, routing_key: routing_key)
+          @logger.info("bind exchange=#{exchange.name} queue=#{queue.name} routing_key=#{routing_key}")
+        end
+
+        consumer_thread = Thread.new do
+          message_queue = Queue.new
+
+          queue.subscribe do |_delivery_info, properties, payload|
+            message = OpenStruct.new
+            message.payload = payload
+            message.correlation_id = properties[:correlation_id]
+            message.reply_to = properties[:reply_to]
+            message.freeze
+
+            message_queue << message
+          end
+
+          while @messages.count < params.config['messages']
+            message = message_queue.pop
+            @logger.info("get queue=#{queue.name}\ncorrelation_id: #{message.correlation_id}\nreply_to: #{message.reply_to}\n#{message.payload}")
+            @messages << message
+          end
+        end
+
+        @threads << consumer_thread
+
+        Thread.new do
+          sleep(params.config['timeout'] || 10)
+          consumer_thread.exit
+        end
+      end
+
+      def publish &block
+        params = PublishActionParams.new(@config, @logger)
+        params.instance_eval(&block)
+
+        connect()
+
+        channel = @conn.create_channel
+
+        exchange = declare_exchange(channel, params)
+
+        routing_key = params.config['routing_keys'].nil? ? nil : params.config['routing_keys'].first
+
+        exchange.publish(
+          params.config['payload'],
+          routing_key: routing_key,
+          correlation_id: params.config['correlation_id'],
+          reply_to: params.config['reply_to']
+        )
+
+        @logger.info("publish exchange=#{params.config['exchange']['name']} routing_key=#{routing_key} payload=\"#{params.config['payload']}\"")
+      end
+
+      def await!
+        @threads.each { |x| x.join }
+      end
+
+      private
+
+      def connect
+        return unless @conn.nil?
+
+        @logger.info("connect #{@config['username']}:*****@#{@config['host']}#{@config['virtual_host']} ssl=#{@config['ssl']}")
+
+        @conn = Bunny.new(
+          host: @config['host'],
+          ssl: @config['ssl'],
+          username: @config['username'],
+          password: @config['password'],
+          virtual_host: @config['virtual_host']
+        )
+
+        @conn.start
+      end
+
+      def declare_exchange(channel, params)
+        exchange = Bunny::Exchange.new(
+          channel,
+          params.config['exchange']['type'].to_s,
+          params.config['exchange']['name'],
+          durable: params.config['exchange']['durable'],
+          auto_delete: params.config['exchange']['auto_delete'],
+        )
+
+        @logger.info("declare exchange name=#{exchange.name} type=#{exchange.type} durable=#{params.config['exchange']['durable']} auto_delete=#{params.config['exchange']['auto_delete']}")
+
+        exchange
+      end
+    end
+
+    class << self
+      @@config = {}
+
+      def rabbitmq name, &block
+        if @@config.key? name
+          config = @@config[name]
+        else
+          config = {
+            'host' => name,
+          }
+        end
+
+        action = RabbitMQAction.new(config, @@logger)
+        action._evaluate(&block) if block_given?
+
+        # Wait for all consumer threads to be finished
+        action.threads.each { |x| x.join }
+
+        action.conn.close
+      end
+    end
+
+    Spectre.register do |config|
+      @@logger = Spectre::Logging::ModuleLogger.new(config, 'spectre/rabbitmq')
+
+      if config.key? 'rabbitmq'
+        config['rabbitmq'].each do |name, cfg|
+          @@config[name] = cfg
+        end
+      end
+    end
+
+    Spectre.delegate :rabbitmq, to: self
+  end
+end

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification
+name: spectre-rabbitmq
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Christian Neubauer
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2023-04-20 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: spectre-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.14.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.14.0
+- !ruby/object:Gem::Dependency
+  name: bunny
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.20.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.20.3
+description: Adds RabbitMQ functionality to the spectre framework
+email:
+- christian.neubauer@ionos.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/spectre/rabbitmq.rb
+homepage: https://github.com/ionos-spectre/spectre-rabbitmq
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/ionos-spectre/spectre-rabbitmq
+  source_code_uri: https://github.com/ionos-spectre/spectre-rabbitmq
+  changelog_uri: https://github.com/ionos-spectre/spectre-rabbitmq/blob/master/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.0.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.3.7
+signing_key:
+specification_version: 4
+summary: RabbitMQ module for spectre
+test_files: []