lex-mesh 0.2.4 → 0.2.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 25cafa977ed6aaf671c903e71895312253a412c32c141e8de9674b1a86747bff
-  data.tar.gz: c7bbdde6f4c065f0fdaea58a5a5124bd53e49f0c48e0d275de0828bd93ec2439
+  metadata.gz: 650958df48e1ad4fa390b3fecb3ab2817dc535c279449dc4ce114e92f59cfff2
+  data.tar.gz: ff13ef5a0e18147daa15b72579de24f3691bec50f5f124627d2b503208cfba82
 SHA512:
-  metadata.gz: 76357a6ca4c31089fd7f5317097f935b07df6349476b344bd2681dccfbd796d79b1061922b22d42ca45b92d045bdbf901ed8d6c502f525cff3a668099dfea56c
-  data.tar.gz: 9e17a698c0b941fd2000ab9affffa87cd3bad983b37acacce794ee0fb4d975a8cdc04eee44480f976856e712fb6af8dfa80f976d07c2fd94ea302ba61f31b389
+  metadata.gz: fc982147fb2a22e4a5bb40fa8f7a767e87aed754f59474d4937059a831adcfeae6b4f4e4c9add47359c3b6a69200a0cdea21c1bafd1b00dfd74501c93415eb40
+  data.tar.gz: e34a1a0d9ae8481bb40949bda0e47d6e1647856777214d133acf10c121b8e70962bb85c10fda16120b9dd7bc958828fc20a73851ab259c89fa813d7df2236f26

data/lib/legion/extensions/mesh/actors/gossip.rb

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require 'legion/extensions/actors/every'
+
+module Legion
+  module Extensions
+    module Mesh
+      module Actor
+        class Gossip < Legion::Extensions::Actors::Every
+          def runner_class
+            Legion::Extensions::Mesh::Runners::Mesh
+          end
+
+          def runner_function
+            :publish_gossip
+          end
+
+          def time
+            15
+          end
+
+          def use_runner?
+            true
+          end
+
+          def check_subtask?
+            false
+          end
+
+          def generate_task?
+            false
+          end
+
+          def args
+            {}
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/mesh/helpers/registry.rb

@@ -13,11 +13,14 @@ module Legion
             @messages = []
           end
 
-          def register_agent(agent_id, capabilities: [], endpoint: nil)
+          def register_agent(agent_id, capabilities: [], endpoint: nil, source: :native, node: nil)
             @agents[agent_id] = {
               agent_id:      agent_id,
               capabilities:  capabilities,
               endpoint:      endpoint,
+              source:        source,
+              node:          node || local_node_name,
+              generation:    1,
               registered_at: Time.now.utc,
               last_seen:     Time.now.utc,
               status:        :online
@@ -39,6 +42,7 @@ module Legion
 
             agent[:last_seen] = Time.now.utc
             agent[:status] = :online
+            agent[:generation] = (agent[:generation] || 0) + 1
           end
 
           def find_by_capability(capability)
@@ -81,9 +85,21 @@ module Legion
             @agents.values.select { |a| a[:status] == :online }
           end
 
+          def all_agents
+            @agents.values
+          end
+
           def count
             @agents.size
           end
+
+          private
+
+          def local_node_name
+            Legion::Settings[:client][:name]
+          rescue StandardError
+            'unknown'
+          end
         end
       end
     end

data/lib/legion/extensions/mesh/runners/mesh.rb

@@ -64,6 +64,50 @@ module Legion
             { success: true, expired: expired, count: expired.size }
           end
 
+          def publish_gossip(**)
+            registry = mesh_registry
+            peers = registry.all_agents.first(gossip_max_peers).map do |agent|
+              agent.slice(:agent_id, :capabilities, :node, :source, :status, :generation,
+                          :last_seen, :registered_at).transform_values { |v| v.is_a?(Time) ? v.to_s : v }
+            end
+
+            @gossip_round = (@gossip_round || 0) + 1
+            publish_gossip_message(peers)
+            { success: true, peers_broadcast: peers.size, gossip_round: @gossip_round }
+          rescue StandardError => e
+            { success: false, reason: :error, message: e.message }
+          end
+
+          def merge_gossip(incoming_peers:, sender: nil, **) # rubocop:disable Lint/UnusedMethodArgument
+            registry = mesh_registry
+            merged = 0
+
+            incoming_peers.each do |peer|
+              peer = peer.transform_keys(&:to_sym)
+              next if peer[:node] == local_node_name
+
+              local = registry.agents[peer[:agent_id]]
+              if local.nil?
+                registry.register_agent(
+                  peer[:agent_id],
+                  capabilities: (peer[:capabilities] || []).map(&:to_sym),
+                  source:       (peer[:source] || :native).to_sym,
+                  node:         peer[:node]
+                )
+                registry.agents[peer[:agent_id]][:generation] = peer[:generation] || 1
+                merged += 1
+              elsif (peer[:generation] || 0) > (local[:generation] || 0)
+                local.merge!(peer.slice(:capabilities, :status, :generation, :last_seen))
+                local[:capabilities] = (local[:capabilities] || []).map(&:to_sym)
+                merged += 1
+              end
+            end
+
+            { success: true, merged: merged, total_peers: incoming_peers.size }
+          rescue StandardError => e
+            { success: false, reason: :error, message: e.message }
+          end
+
           private
 
           def publish_mesh_departure(agent_id:, capabilities:)
@@ -77,6 +121,29 @@ module Legion
             Legion::Logging.warn "[mesh] failed to publish departure signal: #{e.message}"
           end
 
+          def publish_gossip_message(peers)
+            return unless defined?(Legion::Extensions::Mesh::Transport::Messages::Gossip)
+
+            Legion::Extensions::Mesh::Transport::Messages::Gossip.new(
+              sender:       local_node_name,
+              gossip_round: @gossip_round,
+              peers:        peers
+            ).publish
+          end
+
+          def gossip_max_peers
+            settings = Legion::Settings.dig(:mesh, :gossip)
+            (settings.is_a?(Hash) ? settings[:max_peers_per_message] : nil) || 100
+          rescue StandardError
+            100
+          end
+
+          def local_node_name
+            Legion::Settings[:client][:name]
+          rescue StandardError
+            'unknown'
+          end
+
           def mesh_registry
             @mesh_registry ||= Helpers::Registry.new
           end

data/lib/legion/extensions/mesh/transport/messages/gossip.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Mesh
+      module Transport
+        module Messages
+          class Gossip < Legion::Transport::Message
+            def exchange
+              Legion::Transport::Exchanges::Node
+            end
+
+            def routing_key
+              'mesh.gossip'
+            end
+
+            def message
+              {
+                type:         'mesh_gossip',
+                sender:       @options[:sender],
+                gossip_round: @options[:gossip_round] || 0,
+                peers:        @options[:peers] || []
+              }
+            end
+
+            def type
+              'mesh_gossip'
+            end
+
+            def encrypt?
+              false
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/legion/extensions/mesh/transport/queues/gossip.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Legion
+  module Extensions
+    module Mesh
+      module Transport
+        module Queues
+          class Gossip < Legion::Transport::Queue
+            def queue_name
+              'mesh.gossip'
+            end
+
+            def exchange
+              Legion::Transport::Exchanges::Node
+            end
+
+            def routing_key
+              'mesh.gossip'
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -3,7 +3,7 @@
 module Legion
   module Extensions
     module Mesh
-      VERSION = '0.2.4'
+      VERSION = '0.2.5'
     end
   end
 end

data/spec/legion/extensions/mesh/actors/gossip_spec.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+# Stub the framework actor base class since legionio gem is not available in test
+unless defined?(Legion::Extensions::Actors::Every)
+  module Legion
+    module Extensions
+      module Actors
+        class Every # rubocop:disable Lint/EmptyClass
+        end
+      end
+    end
+  end
+end
+
+$LOADED_FEATURES << 'legion/extensions/actors/every' unless $LOADED_FEATURES.include?('legion/extensions/actors/every')
+
+require 'legion/extensions/mesh/actors/gossip'
+
+RSpec.describe Legion::Extensions::Mesh::Actor::Gossip do
+  it 'fires every 15 seconds' do
+    expect(described_class.new.time).to eq(15)
+  end
+
+  it 'calls publish_gossip runner function' do
+    instance = described_class.allocate
+    expect(instance.runner_function).to eq(:publish_gossip)
+  end
+end

data/spec/legion/extensions/mesh/helpers/registry_spec.rb

@@ -313,6 +313,32 @@ RSpec.describe Legion::Extensions::Mesh::Helpers::Registry do
     end
   end
 
+  describe 'gossip fields' do
+    it 'stores source, node, and generation on registration' do
+      registry.register_agent('agent-1', capabilities: [:test], source: :native, node: 'node-01')
+      agent = registry.agents['agent-1']
+      expect(agent[:source]).to eq(:native)
+      expect(agent[:node]).to eq('node-01')
+      expect(agent[:generation]).to eq(1)
+    end
+
+    it 'increments generation on heartbeat' do
+      registry.register_agent('agent-1', capabilities: [:test])
+      registry.heartbeat('agent-1')
+      expect(registry.agents['agent-1'][:generation]).to eq(2)
+    end
+  end
+
+  describe '#all_agents' do
+    it 'returns all agent records as an array' do
+      registry.register_agent('a1', capabilities: [:search])
+      registry.register_agent('a2', capabilities: [:compute])
+      all = registry.all_agents
+      expect(all).to be_an(Array)
+      expect(all.size).to eq(2)
+    end
+  end
+
   describe '#count' do
     it 'returns 0 for an empty registry' do
       expect(registry.count).to eq(0)

data/spec/legion/extensions/mesh/runners/mesh_gossip_spec.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe 'Gossip runner methods' do
+  subject { Object.new.extend(Legion::Extensions::Mesh::Runners::Mesh) }
+
+  before do
+    allow(subject).to receive(:mesh_registry).and_return(
+      Legion::Extensions::Mesh::Helpers::Registry.new
+    )
+  end
+
+  describe '#merge_gossip' do
+    it 'adds unknown agents from incoming gossip' do
+      incoming = [
+        { agent_id: 'remote-1', capabilities: [:test], node: 'node-02',
+          source: :native, status: :online, generation: 5,
+          last_seen: Time.now.utc.to_s, registered_at: Time.now.utc.to_s }
+      ]
+      result = subject.merge_gossip(incoming_peers: incoming, sender: 'node-02')
+      expect(result[:success]).to be true
+      expect(result[:merged]).to eq(1)
+    end
+
+    it 'updates agents with higher generation' do
+      registry = subject.send(:mesh_registry)
+      registry.register_agent('agent-1', capabilities: [:old], node: 'node-01')
+
+      incoming = [
+        { agent_id: 'agent-1', capabilities: [:updated], node: 'node-01',
+          source: :native, status: :online, generation: 99,
+          last_seen: Time.now.utc.to_s, registered_at: Time.now.utc.to_s }
+      ]
+      result = subject.merge_gossip(incoming_peers: incoming, sender: 'node-01')
+      expect(result[:merged]).to eq(1)
+      expect(registry.agents['agent-1'][:generation]).to eq(99)
+    end
+
+    it 'skips agents from local node' do
+      local_name = 'local-node'
+      allow(subject).to receive(:local_node_name).and_return(local_name)
+
+      incoming = [
+        { agent_id: 'local-agent', node: local_name, generation: 1 }
+      ]
+      result = subject.merge_gossip(incoming_peers: incoming, sender: 'other-node')
+      expect(result[:merged]).to eq(0)
+    end
+
+    it 'ignores agents with lower or equal generation' do
+      registry = subject.send(:mesh_registry)
+      registry.register_agent('agent-1', capabilities: [:test], node: 'node-01')
+      # generation is 1 after register
+
+      incoming = [
+        { agent_id: 'agent-1', capabilities: [:test], node: 'node-01',
+          generation: 1, source: :native, status: :online }
+      ]
+      result = subject.merge_gossip(incoming_peers: incoming, sender: 'node-02')
+      expect(result[:merged]).to eq(0)
+    end
+  end
+
+  describe '#publish_gossip' do
+    it 'returns success with peer count' do
+      allow(subject).to receive(:publish_gossip_message)
+      result = subject.publish_gossip
+      expect(result[:success]).to be true
+      expect(result[:peers_broadcast]).to be_a(Integer)
+    end
+  end
+end

data/spec/legion/extensions/mesh/transport/messages/gossip_spec.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+unless defined?(Legion::Transport::Message)
+  module Legion
+    module Transport
+      class Message
+        def initialize(**options)
+          @options = options
+        end
+      end
+    end
+  end
+end
+
+unless defined?(Legion::Transport::Exchanges::Node)
+  module Legion
+    module Transport
+      module Exchanges
+        class Node # rubocop:disable Lint/EmptyClass
+        end
+      end
+    end
+  end
+end
+
+require 'legion/extensions/mesh/transport/messages/gossip'
+
+RSpec.describe Legion::Extensions::Mesh::Transport::Messages::Gossip do
+  subject do
+    described_class.new(
+      sender:       'node-01',
+      gossip_round: 5,
+      peers:        [{ agent_id: 'w1', capabilities: [:test], node: 'node-01', generation: 3 }]
+    )
+  end
+
+  it 'uses the node exchange' do
+    expect(subject.exchange).to eq(Legion::Transport::Exchanges::Node)
+  end
+
+  it 'routes to mesh.gossip' do
+    expect(subject.routing_key).to eq('mesh.gossip')
+  end
+
+  it 'includes sender and peers in the message body' do
+    msg = subject.message
+    expect(msg[:type]).to eq('mesh_gossip')
+    expect(msg[:sender]).to eq('node-01')
+    expect(msg[:peers]).to be_an(Array)
+    expect(msg[:gossip_round]).to eq(5)
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lex-mesh
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.2.5
 platform: ruby
 authors:
 - Esity
@@ -48,6 +48,7 @@ files:
 - Gemfile
 - lex-mesh.gemspec
 - lib/legion/extensions/mesh.rb
+- lib/legion/extensions/mesh/actors/gossip.rb
 - lib/legion/extensions/mesh/actors/heartbeat.rb
 - lib/legion/extensions/mesh/actors/pending_expiry.rb
 - lib/legion/extensions/mesh/actors/preference_listener.rb
@@ -62,11 +63,14 @@ files:
 - lib/legion/extensions/mesh/runners/delegation.rb
 - lib/legion/extensions/mesh/runners/mesh.rb
 - lib/legion/extensions/mesh/runners/preferences.rb
+- lib/legion/extensions/mesh/transport/messages/gossip.rb
 - lib/legion/extensions/mesh/transport/messages/mesh_departure.rb
 - lib/legion/extensions/mesh/transport/messages/preference_query.rb
 - lib/legion/extensions/mesh/transport/messages/preference_response.rb
+- lib/legion/extensions/mesh/transport/queues/gossip.rb
 - lib/legion/extensions/mesh/transport/queues/preference.rb
 - lib/legion/extensions/mesh/version.rb
+- spec/legion/extensions/mesh/actors/gossip_spec.rb
 - spec/legion/extensions/mesh/actors/heartbeat_spec.rb
 - spec/legion/extensions/mesh/actors/pending_expiry_spec.rb
 - spec/legion/extensions/mesh/actors/preference_listener_spec.rb
@@ -79,8 +83,10 @@ files:
 - spec/legion/extensions/mesh/helpers/registry_spec.rb
 - spec/legion/extensions/mesh/helpers/topology_spec.rb
 - spec/legion/extensions/mesh/runners/delegation_spec.rb
+- spec/legion/extensions/mesh/runners/mesh_gossip_spec.rb
 - spec/legion/extensions/mesh/runners/mesh_spec.rb
 - spec/legion/extensions/mesh/runners/preferences_spec.rb
+- spec/legion/extensions/mesh/transport/messages/gossip_spec.rb
 - spec/legion/extensions/mesh/transport/messages/mesh_departure_spec.rb
 - spec/legion/extensions/mesh/transport/messages/preference_query_spec.rb
 - spec/legion/extensions/mesh/transport/messages/preference_response_spec.rb