lex-node 0.1.7 → 0.2.3

data/lib/legion/extensions/node/transport/exchanges/node.rb

@@ -1,9 +1,18 @@
+# frozen_string_literal: true
+
 require 'legion/transport/exchanges/node'
 
-module Legion::Extensions::Node
-  module Transport
-    module Exchanges
-      class Node < Legion::Transport::Exchanges::Node
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Exchanges
+          class Node < Legion::Transport::Exchanges::Node
+            def exchange_name
+              'node'
+            end
+          end
+        end
       end
     end
   end

data/lib/legion/extensions/node/transport/messages/beat.rb

@@ -1,31 +1,94 @@
-module Legion::Extensions::Node::Transport::Messages
-  class Beat < Legion::Transport::Message
-    def routing_key
-      'health'
-    end
+# frozen_string_literal: true
 
-    def type
-      'heartbeat'
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class Beat < Legion::Transport::Message
+            def routing_key
+              'status'
+            end
 
-    def expiration
-      5000
-    end
+            def type
+              'heartbeat'
+            end
 
-    def encrypt?
-      false
-    end
+            def expiration
+              5000
+            end
 
-    def message
-      hash = { hostname: Legion::Settings[:client][:hostname], pid: Process.pid, timestamp: Time.now }
-      hash[:status] = @options[:status].nil? ? 'healthy' : @options[:status]
-      hash
-    end
+            def encrypt?
+              false
+            end
+
+            def message
+              hash = {
+                name:      Legion::Settings[:client][:hostname],
+                pid:       ::Process.pid,
+                timestamp: Time.now,
+                status:    @options[:status].nil? ? 'healthy' : @options[:status]
+              }
+              hash[:version] = Legion::VERSION if defined?(Legion::VERSION)
+              hash[:metrics] = collect_metrics
+              hash[:hosted_worker_ids] = collect_worker_ids
+              hash
+            end
+
+            def validate
+              raise 'status should be a string' unless @options[:status].is_a?(String) || @options[:status].nil?
+
+              @valid = true
+            end
+
+            private
+
+            def collect_metrics
+              times = ::Process.times
+              {
+                memory_rss_mb:      rss_mb,
+                cpu_user_seconds:   times.utime.round(2),
+                cpu_system_seconds: times.stime.round(2),
+                thread_count:       Thread.list.count,
+                loaded_extensions:  loaded_extension_count,
+                uptime_seconds:     uptime_seconds
+              }
+            end
+
+            def rss_mb
+              if RUBY_PLATFORM.include?('darwin')
+                `ps -o rss= -p #{::Process.pid}`.strip.to_i / 1024.0
+              else
+                File.read("/proc/#{::Process.pid}/statm").split[1].to_i * (4096.0 / 1_048_576)
+              end
+            rescue StandardError
+              0.0
+            end
+
+            def loaded_extension_count
+              return 0 unless defined?(Legion::Extensions)
+
+              Legion::Extensions.respond_to?(:loaded_extensions) ? Legion::Extensions.loaded_extensions.count : 0
+            end
+
+            def uptime_seconds
+              (::Process.clock_gettime(::Process::CLOCK_MONOTONIC) - boot_time).round(0)
+            end
+
+            def boot_time
+              @boot_time ||= ::Process.clock_gettime(::Process::CLOCK_MONOTONIC)
+            end
 
-    def validate
-      raise 'status should be a string' unless @options[:status].is_a?(String) || @options[:status].nil?
+            def collect_worker_ids
+              return [] unless defined?(Legion::DigitalWorker)
 
-      @valid = true
+              Legion::DigitalWorker.active_local_ids
+            rescue StandardError
+              []
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/public_key.rb

@@ -1,21 +1,31 @@
-module Legion::Extensions::Node::Transport::Messages
-  class PublicKey < Legion::Transport::Message
-    def routing_key
-      'node.crypt.update_public_key'
-    end
+# frozen_string_literal: true
 
-    def type
-      'task'
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class PublicKey < Legion::Transport::Message
+            def routing_key
+              'node.crypt.update_public_key'
+            end
 
-    def encrypt?
-      false
-    end
+            def type
+              'task'
+            end
+
+            def encrypt?
+              false
+            end
 
-    def validate
-      raise 'public_key should be a string' unless @options[:public_key].is_a?(String)
+            def validate
+              raise 'public_key should be a string' unless @options[:public_key].is_a?(String)
 
-      @valid = true
+              @valid = true
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/push_cluster_secret.rb

@@ -1,32 +1,42 @@
-module Legion::Extensions::Node::Transport::Messages
-  class PushClusterSecret < Legion::Transport::Message
-    def routing_key
-      @options[:queue_name]
-    end
+# frozen_string_literal: true
 
-    def exchange
-      Legion::Transport::Exchanges::Node
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class PushClusterSecret < Legion::Transport::Message
+            def routing_key
+              @options[:queue_name]
+            end
 
-    def message
-      { function: 'receive_cluster_secret',
-        runner_class: 'Legion::Extensions::Node::Runners::Crypt',
-        message: @options[:message],
-        validation_string: @options[:validation_string] || nil,
-        encrypted_string: @options[:encrypted_string] || nil,
-        public_key: Base64.encode64(Legion::Crypt.public_key) }
-    end
+            def exchange
+              Legion::Transport::Exchanges::Node
+            end
 
-    def type
-      'task'
-    end
+            def message
+              { function:          'receive_cluster_secret',
+                runner_class:      'Legion::Extensions::Node::Runners::Crypt',
+                message:           @options[:message],
+                validation_string: @options[:validation_string] || nil,
+                encrypted_string:  @options[:encrypted_string] || nil,
+                public_key:        Base64.encode64(Legion::Crypt.public_key) }
+            end
 
-    def encrypt?
-      false
-    end
+            def type
+              'task'
+            end
+
+            def encrypt?
+              false
+            end
 
-    def validate
-      @valid = true
+            def validate
+              @valid = true
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/push_vault_token.rb

@@ -1,35 +1,42 @@
-module Legion::Extensions::Node::Transport::Messages
-  class PushVaultToken < Legion::Transport::Message
-    def routing_key
-      "node.#{@options[:queue_name]}"
-    end
+# frozen_string_literal: true
 
-    def exchange
-      Legion::Transport::Exchanges::Node
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class PushVaultToken < Legion::Transport::Message
+            def routing_key
+              "node.#{@options[:queue_name]}"
+            end
 
-    def message
-      data =
-        { function: 'receive_vault_token',
-          runner_class: 'Legion::Extensions::Node::Runners::Vault',
-          message: @options[:token],
-          public_key: Base64.encode64(Legion::Crypt.public_key) }
+            def exchange
+              Legion::Transport::Exchanges::Node
+            end
 
-      Legion::Logging.unknown data
-      Legion::Logging.unknown routing_key
-      data
-    end
+            def message
+              {
+                function:     'receive_vault_token',
+                runner_class: 'Legion::Extensions::Node::Runners::Vault',
+                message:      @options[:token],
+                public_key:   Base64.encode64(Legion::Crypt.public_key)
+              }
+            end
 
-    def type
-      'task'
-    end
+            def type
+              'task'
+            end
 
-    def encrypt?
-      false
-    end
+            def encrypt?
+              false
+            end
 
-    def validate
-      @valid = true
+            def validate
+              @valid = true
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/request_cluster_secret.rb

@@ -1,23 +1,33 @@
-module Legion::Extensions::Node::Transport::Messages
-  class RequestClusterSecret < Legion::Transport::Message
-    def routing_key
-      'node.crypt.push_cluster_secret'
-    end
+# frozen_string_literal: true
 
-    def message
-      { function: 'push_cluster_secret', node_name: Legion::Settings[:client][:name] }
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class RequestClusterSecret < Legion::Transport::Message
+            def routing_key
+              'node.crypt.push_cluster_secret'
+            end
 
-    def type
-      'task'
-    end
+            def message
+              { function: 'push_cluster_secret', node_name: Legion::Settings[:client][:name] }
+            end
 
-    def encrypt?
-      false
-    end
+            def type
+              'task'
+            end
+
+            def encrypt?
+              false
+            end
 
-    def validate
-      @valid = true
+            def validate
+              @valid = true
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/request_public_keys.rb

@@ -1,19 +1,29 @@
-module Legion::Extensions::Node::Transport::Messages
-  class RequestPublicKeys < Legion::Transport::Message
-    def routing_key
-      'node'
-    end
+# frozen_string_literal: true
 
-    def type
-      'task'
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class RequestPublicKeys < Legion::Transport::Message
+            def routing_key
+              'node'
+            end
 
-    def encrypt?
-      false
-    end
+            def type
+              'task'
+            end
+
+            def encrypt?
+              false
+            end
 
-    def validate
-      @valid = true
+            def validate
+              @valid = true
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/request_vault_token.rb

@@ -1,28 +1,38 @@
-module Legion::Extensions::Node::Transport::Messages
-  class RequestVaultToken < Legion::Transport::Message
-    def routing_key
-      'vault'
-    end
+# frozen_string_literal: true
 
-    def message
-      {
-        function: 'push_vault_token',
-        node_name: Legion::Settings[:client][:name],
-        runner_class: 'Legion::Extensions::Node::Runners::Vault',
-        public_key: Legion::Crypt.public_key
-      }
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class RequestVaultToken < Legion::Transport::Message
+            def routing_key
+              'vault'
+            end
 
-    def type
-      'task'
-    end
+            def message
+              {
+                function:     'push_vault_token',
+                node_name:    Legion::Settings[:client][:name],
+                runner_class: 'Legion::Extensions::Node::Runners::Vault',
+                public_key:   Legion::Crypt.public_key
+              }
+            end
 
-    def encrypt?
-      false
-    end
+            def type
+              'task'
+            end
+
+            def encrypt?
+              false
+            end
 
-    def validate
-      @valid = true
+            def validate
+              @valid = true
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/messages/update_result.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Messages
+          class UpdateResult < Legion::Transport::Message
+            def routing_key
+              "node.#{Legion::Settings[:client][:name]}.update_result"
+            end
+
+            def exchange
+              Legion::Transport::Exchanges::Node
+            end
+
+            def type
+              'task'
+            end
+
+            def encrypt?
+              false
+            end
+
+            def validate
+              raise 'action is required' unless @options[:action].is_a?(String)
+              raise 'status is required' unless @options[:status].is_a?(String)
+
+              @valid = true
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/node/transport/queues/crypt.rb

@@ -1,7 +1,17 @@
-module Legion::Extensions::Node::Transport::Queues
-  class Crypt < Legion::Transport::Queue
-    def queue_options
-      { auto_delete: false }
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Queues
+          class Crypt < Legion::Transport::Queue
+            def queue_options
+              { auto_delete: false }
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/queues/health.rb

@@ -1,7 +1,17 @@
-module Legion::Extensions::Node::Transport::Queues
-  class Health < Legion::Transport::Queue
-    def queue_options
-      { arguments: { 'x-single-active-consumer': true }, auto_delete: false }
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Queues
+          class Health < Legion::Transport::Queue
+            def queue_options
+              { arguments: { 'x-single-active-consumer': true }, auto_delete: false }
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/queues/node.rb

@@ -1,13 +1,24 @@
+# frozen_string_literal: true
+
 require 'socket'
 
-module Legion::Extensions::Node::Transport::Queues
-  class Node < Legion::Transport::Queue
-    def queue_name
-      "node.#{Legion::Settings[:client][:name]}"
-    end
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Queues
+          class Node < Legion::Transport::Queue
+            def queue_name
+              "node.#{Legion::Settings[:client][:name]}"
+            end
 
-    def queue_options
-      { durable: false, exclusive: true, auto_delete: true }
+            def queue_options
+              { durable: false, exclusive: true, auto_delete: true,
+                arguments: { 'x-queue-type': 'classic' } }
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport/queues/vault.rb

@@ -1,7 +1,17 @@
-module Legion::Extensions::Node::Transport::Queues
-  class Vault < Legion::Transport::Queue
-    def queue_options
-      { auto_delete: false }
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        module Queues
+          class Vault < Legion::Transport::Queue
+            def queue_options
+              { auto_delete: false }
+            end
+          end
+        end
+      end
     end
   end
 end

data/lib/legion/extensions/node/transport.rb

@@ -1,12 +1,21 @@
+# frozen_string_literal: true
+
 require 'legion/extensions/transport'
 
-module Legion::Extensions::Node::Transport
-  extend Legion::Extensions::Transport
-  def self.additional_e_to_q
-    array = [{ from: 'node', to: 'node', routing_key: "node.#{Legion::Settings[:client][:name]}" }]
-    array.push(from: 'node', to: 'node', routing_key: 'node.data.#') if Legion::Settings[:data][:connected]
-    array.push(from: 'node', to: 'node', routing_key: 'node.cache.#') if Legion::Settings[:cache][:connected]
-    array.push(from: 'node', to: 'node', routing_key: 'node.crypt.#')
-    array
+module Legion
+  module Extensions
+    module Node
+      module Transport
+        extend Legion::Extensions::Transport
+
+        def self.additional_e_to_q
+          array = [{ from: 'node', to: 'node', routing_key: "node.#{Legion::Settings[:client][:name]}" }]
+          array.push(from: 'node', to: 'node', routing_key: 'node.data.#') if Legion::Settings[:data][:connected]
+          array.push(from: 'node', to: 'node', routing_key: 'node.cache.#') if Legion::Settings[:cache][:connected]
+          array.push(from: 'node', to: 'node', routing_key: 'node.crypt.#')
+          array
+        end
+      end
+    end
   end
 end

data/lib/legion/extensions/node/version.rb

@@ -1,7 +1,9 @@
+# frozen_string_literal: true
+
 module Legion
   module Extensions
     module Node
-      VERSION = '0.1.7'.freeze
+      VERSION = '0.2.3'
     end
   end
 end

data/lib/legion/extensions/node.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'legion/extensions/node/version'
 
 module Legion