legion-transport 0.1.0 → 1.1.3

data/lib/legion/transport/connections/bunny.rb

@@ -1,53 +0,0 @@
-require 'legion/transport/connections/common'
-
-module Legion
-  module Transport
-    module Connections
-      module Bunny
-        @@session = nil # rubocop:disable Style/ClassVars
-        include Legion::Transport::Connections::Common
-        attr_accessor :session, :channel
-
-        def initialize(options = {})
-          options = options_builder(options)
-          create_session(options)
-          @channel = @session.create_channel
-          Legion::Settings[:transport][:rabbitmq][:connected] = true
-        end
-
-        def create_session(options = {})
-          Legion::Logging.debug 'Creating Legion::Transport session'
-          if options[:create_new_session]
-            @session = ::Bunny.new(options)
-            @session.start
-          elsif @@session.nil? || @@session.closed?
-            @@session = ::Bunny.new(options) # rubocop:disable Style/ClassVars
-            @@session.start
-            @session = @@session
-          else
-            Legion::Logging.debug "Using session:#{@@session.channel_id_allocator} from @@session"
-            @session = @@session
-          end
-        rescue NameError
-          Legion::Logging.debug 'Legion::Transport::Connection doesn\'t have a class variable session'
-          @session = ::Bunny.new(options)
-          @session.start
-        end
-
-        def valid_session?(session = @session)
-          false if session.nil? || session.empty?
-          false unless session.is_a? Bunny::Session
-          false if session.closed?
-          true
-        end
-
-        def valid_channel?(channel = @channel)
-          false if channel.nil? || session.empty?
-          false unless channel.is_a? Bunny::Session
-          false if channel.closed?
-          true
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/connections/common.rb

@@ -1,36 +0,0 @@
-module Legion
-  module Transport
-    module Connections
-      module Common
-        def options_builder(options = {})
-          settings = Legion::Settings[:transport][:rabbitmq] if Legion::Settings[:transport].is_a? Hash
-          hash = default_options.merge(settings) unless settings.nil?
-          hash = {} if settings.nil?
-          hash = hash.merge(options) if options.is_a? Hash
-          hash
-        end
-
-        def url_builder(options = default_options)
-          "amqp://#{options[:user]}:#{options[:password]}@#{options[:host]}:#{options[:port]}"
-        end
-
-        def default_options
-          default = {}
-          default[:read_timeout] = 30
-          default[:heartbeat] = 1
-          default[:automatically_recover] = true
-          default[:continuation_timeout] = 4000
-          default[:network_recovery_interval] = 5
-          default[:host] = '127.0.0.1'
-          default[:port] = '5672'
-          default[:user] = 'guest'
-          default[:password] = 'guest'
-          default[:vhost] = '/'
-          default[:log_level] = :info
-
-          default
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/connections/marchhare.rb

@@ -1,20 +0,0 @@
-require 'legion/transport/connections/common'
-
-module Legion
-  module Transport
-    module Connections
-      module Marchhare
-        include Legion::Transport::Connections::Common
-        attr_accessor :session, :channel
-
-        def initialize(options = {})
-          options = options_builder(options)
-          @session = MarchHare.connect(options)
-          @session.start
-          @channel = @session.create_channel
-          Legion::Settings[:transport][:rabbitmq][:connected] = true
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/consumers/bunny.rb

@@ -1,11 +0,0 @@
-module Legion
-  module Transport
-    module Consumer
-      class Bunny < ::Bunny::Consumer
-        def initialize(queue, no_ack = false, exclusive = false, arguments = {})
-          super(Legion::Transport::Connections.new.channel, queue, '', no_ack, exclusive, arguments)
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/consumers/common.rb

@@ -1,8 +0,0 @@
-module Legion
-  module Transport
-    module Consumer
-      module Common
-      end
-    end
-  end
-end

data/lib/legion/transport/consumers/marchhare.rb

@@ -1,11 +0,0 @@
-module Legion
-  module Transport
-    module Consumer
-      class Marchhare < ::MarchHare::Consumer
-        def initialize(queue, no_ack = false, exclusive = false, arguments = {})
-          super(Legion::Transport::Connection::Marchhare.new.channel, queue, '', no_ack, exclusive, arguments)
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/exchanges/bunny.rb

@@ -1,28 +0,0 @@
-require 'legion/transport/exchanges/common.rb'
-
-module Legion
-  module Transport
-    module Exchanges
-      class Bunny < ::Bunny::Exchange
-        include Legion::Transport::Exchanges::Common
-        def initialize(exchange = exchange_name, options = {})
-          retries ||= 0
-          @options = options
-          @type = options[:type] || default_type
-
-          super(channel, @type, exchange, options_builder(default_options, exchange_options, @options))
-        rescue ::Bunny::PreconditionFailed
-          retries.zero? ? retries = 1 : raise
-          recreate_exchange(channel, @type, exchange)
-          retry
-        end
-
-        def recreate_exchange(channel, type, exchange)
-          Legion::Logging.warn "Exchange:#{exchange} exists with wrong parameters, deleting and creating"
-          exchange = ::Bunny::Exchange.new(channel, type, exchange, no_declare: true)
-          exchange.delete
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/exchanges/common.rb

@@ -1,32 +0,0 @@
-module Legion
-  module Transport
-    module Exchanges
-      module Common
-        include Legion::Transport::Common
-
-        def default_options
-          hash = {}
-          hash[:durable] = true
-          hash[:auto_delete] = false
-          hash[:arguments] = {}
-          hash
-        end
-
-        def exchange_options
-          {}
-        end
-
-        def delete(options = {})
-          super(options)
-          true
-        rescue ::Bunny::PreconditionFailed
-          false
-        end
-
-        def default_type
-          'topic'
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/exchanges/marchhare.rb

@@ -1,17 +0,0 @@
-module Legion
-  module Transport
-    module Exchanges
-      class Marchhare < ::MarchHare::Exchange
-        attr_accessor :channel
-        def self.create(exchange, _type = :direct, _opts = {})
-          @channel = Legion::Transport::Connections::Marchhare.new
-          new(@channel.channel, exchange, type: :direct, durable: true)
-        end
-
-        def self.close
-          @channel.close
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/messages/base.rb

@@ -1,22 +0,0 @@
-module Legion
-  module Transport
-    module Messages
-      module Base
-        include Legion::Transport::Common
-        def publish(_options = {}) # rubocop:disable Metrics/AbcSize
-          validate if @valid.nil?
-          raise unless @valid
-
-          thing = exchange.new
-          thing.publish(message) if routing_key.nil?
-          thing.publish(message, routing_key: routing_key) unless routing_key.nil?
-          thing.close
-        end
-
-        def validate
-          @valid = true
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/messages/node_status.rb

@@ -1,49 +0,0 @@
-require 'legion/transport/exchanges/task_update'
-
-module Legion
-  module Exception
-    class InvalidTaskStatus; end
-    class InvalidTaskId; end
-  end
-end
-
-module Legion
-  module Transport
-    module Messages
-      # Used for sending a task update, to be moved
-      class TaskUpdate < Legion::Transport::Message
-        attr_accessor :status
-        attr_reader :task_id, :valid
-        def initialize(task_id, status = 'unknown', options = {})
-          @status = status
-          @task_id = task_id
-          @routing_key = options[:routing_key] unless options[:routing_key].nil?
-          @valid = validate
-        end
-
-        def exchange
-          Legion::Transport::Exchanges::Task
-        end
-
-        def message(status = @status, task_id = @task_id)
-          Legion::JSON.dump(status: status, task_id: task_id)
-        end
-
-        def validate(status = @status, task_id = @task_id)
-          raise Legion::Exception::InvalidTaskId unless task_id.is_a? Integer
-          raise Legion::Exception::InvalidTaskStatus unless valid_status.include? status
-
-          @valid = true
-        end
-
-        def valid_status
-          conditioner = ['conditioner.queued', 'conditioner.failed', 'conditioner.exception']
-          transfomer = ['transformer.queued', 'transformer.succeeded', 'transformer.exception']
-          task = ['task.scheduled', 'task.queued', 'task.completed']
-          status = conditioner + transfomer + task
-          status
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/messages/task_check_subtask.rb

@@ -1,35 +0,0 @@
-require 'legion/transport/exchanges/task'
-
-module Legion
-  module Transport
-    module Messages
-      class TaskCheckSubtask < Legion::Transport::Message
-        attr_accessor :status
-        attr_reader :task_id, :valid
-        def initialize(namespace, method, result, _options = {})
-          @namespace = namespace
-          @method = method
-          @result = result
-          @routing_key = 'task.subtask.check'
-          validate
-        end
-
-        def exchange
-          Legion::Transport::Exchanges::Task
-        end
-
-        def message(namespace = @namespace, method = @method, result = @result)
-          Legion::JSON.dump(args: { namespace: namespace, method: method, result: result })
-        end
-
-        def routing_key
-          'task.subtask.check'
-        end
-
-        def validate(_namespace = @namespace, _method = @method, _result = @result, _options = {})
-          @valid = true
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/messages/task_subtask.rb

@@ -1,45 +0,0 @@
-require 'legion/transport/exchanges/task'
-
-module Legion
-  module Transport
-    module Messages
-      class TaskSubTask < Legion::Transport::Message
-        attr_accessor :status
-        attr_reader :task_id, :valid
-        def initialize(relationship_id, trigger_id, conditions, transformations, task_id = nil, options = {}) # rubocop:disable Metrics/ParameterLists,Metrics/LineLength
-          @relationship_id = relationship_id
-          @trigger_id = trigger_id
-          @conditions = conditions
-          @tranformations = transformations
-          @task_id = task_id
-          @options = options
-          @routing_key = 'task.subtask'
-          validate
-        end
-
-        def exchange
-          Legion::Transport::Exchanges::Task
-        end
-
-        def message(relationship_id = @relationship_id, trigger_id = @trigger_id, conditions = @conditions, transformations = @tranformations, task_id = @task_id, options = @options) # rubocop:disable Metrics/ParameterLists,Metrics/LineLength
-          Legion::JSON.dump(args: {
-                              relationship_id: relationship_id,
-                              trigger_id:      trigger_id,
-                              conditions:      conditions,
-                              task_id:         task_id,
-                              transformations: transformations,
-                              options:         options
-                            })
-        end
-
-        def routing_key
-          'task.subtask'
-        end
-
-        def validate(_namespace = @namespace, _method = @method, _result = @result, _options = {})
-          @valid = true
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/queues/bunny.rb

@@ -1,26 +0,0 @@
-require 'legion/transport/queues/common'
-
-module Legion
-  module Transport
-    module Queues
-      class Bunny < ::Bunny::Queue
-        include Legion::Transport::Queues::Common
-        def initialize(queue = queue_name, options = {})
-          retries ||= 0
-          @options = options
-          super(channel, queue, options_builder(default_options, queue_options, @options))
-        rescue ::Bunny::PreconditionFailed
-          retries.zero? ? retries = 1 : raise
-          recreate_queue(channel, queue)
-          retry
-        end
-
-        def recreate_queue(channel, queue)
-          Legion::Logging.warn "Queue:#{queue} exists with wrong parameters, deleting and creating"
-          queue = ::Bunny::Queue.new(channel, queue, no_declare: true, passive: true)
-          queue.delete(if_empty: true)
-        end
-      end
-    end
-  end
-end

data/lib/legion/transport/queues/common.rb

@@ -1,45 +0,0 @@
-module Legion
-  module Transport
-    module Queues
-      module Common
-        include Legion::Transport::Common
-        def default_options
-          hash = {}
-          hash[:manual_ack] = true
-          hash[:durable] = true
-          hash[:exclusive] = false
-          hash[:block] = false
-          hash[:arguments] = { 'x-max-priority': 255, 'x-overflow': 'reject-publish' }
-          hash
-        end
-
-        def queue_options
-          {}
-        end
-
-        def delete(options = { if_unused: true, if_empty: true })
-          super(options)
-          true
-        rescue ::Bunny::PreconditionFailed
-          false
-        end
-
-        def channel(options = @options || {})
-          @channel if channel_valid?
-          @channel = options[:channel] if channel_valid?(options[:channel])
-          @channel = open_channel(options) unless channel_valid?(@channel)
-          @channel
-        end
-
-        #  @queue.channel.acknowledge(delivery_info.delivery_tag)
-        def acknowledge(delivery_tag)
-          channel.acknowledge(delivery_tag)
-        end
-
-        def reject(delivery_tag, requeue = false)
-          channel.reject(delivery_tag, requeue)
-        end
-      end
-    end
-  end
-end