|
| 1 | +# frozen_string_literal: true |
| 2 | + |
| 3 | +require "forwardable" |
| 4 | + |
| 5 | +class MaintenanceMode |
| 6 | + extend Forwardable |
| 7 | + |
| 8 | + def_delegators :@command, :config, :progress, :cp, :step, :run_cpflow_command |
| 9 | + |
| 10 | + def initialize(command) |
| 11 | + @command = command |
| 12 | + end |
| 13 | + |
| 14 | + def enabled? |
| 15 | + validate_domain_exists! |
| 16 | + cp.domain_workload_matches?(domain_data, maintenance_workload) |
| 17 | + end |
| 18 | + |
| 19 | + def disabled? |
| 20 | + validate_domain_exists! |
| 21 | + cp.domain_workload_matches?(domain_data, one_off_workload) |
| 22 | + end |
| 23 | + |
| 24 | + def enable! |
| 25 | + if enabled? |
| 26 | + progress.puts("Maintenance mode is already enabled for app '#{config.app}'.") |
| 27 | + else |
| 28 | + enable_maintenance_mode |
| 29 | + end |
| 30 | + end |
| 31 | + |
| 32 | + def disable! |
| 33 | + if disabled? |
| 34 | + progress.puts("Maintenance mode is already disabled for app '#{config.app}'.") |
| 35 | + else |
| 36 | + disable_maintenance_mode |
| 37 | + end |
| 38 | + end |
| 39 | + |
| 40 | + private |
| 41 | + |
| 42 | + def enable_maintenance_mode |
| 43 | + validate_maintenance_workload_exists! |
| 44 | + |
| 45 | + start_or_stop_maintenance_workload(:start) |
| 46 | + switch_domain_workload(to: maintenance_workload) |
| 47 | + start_or_stop_all_workloads(:stop) |
| 48 | + |
| 49 | + progress.puts("\nMaintenance mode enabled for app '#{config.app}'.") |
| 50 | + end |
| 51 | + |
| 52 | + def disable_maintenance_mode |
| 53 | + validate_maintenance_workload_exists! |
| 54 | + |
| 55 | + start_or_stop_maintenance_workload(:start) |
| 56 | + switch_domain_workload(to: one_off_workload) |
| 57 | + start_or_stop_all_workloads(:stop) |
| 58 | + |
| 59 | + progress.puts("\nMaintenance mode disabled for app '#{config.app}'.") |
| 60 | + end |
| 61 | + |
| 62 | + def validate_domain_exists! |
| 63 | + return if domain_data |
| 64 | + |
| 65 | + raise "Can't find domain. " \ |
| 66 | + "Maintenance mode is only supported for domains that use path based routing mode " \ |
| 67 | + "and have a route configured for the prefix '/' on either port 80 or 443." |
| 68 | + end |
| 69 | + |
| 70 | + def validate_maintenance_workload_exists! |
| 71 | + cp.fetch_workload!(maintenance_workload) |
| 72 | + end |
| 73 | + |
| 74 | + def start_or_stop_all_workloads(action) |
| 75 | + run_cpflow_command("ps:#{action}", "-a", config.app, "--wait") |
| 76 | + |
| 77 | + progress.puts |
| 78 | + end |
| 79 | + |
| 80 | + def start_or_stop_maintenance_workload(action) |
| 81 | + run_cpflow_command("ps:#{action}", "-a", config.app, "-w", maintenance_workload, "--wait") |
| 82 | + |
| 83 | + progress.puts |
| 84 | + end |
| 85 | + |
| 86 | + def switch_domain_workload(to:) |
| 87 | + step("Switching workload for domain '#{domain_data['name']}' to '#{to}'") do |
| 88 | + cp.set_domain_workload(domain_data, to) |
| 89 | + |
| 90 | + # Give it a bit of time for the domain to update |
| 91 | + Kernel.sleep(30) |
| 92 | + end |
| 93 | + |
| 94 | + progress.puts |
| 95 | + end |
| 96 | + |
| 97 | + def domain_data |
| 98 | + @domain_data ||= |
| 99 | + if config.domain |
| 100 | + cp.fetch_domain(config.domain) |
| 101 | + else |
| 102 | + cp.find_domain_for([one_off_workload, maintenance_workload]) |
| 103 | + end |
| 104 | + end |
| 105 | + |
| 106 | + def one_off_workload |
| 107 | + @one_off_workload ||= config[:one_off_workload] |
| 108 | + end |
| 109 | + |
| 110 | + def maintenance_workload |
| 111 | + @maintenance_workload ||= config.current[:maintenance_workload] || "maintenance" |
| 112 | + end |
| 113 | +end |
0 commit comments