X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/fc100474c5d74b20b2576bd3a8f633746c0c6fb2..03395937ba05b9e3192e346a355c691f45cc7c85:/services/api/app/models/node.rb diff --git a/services/api/app/models/node.rb b/services/api/app/models/node.rb index 358e12b934..512f0e0a59 100644 --- a/services/api/app/models/node.rb +++ b/services/api/app/models/node.rb @@ -1,5 +1,5 @@ class Node < ArvadosModel - include AssignUuid + include HasUuid include KindAndEtag include CommonApiTemplate serialize :info, Hash @@ -8,13 +8,7 @@ class Node < ArvadosModel MAX_SLOTS = 64 - @@confdir = if Rails.configuration.respond_to? :dnsmasq_conf_dir - Rails.configuration.dnsmasq_conf_dir - elsif File.exists? '/etc/dnsmasq.d/.' - '/etc/dnsmasq.d' - else - nil - end + @@confdir = Rails.configuration.dnsmasq_conf_dir @@domain = Rails.configuration.compute_node_domain rescue `hostname --domain`.strip @@nameservers = Rails.configuration.compute_node_nameservers @@ -72,7 +66,7 @@ class Node < ArvadosModel if o[:ping_secret] != self.info[:ping_secret] logger.info "Ping: secret mismatch: received \"#{o[:ping_secret]}\" != \"#{self.info[:ping_secret]}\"" - return nil + raise ArvadosModel::UnauthorizedError.new("Incorrect ping_secret") end self.last_ping_at = Time.now @@ -89,7 +83,11 @@ class Node < ArvadosModel if o[:ec2_instance_id] if !self.info[:ec2_instance_id] self.info[:ec2_instance_id] = o[:ec2_instance_id] - `ec2-create-tags #{o[:ec2_instance_id]} --tag 'Name=#{self.uuid}'` + if (Rails.configuration.compute_node_ec2_tag_enable rescue true) + tag_cmd = ("ec2-create-tags #{o[:ec2_instance_id]} " + + "--tag 'Name=#{self.uuid}'") + `#{tag_cmd}` + end elsif self.info[:ec2_instance_id] != o[:ec2_instance_id] logger.debug "Multiple nodes have credentials for #{self.uuid}" raise "#{self.uuid} is already running at #{self.info[:ec2_instance_id]} so rejecting ping from #{o[:ec2_instance_id]}" @@ -111,7 +109,9 @@ class Node < ArvadosModel end while true self.hostname = self.class.hostname_for_slot(self.slot_number) if info[:ec2_instance_id] - `ec2-create-tags #{self.info[:ec2_instance_id]} --tag 'hostname=#{self.hostname}'` + if (Rails.configuration.compute_node_ec2_tag_enable rescue true) + `ec2-create-tags #{self.info[:ec2_instance_id]} --tag 'hostname=#{self.hostname}'` + end end end @@ -119,19 +119,25 @@ class Node < ArvadosModel end def start!(ping_url_method) - ensure_permission_to_update - ping_url = ping_url_method.call({ uuid: self.uuid, ping_secret: self.info[:ping_secret] }) - ec2_args = ["--user-data '#{ping_url}'", - "-t c1.xlarge -n 1", - Rails.configuration.compute_node_ec2run_args, - Rails.configuration.compute_node_ami - ] - ec2run_cmd = ["ec2-run-instances", - "--client-token", self.uuid, - ec2_args].flatten.join(' ') - ec2spot_cmd = ["ec2-request-spot-instances", - "-p #{Rails.configuration.compute_node_spot_bid} --type one-time", - ec2_args].flatten.join(' ') + ensure_permission_to_save + ping_url = ping_url_method.call({ id: self.uuid, ping_secret: self.info[:ping_secret] }) + if (Rails.configuration.compute_node_ec2run_args and + Rails.configuration.compute_node_ami) + ec2_args = ["--user-data '#{ping_url}'", + "-t c1.xlarge -n 1", + Rails.configuration.compute_node_ec2run_args, + Rails.configuration.compute_node_ami + ] + ec2run_cmd = ["ec2-run-instances", + "--client-token", self.uuid, + ec2_args].flatten.join(' ') + ec2spot_cmd = ["ec2-request-spot-instances", + "-p #{Rails.configuration.compute_node_spot_bid} --type one-time", + ec2_args].flatten.join(' ') + else + ec2run_cmd = '' + ec2spot_cmd = '' + end self.info[:ec2_run_command] = ec2run_cmd self.info[:ec2_spot_command] = ec2spot_cmd self.info[:ec2_start_command] = ec2spot_cmd @@ -142,12 +148,16 @@ class Node < ArvadosModel result.match(/INSTANCE\s*(i-[0-9a-f]+)/) do |m| instance_id = m[1] self.info[:ec2_instance_id] = instance_id - `ec2-create-tags #{instance_id} --tag 'Name=#{self.uuid}'` + if (Rails.configuration.compute_node_ec2_tag_enable rescue true) + `ec2-create-tags #{instance_id} --tag 'Name=#{self.uuid}'` + end end result.match(/SPOTINSTANCEREQUEST\s*(sir-[0-9a-f]+)/) do |m| sir_id = m[1] self.info[:ec2_sir_id] = sir_id - `ec2-create-tags #{sir_id} --tag 'Name=#{self.uuid}'` + if (Rails.configuration.compute_node_ec2_tag_enable rescue true) + `ec2-create-tags #{sir_id} --tag 'Name=#{self.uuid}'` + end end self.save! end