double_decker 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 84c51fbc0973e6d5c7fc7f6522de3c6a015423fef30a9bc59726bea16612db7e
-  data.tar.gz: c4ffeff4e80278ee3213d12022bd5766ca0033edf55926334c0b4a3a0be00684
+  metadata.gz: 720e2cbc7fe7d4e394fa6334c31323777fbd849146b3ea71b919b4683fbedc7b
+  data.tar.gz: c076994dcb91a33c6d58539eb0e69e3dd4a27262fd7e8f299ca84da389cec7a9
 SHA512:
-  metadata.gz: 4bf1066987f51577ae81fe50923f83217b696c501ca6f6ca29e33d76ac03497287ce57f055a092bf3ac11d4c59d0eb813501ab12967662f7bfeb12cabf27ac04
-  data.tar.gz: 26423775150edf5cba75744e82bcfaf89e9fda5927808066442e0d5a8049723f5aef7a2c60af45ed512355d1427daaed474a079ac595b11dfb011f6acae6d6b7
+  metadata.gz: 9702a9da87d92ad5aa9271ac64937f5e975c17ca321b462311413d66e06413f5d3434986b488562917bb1575de73735b7580e6eb90ec39b71f34b1e02437d89e
+  data.tar.gz: 15877c849403d831319ffce729adfab623140c470bd23d3f574cf7bdc1cda68c9b6c6c47c596082b9d1103dc828b1c3c1d293b12388b7b4374f790e936c83d3a

data/src/double_decker/agent.rb

@@ -0,0 +1,65 @@
+module DoubleDecker
+  class Agent
+    attr_reader :id, :end_time
+
+    def initialize(bus, bus_data, expected_agents, merge_proc: nil)
+      @bus = bus
+      @bus_data = bus_data
+      @expected_agents = expected_agents
+      @finished = nil
+      @merge_proc = merge_proc
+      @id = setup!
+      merge({})
+    end
+
+    def merge_like(&block)
+      @merge_proc = block
+    end
+
+    def merge(hash)
+      @bus_data.merge(id, hash, &@merge_proc)
+    end
+
+    def to_h
+      @bus_data.to_h[id.to_s]
+    end
+
+    def on_finished(&block)
+      @finished = block
+    end
+
+    def wait_for_expected_agents(max_tries=0)
+      if @expected_agents
+        puts "AGENTS: #{@bus_data.active_agents} EXPECTED: #{@expected_agents}"
+        loop do 
+          break if (@bus_data.active_agents == @expected_agents.to_i) || (max_tries == 5)
+          sleep 1
+          max_tries += 1
+        end
+      end
+    end
+
+    def finish!(end_time = DateTime.now)
+      @end_time = end_time
+      @finished&.call(to_h)
+      if last
+        @bus.finished&.call(@bus_data.to_h)
+        @bus_data.teardown_bus
+      end
+    end
+
+    private
+    
+    def last
+      teardown!.zero?
+    end
+
+    def teardown!
+      @bus_data.teardown_agent
+    end
+
+    def setup!
+      @bus_data.setup_agent
+    end
+  end
+end

data/src/double_decker/bus.rb

@@ -0,0 +1,34 @@
+require "redis" 
+require "date"
+require "json"
+require_relative "./agent"
+require_relative "./bus_data"
+
+module DoubleDecker
+  class Bus
+    attr_reader :run_id, :expected_agents, :finished
+
+    def initialize(run_id, url: "redis://localhost", expected_agents: nil, redis: Redis.new(url: url))
+      @store = redis
+      @run_id = run_id
+      @expected_agents = expected_agents
+      setup!
+    end
+
+    def register(&merge_proc)
+      agent = Agent.new(self, BusData.new(run_id, @store), expected_agents, merge_proc: merge_proc)
+      agent.wait_for_expected_agents
+      agent
+    end
+
+    def on_finished(&block)
+      @finished = block
+    end
+
+    protected
+
+    def setup!
+      @store.get(run_id) || @store.set(run_id, {}.to_json)
+    end
+  end
+end

data/src/double_decker/bus_data.rb

@@ -0,0 +1,42 @@
+require "json"
+
+module DoubleDecker
+  class BusData
+    attr_reader :run_id, :store
+
+    def initialize(run_id, store)
+      @run_id = run_id
+      @store = store
+    end
+
+    def to_h
+     data = store.get(run_id) 
+     data && JSON.parse(data)
+    end
+
+    def setup_agent
+      store.incr("#{run_id}_agents").to_i
+    end
+
+    def teardown_bus
+      store.del(run_id)
+      store.del("#{run_id}_agents")
+    end
+
+    def teardown_agent
+      store.decr("#{run_id}_agents").to_i
+    end
+
+    def active_agents
+      store.get("#{run_id}_agents").to_i
+    end
+
+    def merge(agent_id, hash, &block)
+      block.call(hash) if block
+      data = to_h || {}
+      data[agent_id.to_s] ||= {}
+      data[agent_id.to_s].merge!(hash)
+      store.set(run_id, data.to_json)
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: double_decker
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Sean Gregory
@@ -17,6 +17,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - src/double_decker.rb
+- src/double_decker/agent.rb
+- src/double_decker/bus.rb
+- src/double_decker/bus_data.rb
 homepage: https://rubygems.org/gems/double_decker
 licenses:
 - MIT