54 lines
1.8 KiB
Ruby
54 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../../lib/util"
|
|
|
|
class Prog::Postgres::ConvergePostgresResource < Prog::Base
|
|
subject_is :postgres_resource
|
|
|
|
label def start
|
|
register_deadline(nil, 2 * 60 * 60)
|
|
hop_provision_servers
|
|
end
|
|
|
|
label def provision_servers
|
|
hop_wait_servers_to_be_ready if postgres_resource.has_enough_fresh_servers?
|
|
|
|
if postgres_resource.servers.all? { _1.vm.vm_host }
|
|
exclude_host_ids = (Config.development? || Config.is_e2e) ? [] : postgres_resource.servers.map { _1.vm.vm_host.id }
|
|
Prog::Postgres::PostgresServerNexus.assemble(resource_id: postgres_resource.id, timeline_id: postgres_resource.timeline.id, timeline_access: "fetch", exclude_host_ids: exclude_host_ids)
|
|
end
|
|
|
|
nap 5
|
|
end
|
|
|
|
label def wait_servers_to_be_ready
|
|
hop_provision_servers unless postgres_resource.has_enough_fresh_servers?
|
|
hop_recycle_representative_server if postgres_resource.has_enough_ready_servers?
|
|
|
|
nap 60
|
|
end
|
|
|
|
label def recycle_representative_server
|
|
if (rs = postgres_resource.representative_server)
|
|
hop_prune_servers unless rs.needs_recycling?
|
|
rs.trigger_failover
|
|
end
|
|
|
|
nap 60
|
|
end
|
|
|
|
label def prune_servers
|
|
# Below we only keep servers that does not need recycling. If there are
|
|
# more such servers than required, we prefer ready and recent servers (in that order)
|
|
servers_to_keep = postgres_resource.servers
|
|
.reject { _1.representative_at || _1.needs_recycling? }
|
|
.sort_by { [(_1.strand.label == "wait") ? 0 : 1, Time.now - _1.created_at] }
|
|
.take(postgres_resource.target_standby_count) + [postgres_resource.representative_server]
|
|
(postgres_resource.servers - servers_to_keep).each.each(&:incr_destroy)
|
|
|
|
postgres_resource.incr_update_billing_records
|
|
|
|
pop "postgres resource is converged"
|
|
end
|
|
end
|