rswim 1.0.0

data/lib/rswim/integration/deserializer.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Integration
+    class Deserializer
+      def initialize(directory, my_id)
+        @directory = directory
+        @my_id = my_id
+      end
+
+      def deserialize(sender_host, wire_message)
+        l, *ls = wire_message.split("\n")
+        # First line is
+        # type [target_id]
+        ary = l.strip.split(' ')
+        type = ary[0].gsub(/-/, '_').to_sym
+        payload = type == :ping_req ? { target_id: @directory.id(ary[1]) } : {}
+        payload[:updates] = parse_updates(ls)
+        from = @directory.id(sender_host)
+        RSwim::Message.new(@my_id, from, type, payload)
+      rescue StandardError => e
+        logger.debug("Failed to parse line `#{l}`: #{e}")
+        nil
+      end
+
+      protected
+
+      def logger
+        @_logger ||= begin
+          RSwim::Logger.new(self.class, STDERR)
+        end
+      end
+
+      private
+
+      def parse_updates(lines)
+        lines.map do |l|
+          begin
+            # host status incarnation_number
+            host, status, incarnation_number = l.strip.split(' ')
+            id = @directory.id(host)
+            UpdateEntry.new(id, status.to_sym, incarnation_number.to_i)
+          rescue StandardError => e
+            logger.debug("Failed to parse line `#{l}`: #{e}")
+            nil
+          end
+        end.tap(&:compact!)
+      end
+    end
+  end
+end

data/lib/rswim/integration/serializer.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Integration
+    class Serializer
+      def initialize(directory)
+        @directory = directory
+      end
+
+      def serialize(message)
+        l1 = message.type.to_s.gsub(/_/, '-')
+        l1 << " #{@directory.host(message.payload[:target_id])}" if message.type == :ping_req
+        message.payload[:updates].to_a.each do |update|
+          # host status incarnation_number
+          l1 << "\n#{@directory.host(update.member_id)} #{update.status} #{update.incarnation_number}"
+        end
+        l1
+      end
+
+      protected
+
+      def logger
+        @_logger ||= begin
+          RSwim::Logger.new(self.class, STDERR)
+        end
+      end
+    end
+  end
+end

data/lib/rswim/integration/udp/node.rb

@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+require 'socket'
+
+module RSwim
+  module Integration
+    module Udp
+      class Node < RSwim::Node
+        def initialize(my_host, seed_hosts, port)
+          my_host ||= Socket.ip_address_list.find { |x| x.ipv4_private? }.ip_address
+          super(my_host, seed_hosts)
+          @port = port
+        end
+
+        protected
+
+        def before_start
+          @in_s = UDPSocket.new
+          @out_s = UDPSocket.new
+          @in_s.bind(@my_host, @port)
+          logger.info "node listening on UDP port #{@port}"
+          Thread.new { receive }.tap { |t| t.abort_on_exception = true }
+          Thread.new { send }.tap { |t| t.abort_on_exception = true }
+        end
+
+        private
+
+        def send
+          loop do
+            begin
+              message = @pipe.q_out.pop
+              wire_message = @serializer.serialize(message)
+              host = @directory.host(message.to)
+              logger.debug "about to send message to #{host}"
+              @out_s.send(wire_message, 0, host, @port)
+            rescue StandardError => e
+              logger.debug("Error while sending: #{e}")
+            end
+          end
+          logger.info 'node no longer receiving'
+        end
+
+        def receive
+          loop do
+            begin
+              logger.debug 'about to recieve message'
+              text, sender = @in_s.recvfrom(10_000)
+
+              message = @deserializer.deserialize(sender[3], text)
+              logger.debug "received #{message}"
+              @pipe.q_in << message
+            rescue StandardError => e
+              logger.debug("Error while receiving: #{e}")
+            end
+          end
+          logger.info 'node no longer receiving'
+        end
+      end
+    end
+  end
+end

data/lib/rswim/logger.rb

@@ -0,0 +1,31 @@
+module RSwim
+  class Logger < ::Logger
+    def initialize(klass, *args, **options)
+      options[:formatter] = LeFormatter.new(klass)
+      super(*args, **options)
+      self.level=Logger.level
+    end
+
+    class << self
+      def level=(severity)
+        @level = severity
+      end
+
+      def level
+        @level ||= ::Logger::INFO
+      end
+    end
+
+    class LeFormatter < ::Logger::Formatter
+      def initialize(klass)
+        super()
+        @klass = klass
+      end
+
+      def call(severity, timestamp, progname, msg)
+        m = String === msg ? msg : msg.inspect
+        "#{timestamp} | #{severity} | #{@klass} | #{m}\n"
+      end
+    end
+  end
+end

data/lib/rswim/member/ack_responder.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    class AckResponder
+      def initialize(id)
+        @id = id
+        @pending = []
+      end
+
+      def schedule_ack(member_id)
+        @pending << Message.new(member_id, @id, :ack)
+      end
+
+      def prepare_output
+        result = @pending
+        @pending = []
+        result
+      end
+    end
+  end
+end

data/lib/rswim/member/base.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    class Base
+      def initialize(id)
+        @id = id
+      end
+    end
+  end
+end

data/lib/rswim/member/forwarding_state/base.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module ForwardingState
+      class Base
+        def initialize(id, node_member_id)
+          @id = id
+          @node_member_id = node_member_id
+          logger.debug("Member with id #{id} entered new state: #{self.class}")
+        end
+
+        def forward_ack_to_member; end
+
+        def advance(_elapsed_seconds)
+          self
+        end
+
+        def prepare_output
+          []
+        end
+
+        protected
+
+        def logger
+          @_logger ||= begin
+            RSwim::Logger.new("Node #{@node_member_id}", STDERR)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rswim/member/forwarding_state/forwarding_ack.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module ForwardingState
+      class ForwardingAck < Base
+        def initialize(id, node_member_id)
+          super
+          @done = false
+        end
+
+        def prepare_output
+          @done = true
+          message = Message.new(@id, @node_member_id, :ack)
+          [message]
+        end
+
+        def advance(elapsed_seconds)
+          if @done then Ready.new(@id, @node_member_id)
+          else self
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rswim/member/forwarding_state/ready.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module ForwardingState
+      class Ready < Base
+        def initialize(id, node_member_id)
+          super
+          @activated = false
+        end
+
+        def forward_ack_to_member
+          @activated = true
+        end
+
+        def advance(_elapsed_seconds)
+          if @activated then ForwardingAck.new(@id, @node_member_id)
+          else self
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rswim/member/health_state/alive.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module HealthState
+      class Alive < Base
+        def initialize(id, member_pool, update_entry = UpdateEntry.new(id, :alive, 0, 0))
+          super
+          @failed_to_reply = false
+        end
+
+        def advance(_elapsed_seconds)
+          if @failed_to_reply
+            ue = UpdateEntry.new(@id, :suspected, @update_entry.incarnation_number, -1)
+            Suspected.new(@id, @member_pool, ue, true)
+          else
+            self
+          end
+        end
+
+        def member_failed_to_reply
+          @failed_to_reply = true
+        end
+
+        def can_be_pinged?
+          true
+        end
+      end
+    end
+  end
+end

data/lib/rswim/member/health_state/base.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module HealthState
+      class Base
+        attr_reader :update_entry
+
+        def initialize(id, member_pool, update_entry)
+          @member_pool = member_pool
+          @id = id
+          @update_entry = update_entry
+          logger.debug("Member with id #{id} entered new state: #{self.class}")
+        end
+
+        def advance(_elapsed_seconds)
+          self
+        end
+
+        def update_suspicion(status, incarnation_number)
+          incarnation_number ||= @update_entry.incarnation_number
+          s0 = @update_entry.status
+          i0 = @update_entry.incarnation_number
+          case status
+          when :confirmed
+            if (s0 == :confirmed)
+              self
+            else
+              ue = UpdateEntry.new(@id, status, incarnation_number, 0)
+              Confirmed.new(@id, @member_pool, ue)
+            end
+          when :suspected
+            if (s0 == :suspected && incarnation_number > i0) ||
+               (s0 == :alive && incarnation_number >= i0)
+              ue = UpdateEntry.new(@id, status, incarnation_number, 0)
+              Suspected.new(@id, @member_pool, ue, false)
+            else
+              self
+            end
+          when :alive
+            if (s0 == :suspected && incarnation_number > i0) ||
+               (s0 == :alive && incarnation_number > i0)
+              ue = UpdateEntry.new(@id, status, incarnation_number, 0)
+              Alive.new(@id, @member_pool, ue)
+            else
+              self
+            end
+          end
+        end
+
+        def member_failed_to_reply; end
+
+        def increment_propagation_count
+          @update_entry.increment_propagation_count
+        end
+
+        def can_be_pinged?
+          false
+        end
+
+        protected
+
+        def logger
+          @_logger ||= begin
+            RSwim::Logger.new("unknown node", STDERR)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rswim/member/health_state/confirmed.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module HealthState
+      class Confirmed < Base
+        def initialize(id, member_pool, update_entry)
+          super
+          @member_halted = false
+          @member_removed = false
+          @life_time_seconds = 0
+        end
+
+        def advance(elapsed_seconds)
+          @life_time_seconds += elapsed_seconds
+          unless @member_halted
+            @member_pool.halt_member(@id)
+            @member_halted = true
+          end
+          
+          if !@member_removed && @life_time_seconds > 10
+            @member_pool.remove_member(@id)
+            @member_removed = true
+          end
+
+          self
+        end
+      end
+    end
+  end
+end

data/lib/rswim/member/health_state/suspected.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module RSwim
+  module Member
+    module HealthState
+      class Suspected < Base
+        def initialize(id, member_pool, update_entry, send_ping_request)
+          super(id, member_pool, update_entry)
+          @ping_request_sent = !send_ping_request
+          @life_time_seconds = 0
+        end
+
+        def advance(elapsed_seconds)
+          @life_time_seconds += elapsed_seconds
+          unless @ping_request_sent
+            @member_pool.send_ping_request_to_k_members(@id)
+            @ping_request_sent = true
+          end
+          if @life_time_seconds > 60
+            Confirmed.new(@id, @member_pool, UpdateEntry.new(@id, :confirmed, @update_entry.incarnation_number, -2))
+          else
+            self
+          end
+        end
+
+        def can_be_pinged?
+          true
+        end
+      end
+    end
+  end
+end