legion-transport 0.1.0

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

@@ -0,0 +1,20 @@
+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/consumer.rb

@@ -0,0 +1,6 @@
+module Legion
+  module Transport
+    class Connection
+    end
+  end
+end

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

@@ -0,0 +1,11 @@
+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

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

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

@@ -0,0 +1,11 @@
+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/exchange.rb

@@ -0,0 +1,13 @@
+module Legion
+  module Transport
+    if RUBY_ENGINE == 'jruby'
+      require 'legion/transport/exchanges/marchhare'
+      class Exchange < Legion::Transport::Exchanges::Marchhare
+      end
+    else
+      require 'legion/transport/exchanges/bunny'
+      class Exchange < Legion::Transport::Exchanges::Bunny
+      end
+    end
+  end
+end

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

@@ -0,0 +1,28 @@
+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

@@ -0,0 +1,32 @@
+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/lex.rb

@@ -0,0 +1,11 @@
+module Legion
+  module Transport
+    module Exchanges
+      class Lex < Legion::Transport::Exchange
+        def exchange_name
+          'lex'
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,17 @@
+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/exchanges/task.rb

@@ -0,0 +1,11 @@
+module Legion
+  module Transport
+    module Exchanges
+      class Task < Legion::Transport::Exchange
+        def exchange_name
+          'task'
+        end
+      end
+    end
+  end
+end

data/lib/legion/transport/message.rb

@@ -0,0 +1,8 @@
+module Legion
+  module Transport
+    class Message
+      require 'legion/transport/messages/base'
+      include Legion::Transport::Messages::Base
+    end
+  end
+end

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

@@ -0,0 +1,22 @@
+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/lex_register.rb

@@ -0,0 +1,39 @@
+require 'legion/transport/exchanges/lex'
+
+module Legion
+  module Transport
+    module Messages
+      class LexRegister < Legion::Transport::Message
+        def initialize(namespace, method, options = {})
+          @namespace = namespace
+          @method = method
+          @options = options
+        end
+
+        def exchange
+          Legion::Transport::Exchanges::Lex
+        end
+
+        def routing_key
+          'lex.methods.register'
+        end
+
+        def message(namespace = @namespace, method = @method, options = @options)
+          obj = { args: { namespace: namespace, method: method } }
+          obj[:args][:namespace_queue] = options[:namespace][:queue] unless options[:namespace][:queue].nil?
+          obj[:args][:namespace_uri] = options[:namespace][:uri] unless options[:namespace][:uri].nil?
+          Legion::JSON.dump(obj)
+        end
+
+        def validate(namespace = @namespace, _method = @method, _options = @options)
+          raise unless namespace.is_a? String
+
+          # raise unless method.is_a? String
+          # raise unless options.is_a? Hash
+
+          @valid = true
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,49 @@
+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

@@ -0,0 +1,35 @@
+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

@@ -0,0 +1,45 @@
+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/messages/task_update.rb

@@ -0,0 +1,52 @@
+require 'legion/transport/exchanges/task'
+
+module Legion
+  module Exception
+    class InvalidTaskStatus; end
+    class InvalidTaskId; end
+  end
+end
+
+module Legion
+  module Transport
+    module Messages
+      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 = 'task.update'
+          validate
+        end
+
+        def routing_key
+          'task.update'
+        end
+
+        def exchange
+          Legion::Transport::Exchanges::Task
+        end
+
+        def message(status = @status, task_id = @task_id, options = {})
+          Legion::JSON.dump(args: { status: status, task_id: task_id, options: options })
+        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', 'task.exception']
+          status = conditioner + transfomer + task
+          status
+        end
+      end
+    end
+  end
+end