Class: Chef::Knife::UpdateRun

Inherits:
Chef::Knife show all
Includes:
Update
Defined in:
lib/chef/knife/update_run.rb

Constant Summary collapse

BATCH_SIZE = 
50

Instance Method Summary collapse

Methods included from Update

#except, included, #node_list, #only, #recipes, #target_version

Instance Method Details

#run ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
# File 'lib/chef/knife/update_run.rb', line 81

def run
  list = node_list
  compatible = recipes(only(list, target_version))
  unknown = recipes(except(list, target_version))
  compatible.each { |recipe, _| unknown.delete(recipe.to_s) }

  nodes = list.inject([]) do |memo, (version, nodes)|
    if version != target_version
      nodes.each do |node|
        if config[:force] || node.recipes.all? { |recipe| compatible.keys.include?(recipe) }
          memo << node
        end
      end
    end

    memo
  end

  if config[:force] && unknown.any?
    ui.warn "It's unknown if the following recipes are Chef #{target_version} compatible: #{unknown.keys.sort.join(', ')}"
  end

  if nodes.size > 0
    if !config[:dry_run]
      ui.confirm "Are you sure you want to update #{nodes.size} nodes to chef #{target_version}"
      nodes.each_slice(BATCH_SIZE) { |batch| update_ssh(batch, target_version).run }
    else
      ui.msg "Dry run. The following #{nodes.size} nodes would get updated:"
      nodes.each { |node| ui.msg "  #{node.fqdn}" }
    end
  elsif list.size > 0
    ui.msg "None of the nodes can be updated. Try option --force"
  else
    ui.msg "No nodes found."
  end
end

#ssh_command(version) ⇒ Object 



136
137
138
139
140
141
142
143
144
 
# File 'lib/chef/knife/update_run.rb', line 136

def ssh_command(version)
  command = "gem install chef -v #{version} --no-rdoc --no-ri"

  if config[:use_sudo]
    command = "sudo #{command}"
  end

  command
end

#update_ssh(nodes, version) ⇒ Object

Update given nodes to specified version



119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/chef/knife/update_run.rb', line 119

def update_ssh(nodes, version)
  fqdns = nodes.map { |node| node.fqdn }.join(' ')

  ssh = Chef::Knife::Ssh.new
  ssh.ui = ui
  ssh.name_args = [ fqdns, ssh_command(version) ]
  ssh.config[:manual] = true
  ssh.config[:on_error] = :skip
  ssh.config[:ssh_user] = Chef::Config[:knife][:ssh_user] || config[:ssh_user]
  ssh.config[:ssh_password] = config[:ssh_password]
  ssh.config[:ssh_port] = Chef::Config[:knife][:ssh_port] || config[:ssh_port]
  ssh.config[:ssh_gateway] = Chef::Config[:knife][:ssh_gateway] || config[:ssh_gateway]
  ssh.config[:identity_file] = Chef::Config[:knife][:identity_file] || config[:identity_file]
  ssh.config[:host_key_verify] = Chef::Config[:knife][:host_key_verify] || config[:host_key_verify]
  ssh
end