Class: Kontena::Machine::DigitalOcean::NodeProvisioner
- Inherits:
-
Object
- Object
- Kontena::Machine::DigitalOcean::NodeProvisioner
- Includes:
- Cli::ShellSpinner, RandomName
- Defined in:
- lib/kontena/machine/digital_ocean/node_provisioner.rb
Instance Attribute Summary collapse
-
#api_client ⇒ Object
readonly
Returns the value of attribute api_client.
-
#client ⇒ Object
readonly
Returns the value of attribute client.
Instance Method Summary collapse
- #droplet_exists_in_grid?(grid, droplet) ⇒ Boolean
- #erb(template, vars) ⇒ Object
- #generate_name ⇒ Object
-
#initialize(api_client, token) ⇒ NodeProvisioner
constructor
A new instance of NodeProvisioner.
- #run!(opts) ⇒ Object
- #set_labels(node, labels) ⇒ Object
- #ssh_key(public_key) ⇒ Object
- #user_data(vars) ⇒ Object
Constructor Details
#initialize(api_client, token) ⇒ NodeProvisioner
Returns a new instance of NodeProvisioner.
16 17 18 19 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 16 def initialize(api_client, token) @api_client = api_client @client = DropletKit::Client.new(access_token: token) end |
Instance Attribute Details
#api_client ⇒ Object (readonly)
Returns the value of attribute api_client.
12 13 14 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 12 def api_client @api_client end |
#client ⇒ Object (readonly)
Returns the value of attribute client.
12 13 14 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 12 def client @client end |
Instance Method Details
#droplet_exists_in_grid?(grid, droplet) ⇒ Boolean
75 76 77 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 75 def droplet_exists_in_grid?(grid, droplet) api_client.get("grids/#{grid}/nodes")['nodes'].find{|n| n['name'] == droplet.name} end |
#erb(template, vars) ⇒ Object
79 80 81 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 79 def erb(template, vars) ERB.new(template).result(OpenStruct.new(vars).instance_eval { binding }) end |
#generate_name ⇒ Object
67 68 69 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 67 def generate_name "#{super}-#{rand(1..99)}" end |
#run!(opts) ⇒ Object
21 22 23 24 25 26 27 28 29 30 31 32 33 34 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 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 21 def run!(opts) userdata_vars = { version: opts[:version], master_uri: opts[:master_uri], grid_token: opts[:grid_token], } image = "coreos-#{opts[:channel]}" droplets = [] opts[:count].to_i.times do droplet = DropletKit::Droplet.new( name: opts[:name] || generate_name, region: opts[:region], image: image, size: opts[:size], private_networking: true, user_data: user_data(userdata_vars), ssh_keys: [opts[:ssh_key_id]], tags: [opts[:grid]] ) created = client.droplets.create(droplet) spinner "Creating DigitalOcean droplet #{droplet.name.colorize(:cyan)} " do sleep 1 until client.droplets.find(id: created.id).status == 'active' end droplets << droplet end droplets.each do |droplet| node = nil spinner "Waiting for node #{droplet.name.colorize(:cyan)} join to grid #{opts[:grid].colorize(:cyan)} " do sleep 1 until node = droplet_exists_in_grid?(opts[:grid], droplet) end set_labels( node, [ "region=#{opts[:region]}", "az=#{opts[:region]}", "provider=digitalocean" ] ) end end |
#set_labels(node, labels) ⇒ Object
83 84 85 86 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 83 def set_labels(node, labels) data = {labels: labels} api_client.put("nodes/#{node['id']}", data, {}, {'Kontena-Grid-Token' => node['grid']['token']}) end |
#ssh_key(public_key) ⇒ Object
71 72 73 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 71 def ssh_key(public_key) ssh_key = client.ssh_keys.all.find{|key| key.public_key == public_key} end |
#user_data(vars) ⇒ Object
62 63 64 65 |
# File 'lib/kontena/machine/digital_ocean/node_provisioner.rb', line 62 def user_data(vars) cloudinit_template = File.join(__dir__ , '/cloudinit.yml') erb(File.read(cloudinit_template), vars) end |