zold-io/zold

View on GitHub
lib/zold/commands/routines/reconnect.rb

Summary

Maintainability
A
1 hr
Test Coverage
# frozen_string_literal: true

# Copyright (c) 2018-2023 Zerocracy
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the 'Software'), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

require 'shellwords'
require_relative '../routines'
require_relative '../remote'
require_relative '../../node/farm'

# Reconnect routine.
# Author:: Yegor Bugayenko (yegor256@gmail.com)
# Copyright:: Copyright (c) 2018 Yegor Bugayenko
# License:: MIT
class Zold::Routines::Reconnect
  def initialize(opts, remotes, farm = Zold::Farm::Empty.new, log: Log::NULL)
    @opts = opts
    @remotes = remotes
    @farm = farm
    @log = log
  end

  def exec(step = 0)
    sleep(60) unless @opts['routine-immediately']
    cmd = Zold::Remote.new(remotes: @remotes, log: @log, farm: @farm)
    args = ['remote', "--network=#{Shellwords.escape(@opts['network'])}", '--ignore-ping']
    score = @farm.best[0]
    args << "--ignore-node=#{Shellwords.escape("#{score.host}:#{score.port}")}" if score
    cmd.run(args + ['masters']) unless @opts['routine-immediately']
    return if @opts['routine-immediately'] && @remotes.all.empty?
    cmd.run(args + ['select'])
    if (step % 10).zero?
      @log.info("It is round ##{step}, time to update the list of remotes")
      update(cmd, args)
    end
    if @remotes.all.any? { |r| r[:errors] > Zold::Remotes::TOLERANCE }
      @log.info('There are a few remote nodes with too many errors, it\'s time to update')
      update(cmd, args)
    else
      @log.debug("All #{@remotes.all.count} remote nodes are still more or less error-free, won't update")
    end
    if @remotes.all.count < Zold::Remotes::MAX_NODES / 2
      @log.info("There are just #{@remotes.all.count} remotes in the list, time to update")
      update(cmd, args)
    end
    cmd.run(args + ['trim'])
    cmd.run(args + ['select'])
    @log.info("Reconnected, there are #{@remotes.all.count} remote notes: \
#{@remotes.all.map { |r| "#{r[:host]}:#{r[:port]}/#{r[:score]}/#{r[:errors]}" }.join(', ')}")
  end

  private

  def update(cmd, args)
    cmd.run(args + ['update'] + (@opts['never-reboot'] ? [] : ['--reboot']))
  end
end