services/api/app/controllers/arvados/v1/nodes_controller.rb

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: AGPL-3.0
   4
   5 class Arvados::V1::NodesController < ApplicationController
   6   skip_before_filter :require_auth_scope, :only => :ping
   7   skip_before_filter :find_object_by_uuid, :only => :ping
   8   skip_before_filter :render_404_if_no_object, :only => :ping
   9
  10   include DbCurrentTime
  11
  12   def update
  13     if resource_attrs[:job_uuid].is_a? String
  14       @object.job_readable = readable_job_uuids([resource_attrs[:job_uuid]]).any?
  15     end
  16     super
  17   end
  18
  19   def self._ping_requires_parameters
  20     { ping_secret: {required: true} }
  21   end
  22
  23   def self._create_requires_parameters
  24     super.merge(
  25       { assign_slot: {required: false, type: 'boolean', description: 'assign slot and hostname'} })
  26   end
  27
  28   def self._update_requires_parameters
  29     super.merge(
  30       { assign_slot: {required: false, type: 'boolean', description: 'assign slot and hostname'} })
  31   end
  32
  33   def create
  34     @object = model_class.new(resource_attrs)
  35     @object.assign_slot if params[:assign_slot]
  36     @object.save!
  37     show
  38   end
  39
  40   def update
  41     attrs_to_update = resource_attrs.reject { |k,v|
  42       [:kind, :etag, :href].index k
  43     }
  44     @object.update_attributes!(attrs_to_update)
  45     @object.assign_slot if params[:assign_slot]
  46     show
  47   end
  48
  49   def ping
  50     act_as_system_user do
  51       @object = Node.where(uuid: (params[:id] || params[:uuid])).first
  52       if !@object
  53         return render_not_found
  54       end
  55       ping_data = {
  56         ip: params[:local_ipv4] || request.remote_ip,
  57         ec2_instance_id: params[:instance_id]
  58       }
  59       [:ping_secret, :total_cpu_cores, :total_ram_mb, :total_scratch_mb]
  60         .each do |key|
  61         ping_data[key] = params[key] if params[key]
  62       end
  63       @object.ping(ping_data)
  64       if @object.info['ping_secret'] == params[:ping_secret]
  65         send_json @object.as_api_response(:superuser)
  66       else
  67         raise "Invalid ping_secret after ping"
  68       end
  69     end
  70   end
  71
  72   def find_objects_for_index
  73     if !current_user.andand.is_admin && current_user.andand.is_active
  74       # active non-admin users can list nodes that are (or were
  75       # recently) working
  76       @objects = model_class.where('last_ping_at >= ?', db_current_time - 1.hours)
  77     end
  78     super
  79     if @select.nil? or @select.include? 'job_uuid'
  80       job_uuids = @objects.map { |n| n[:job_uuid] }.compact
  81       assoc_jobs = readable_job_uuids(job_uuids)
  82       @objects.each do |node|
  83         node.job_readable = assoc_jobs.include?(node[:job_uuid])
  84       end
  85     end
  86   end
  87
  88   protected
  89
  90   def readable_job_uuids(uuids)
  91     Job.readable_by(*@read_users).select(:uuid).where(uuid: uuids).map(&:uuid)
  92   end
  93 end