Class: KubernetesDeploy::RestartTask

Inherits:
Object
  • Object
show all
Includes:
KubeclientBuilder
Defined in:
lib/kubernetes-deploy/restart_task.rb

Defined Under Namespace

Classes: DeploymentNotFoundError, RestartError

Constant Summary collapse

HTTP_OK_RANGE =
200..299
ANNOTATION =
"shipit.shopify.io/restart"

Instance Method Summary collapse

Constructor Details

#initialize(context:, namespace:, logger:) ⇒ RestartTask

Returns a new instance of RestartTask.



26
27
28
29
30
31
32
33
# File 'lib/kubernetes-deploy/restart_task.rb', line 26

def initialize(context:, namespace:, logger:)
  @context = context
  @namespace = namespace
  @logger = logger
  @kubeclient = build_v1_kubeclient(context)
  @v1beta1_kubeclient = build_v1beta1_kubeclient(context)
  @policy_v1beta1_kubeclient = build_policy_v1beta1_kubeclient(context)
end

Instance Method Details

#perform(deployments_names = nil) ⇒ Object



35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# File 'lib/kubernetes-deploy/restart_task.rb', line 35

def perform(deployments_names = nil)
  @logger.reset
  verify_namespace

  if deployments_names
    deployments = fetch_deployments(deployments_names.uniq)

    if deployments.none?
      raise ArgumentError, "no deployments with names #{deployments_names} found in namespace #{@namespace}"
    end
  else
    deployments = @v1beta1_kubeclient
      .get_deployments(namespace: @namespace)
      .select { |d| d..annotations[ANNOTATION] }

    if deployments.none?
      raise ArgumentError, "no deployments found in namespace #{@namespace} with #{ANNOTATION} annotation available"
    end
  end

  @logger.phase_heading("Triggering restart by touching ENV[RESTARTED_AT]")
  patch_kubeclient_deployments(deployments)

  @logger.phase_heading("Waiting for rollout")
  wait_for_rollout(deployments)

  names = deployments.map { |d| "`#{d..name}`" }
  @logger.info "Restart of #{names.sort.join(', ')} deployments succeeded"
  true
rescue FatalDeploymentError => error
  @logger.fatal "#{error.class}: #{error.message}"
  false
end