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
 
# 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

  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 



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

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



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

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