ManageIQ/manageiq

class MiqServer::WorkerManagement
  include Vmdb::Logging

  include Dequeue
  include Heartbeat
  include Monitor

  attr_reader :my_server, :workers

  def self.build(my_server)
    klass = if podified?
              Kubernetes
            elsif systemd?
              Systemd
            else
              Process
            end

    klass.new(my_server)
  end

  def self.podified?
    MiqEnvironment::Command.is_podified?
  end

  def self.systemd?
    MiqEnvironment::Command.supports_systemd?
  end

  def initialize(my_server)
    @my_server           = my_server
    @workers_lock        = Sync.new
    @workers             = {}
    @queue_messages_lock = Sync.new
    @queue_messages      = {}
  end

  delegate :miq_workers, :to => :my_server

  def start_workers
    clean_heartbeat_files # Appliance specific
    sync_config
    start_drb_server
    sync_workers
    wait_for_started_workers
  end

  def sync_starting_workers
    raise NotImplementedError, _("must be implemented in a subclass")
  end

  def start_drb_server
    require 'drb'
    require 'drb/acl'

    acl = ACL.new(%w[deny all allow 127.0.0.1/32])
    DRb.install_acl(acl)

    require 'tmpdir'
    Dir::Tmpname.create("worker_monitor", nil) do |path|
      drb = DRb.start_service("drbunix://#{path}", self)
      FileUtils.chmod(0o750, path)
      my_server.update(:drb_uri => drb.uri)
    end
  end

  def worker_add(worker_pid)
    @workers_lock.synchronize(:EX) { @workers[worker_pid] ||= {} }
  end

  def worker_delete(worker_pid)
    @workers_lock.synchronize(:EX) { @workers.delete(worker_pid) }
  end

  def find_worker(worker)
    worker = miq_workers.find_by(:id => worker) if worker.kind_of?(Integer)
    _log.warn("Cannot find Worker <#{worker.inspect}>") if worker.nil?

    worker
  end

  # remove workers from the miq_workers table and the in memory cache.
  def remove_workers(workers)
    return if workers.empty?

    workers.each do |w|
      yield(w)
      worker_delete(w.pid)
      w.destroy
    end
    miq_workers.reload
  end
end