Class: Cute::G5K::API
- Inherits:
-
Object
- Object
- Cute::G5K::API
- Defined in:
- lib/cute/g5k_api.rb
Overview
This class helps you to access Grid’5000 REST API. Thus, the most common actions such as reservation of nodes and deployment can be easily scripted. If you are using this module outside Grid’5000 you should created a configuration file with the following contents:
$ cat > ~/.grid5000_api.yml << EOF
uri: https://api.grid5000.fr/
username: user
password: **********
version: stable
EOF
You can take a look at the G5K::API constructor to see more details of this configuration.
Getting started
As already said, the goal of G5K::API class is to present a high level abstraction to manage the most common activities in Grid’5000 such as: the reservation of resources and the deployment of environments. Consequently, these activities can be easily scripted using Ruby. The advantage of this is that you can use all Ruby constructs (e.g., loops, conditionals, blocks, iterators, etc) to script your experiments. In the presence of error, G5K::API raises exceptions (see G5K exceptions), that you can handle to decide the workflow of your experiment (see wait_for_deploy and wait_for_job). In the following example it is shown how G5K::API is used. The example represents the reservation of 3 nodes in Nancy site for 1 hour:
require 'cute'
g5k = Cute::G5K::API.new()
job = g5k.reserve(:nodes => 3, :site => 'nancy', :walltime => '01:00:00')
puts "Assigned nodes : #{job['assigned_nodes']}"
If that is all you want to do, you can write that into a file, let’s say example.rb and execute it using the Ruby interpreter.
$ ruby example.rb
The execution will block until you got the reservation. Then, you can interact with the nodes you reserved in the way you used to or add more code to the previous script for controlling your experiment with Ruby-Cute as shown in this example. We have just used the method reserve that allow us to reserve resources in Grid’5000. This method can be used to reserve resources in deployment mode and deploy our own software environment on them using / Kadeploy. For this we use the option :env of the reserve method. Therefore, it will first reserve the resources and then deploy the specified environment. The method reserve will block until the deployment is done. The following Ruby script illustrates all we have just said.
require 'cute'
g5k = Cute::G5K::API.new()
job = g5k.reserve(:nodes => 1, :site => 'grenoble', :walltime => '00:40:00', :env => 'wheezy-x64-base')
puts "Assigned nodes : #{job['assigned_nodes']}"
By default your public ssh key ‘~/.ssh/id_rsa.pub’ will be copied on the deployed machines, you can specify another path for your keys with the option :keys. In order to deploy your own environment, you have to put the tar file that contains the operating system you want to deploy and the environment description file, under the public directory of a given site. VLANS are supported by adding the parameter :vlan => type where type can be: kavlan, kavlan-global, kavlan-local. The following example, reserves 10 nodes in the Lille site, starts the deployment of a custom environment over the nodes and puts the nodes under a routed VLAN. The different types of VLANs are described in KaVLAN We used the method get_vlan_nodes to get the new hostnames assigned to your nodes.
require 'cute'
g5k = Cute::G5K::API.new()
job = g5k.reserve(:site => "lille", :nodes => 10,
:env => 'http://public.lyon.grid5000.fr/~user/debian_custom_img.yaml',
:vlan => "kavlan", :keys => "~/my_ssh_key")
puts "Log in into the nodes using the following hostnames: #{g5k.get_vlan_nodes(job)}"
You should make sure when using custom environments that in the Kadeploy description file, you use an URL to specify the path to the tarball. Otherwise, you will get the error ‘Invalid client’s export’. If you do not want that the method reserve perform the deployment for you, you have to use the option :type => :deploy. This can be useful when deploying different environments in your reserved nodes, for example: deploying the environments for a small HPC cluster. You have to use the method deploy for performing the deploy. This method do not block by default, that is why you have to use the method wait_for_deploy in order to block the execution until the deployment is done.
require 'cute'
g5k = Cute::G5K::API.new()
job = g5k.reserve(:site => "lyon", :nodes => 5, :walltime => "03:00:00", :type => :deploy)
nodes = job["assigned_nodes"]
slaves = nodes[1..4]
master = nodes-slaves
g5k.deploy(job,:nodes => master, :env => 'http://public.lyon.grid5000.fr/~user/debian_master_img.yaml')
g5k.deploy(job,:nodes => slaves, :env => 'http://public.lyon.grid5000.fr/~user/debian_slaves_img.yaml')
g5k.wait_for_deploy(job)
puts "master node: #{master}"
puts "slaves nodes: #{slaves}"
You can check out the documentation of reserve and deploy methods to know all the parameters supported and more complex uses.
Another useful methods
Let’s use pry to show other useful methods. As shown in Ruby Cute the cute command will open a pry shell with some modules preloaded and it will create the variable $g5k to access G5K::API class. Therefore, we can consult the name of the cluster available in a specific site.
[4] pry(main)> $g5k.cluster_uids("grenoble")
=> ["adonis", "edel", "genepi"]
As well as the deployable environments:
[6] pry(main)> $g5k.environment_uids("grenoble")
=> ["squeeze-x64-base", "squeeze-x64-big", "squeeze-x64-nfs", "wheezy-x64-base", "wheezy-x64-big", "wheezy-x64-min", "wheezy-x64-nfs", "wheezy-x64-xen"]
For getting a list of sites available in Grid’5000 you can use:
[7] pry(main)> $g5k.site_uids()
=> ["grenoble", "lille", "luxembourg", "lyon",...]
We can get the status of nodes in a given site by using:
[8] pry(main)> $g5k.nodes_status("lyon")
=> {"taurus-2.lyon.grid5000.fr"=>"besteffort", "taurus-16.lyon.grid5000.fr"=>"besteffort", "taurus-15.lyon.grid5000.fr"=>"besteffort", ...}
We can get information about our submitted jobs by using:
[11] pry(main)> $g5k.get_my_jobs("grenoble")
=> [{"uid"=>1679094,
"user_uid"=>"cruizsanabria",
"user"=>"cruizsanabria",
"walltime"=>3600,
"queue"=>"default",
"state"=>"running", ...}, ...]
If we are done with our experiment, we can release the submitted job or all jobs in a given site as follows:
[12] pry(main)> $g5k.release(job)
[13] pry(main)> $g5k.release_all("grenoble")
How to debug
There are two ways for debugging: (1) by using the option :debug when initializing the G5K::API object or (2) by enabling the RestClient’s logging:
RESTCLIENT_LOG=stdout cute
RESTCLIENT_LOG=stdout path/to/my/script
Instance Attribute Summary collapse
-
#logger ⇒ Object
Assigns a logger.
Instance Method Summary collapse
-
#api_uri(path) ⇒ String
Returns a valid URI using the current G5K API version.
-
#check_deployment(deploy_info) ⇒ Array
It returns an array of machines that did not deploy successfully = Example It can be used to try a new deploy:.
-
#cluster_uids(site) ⇒ Array
Returns all cluster identifiers.
-
#clusters(site) ⇒ Array
The description of clusters that belong to a given Grid’5000 site.
-
#deploy(job, opts = {}) ⇒ G5KJSON
Deploys an environment in a set of reserved nodes using / Kadeploy.
-
#deploy_status(job, filter = {}) ⇒ Array
Returns the status of all deployments performed within a job.
-
#environment_uids(site) ⇒ Array
Returns the name of the environments deployable in a given site.
-
#environments(site) ⇒ Array
The description of all environments registered in a Grid’5000 site.
-
#g5k_user ⇒ String
Grid’5000 user.
-
#get_deployments(site, uid = nil) ⇒ Hash
The last 50 deployments performed in a Grid’5000 site.
-
#get_job(site, jid) ⇒ Hash
Information concerning a given job submitted in a Grid’5000 site.
-
#get_jobs(site, uid = nil, states = nil) ⇒ Hash
All the jobs submitted in a given Grid’5000 site, if a uid is provided only the jobs owned by the user are shown.
-
#get_metric(site, opts = {}) ⇒ Array
Returns information using the Metrology API.
-
#get_my_jobs(site, states = "running") ⇒ Array
Returns information of all my jobs submitted in a given site.
-
#get_subnets(job) ⇒ Array
Returns an Array with all subnets reserved by a given job.
-
#get_switch(site, name) ⇒ Hash
Information of a specific switch available in a given Grid’5000 site.
-
#get_switches(site) ⇒ Hash
Switches information available in a given Grid’5000 site.
-
#get_vlan_nodes(job) ⇒ Array
All the nodes in the VLAN.
-
#initialize(params = {}) ⇒ API
constructor
Initializes a REST connection for Grid’5000 API.
-
#nodes_status(site) ⇒ Hash
The nodes state (e.g, free, busy, etc) that belong to a given Grid’5000 site.
-
#release(resource) ⇒ Object
Releases a resource, it can be a job or a deploy.
-
#release_all(site) ⇒ Object
Releases all jobs on a site.
-
#reserve(opts) ⇒ G5KJSON
Performs a reservation in Grid’5000.
-
#rest ⇒ Object
It returns the RestClient::Resource object which provides you the get_json and post_json methods.
-
#site ⇒ String
It returns the site name.
-
#site_status(site) ⇒ Hash
All the status information of a given Grid’5000 site.
-
#site_uids ⇒ Array
Returns all sites identifiers.
-
#sites ⇒ Array
The description of all Grid’5000 sites.
-
#wait_for_deploy(job, opts = {}) ⇒ Object
Blocks until deployments have terminated status.
-
#wait_for_job(job, opts = {}) ⇒ Object
Blocks until job is in running state.
Constructor Details
#initialize(params = {}) ⇒ API
Initializes a REST connection for Grid’5000 API
Example
You can specify another configuration file using the option :conf_file, for example:
g5k = Cute::G5K::API.new(:conf_file =>"config file path")
You can specify other parameters to use:
g5k = Cute::G5K::API.new(:uri => "https://api.grid5000.fr", :version => "sid")
If you want to ignore ResquestFailed exceptions you can use:
g5k = Cute::G5K::API.new(:on_error => :ignore)
To activate debugging mode you can use the option :debug:
g5k = Cute::G5K::API.new(:debug => true)
This will provide you with a curl command to try by hand the same request the library is trying to perform. For example:
2016-09-27 11:25:34.039 => CMD debug: curl -kn https://api.grid5000.fr/3.0/sites/nancy/deployments/?user=cruizsanabria
552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 |
# File 'lib/cute/g5k_api.rb', line 552 def initialize(params={}) config = {} default_file = "#{ENV['HOME']}/.grid5000_api.yml" if params[:conf_file].nil? then params[:conf_file] = default_file if File.exist?(default_file) end params[:username] ||= params[:user] params[:password] ||= params[:pass] # backward compatibility config = YAML.load(File.open(params[:conf_file],'r')) unless params[:conf_file].nil? @user = params[:username] || config["username"] @pass = params[:password] || config["password"] @uri = params[:uri] || config["uri"] || "https://api.grid5000.fr/" @api_version = params[:version] || config["version"] || "stable" @logger = nil @debug = params[:debug] || false begin @g5k_connection = G5KRest.new(@uri,@api_version,@user,@pass,params[:on_error]) rescue => e msg_create_file = "" if (not File.exist?(default_file)) && params[:conf_file].nil? then msg_create_file = "Please create the file: ~/.grid5000_api.yml and put the necessary credentials or use the option :conf_file to indicate another file for the credentials" end raise "Unable to authorize against the Grid'5000 API. #{e.message} #{msg_create_file}" end end |
Instance Attribute Details
#logger ⇒ Object
Assigns a logger
Examples
You can use this attribute to control how to log all messages produce by G5K::API. For example, below we use the logger available in Ruby standard library.
require 'cute'
require 'logger'
g5k = Cute::G5K::API.new()
g5k.logger = Logger.new(File.new('experiment_1.log'))
519 520 521 |
# File 'lib/cute/g5k_api.rb', line 519 def logger @logger end |
Instance Method Details
#api_uri(path) ⇒ String
Returns a valid URI using the current G5K API version.
1354 1355 1356 1357 |
# File 'lib/cute/g5k_api.rb', line 1354 def api_uri(path) path = path[1..-1] if path.start_with?('/') return "#{@api_version}/#{path}" end |
#check_deployment(deploy_info) ⇒ Array
It returns an array of machines that did not deploy successfully
Example
It can be used to try a new deploy:
badnodes = g5k.check_deployment(job["deploy"].last)
g5k.deploy(job,:nodes => badnodes, :env => 'wheezy-x64-base')
g5k.wait_for_deploy(job)
1348 1349 1350 |
# File 'lib/cute/g5k_api.rb', line 1348 def check_deployment(deploy_info) deploy_info["result"].select{ |p,v| v["state"] == "KO"}.keys end |
#cluster_uids(site) ⇒ Array
Returns all cluster identifiers
Example:
cluster_uids("grenoble") #=> ["adonis", "edel", "genepi"]
635 636 637 |
# File 'lib/cute/g5k_api.rb', line 635 def cluster_uids(site) return clusters(site).uids end |
#clusters(site) ⇒ Array
Returns the description of clusters that belong to a given Grid’5000 site.
684 685 686 |
# File 'lib/cute/g5k_api.rb', line 684 def clusters(site) @g5k_connection.get_json(api_uri("sites/#{site}/clusters")).items end |
#deploy(job, opts = {}) ⇒ G5KJSON
Deploys an environment in a set of reserved nodes using / Kadeploy. A job structure returned by reserve or get_my_jobs methods is mandatory as a parameter as well as the environment to deploy. By default this method does not block, for that you have to set the option :wait to true.
Examples
Deploying the production environment wheezy-x64-base on all the reserved nodes and wait until the deployment is done:
deploy(job, :env => "wheezy-x64-base", :wait => true)
Other parameters you can specify are :nodes [Array] for deploying on specific nodes within a job and :keys [String] to specify the public key to use during the deployment.
deploy(job, :nodes => ["genepi-2.grid5000.fr"], :env => "wheezy-x64-xen", :keys => "~/my_key")
The parameter :keys [String] can be a string specifying the path of the key (as the previous case) or the contents of the public ssh key as the example given below:
deploy(job,:env => "jessie-x64-big", :keys => File.read("/tmp/test_key/test_key.pub"))
1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 |
# File 'lib/cute/g5k_api.rb', line 1187 def deploy(job, opts = {}) # checking valid options, same as reserve option even though some option dont make any sense valid_opts = [:site, :cluster, :switches, :cpus, :cores, :nodes, :walltime, :cmd, :type, :name, :subnets, :env, :vlan_id, :properties, :resources, :reservation, :wait, :keys, :queue, :env_user] unre_opts = opts.keys - valid_opts raise ArgumentError, "Unrecognized option #{unre_opts}" unless unre_opts.empty? raise ArgumentError, "Unrecognized job format" unless job.is_a?(G5KJSON) env = opts[:env] raise ArgumentError, "Environment must be given" if env.nil? nodes = opts[:nodes].nil? ? job['assigned_nodes'] : opts[:nodes] raise "Unrecognized nodes format, use an Array" unless nodes.is_a?(Array) site = @g5k_connection.follow_parent(job).uid if opts[:keys].nil? then public_key_path = File.("~/.ssh/id_rsa.pub") if File.exist?(public_key_path) then public_key_file = File.read(public_key_path) else raise ArgumentError, "No public ssh key found" end else # We check if the string passed contains the ssh public key if (opts[:keys].length < 300 && (opts[:keys] =~ /^ssh.*/).nil?) public_key_file = File.read("#{File.expand_path(opts[:keys])}.pub").chop else public_key_file = opts[:keys] end end payload = { 'nodes' => nodes, 'environment' => env, 'key' => public_key_file, } payload['vlan'] = opts[:vlan_id] if opts[:vlan_id] payload['user'] = opts[:env_user] unless opts[:env_user].nil? info "Creating deployment" begin info debug_cmd(api_uri("sites/#{site}/deployments"),"POST",payload.to_json), :debug r = @g5k_connection.post_json(api_uri("sites/#{site}/deployments"), payload) rescue Error => e info "Fail to deploy" info e. e.http_body.split("\\n").each{ |line| info line} raise end job["deploy"] = [] if job["deploy"].nil? job["deploy"].push(r) job = wait_for_deploy(job) if opts[:wait] == true return job end |
#deploy_status(job, filter = {}) ⇒ Array
Returns the status of all deployments performed within a job. The results can be filtered using a Hash with valid deployment properties described in Grid’5000 API spec.
Example
deploy_status(job, :nodes => ["adonis-10.grenoble.grid5000.fr"], :status => "terminated")
1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 |
# File 'lib/cute/g5k_api.rb', line 1267 def deploy_status(job,filter = {}) job["deploy"].map!{ |d| d.refresh(@g5k_connection) } filter.keep_if{ |k,v| v} # removes nil values if filter.empty? status = job["deploy"].map{ |d| d["status"] } else status = job["deploy"].map{ |d| d["status"] if filter.select{ |k,v| d[k.to_s] != v }.empty? } end return status.compact end |
#environment_uids(site) ⇒ Array
646 647 648 649 650 651 652 653 654 655 656 |
# File 'lib/cute/g5k_api.rb', line 646 def environment_uids(site) # environments are returned by the API following the format squeeze-x64-big-1.8 # it returns environments without the version environment_uids = environments(site).uids.map{ |e| e_match = /(.*)-(.*)/.match(e) new_name = e_match.nil? ? "" : e_match[1] new_name } return environment_uids.uniq end |
#environments(site) ⇒ Array
Returns the description of all environments registered in a Grid’5000 site.
689 690 691 |
# File 'lib/cute/g5k_api.rb', line 689 def environments(site) @g5k_connection.get_json(api_uri("sites/#{site}/environments")).items end |
#g5k_user ⇒ String
Returns Grid’5000 user.
615 616 617 |
# File 'lib/cute/g5k_api.rb', line 615 def g5k_user return @user.nil? ? ENV['USER'] : @user end |
#get_deployments(site, uid = nil) ⇒ Hash
Returns the last 50 deployments performed in a Grid’5000 site.
717 718 719 720 |
# File 'lib/cute/g5k_api.rb', line 717 def get_deployments(site, uid = nil) info(debug_cmd(api_uri("sites/#{site}/deployments/?user=#{uid}"),"GET"), :debug) @g5k_connection.get_json(api_uri("sites/#{site}/deployments/?user=#{uid}")).items end |
#get_job(site, jid) ⇒ Hash
Returns information concerning a given job submitted in a Grid’5000 site.
725 726 727 728 |
# File 'lib/cute/g5k_api.rb', line 725 def get_job(site, jid) info(debug_cmd(api_uri("/sites/#{site}/jobs/#{jid}"),"GET"), :debug) @g5k_connection.get_json(api_uri("/sites/#{site}/jobs/#{jid}")) end |
#get_jobs(site, uid = nil, states = nil) ⇒ Hash
Returns all the jobs submitted in a given Grid’5000 site, if a uid is provided only the jobs owned by the user are shown.
698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 |
# File 'lib/cute/g5k_api.rb', line 698 def get_jobs(site, uid = nil, states = nil) parameters = [] if states then states = [states] if states.is_a?(String) parameters.push("state=#{states.join(",")}") end parameters.push("user=#{uid}") if uid parameters.push("limit=25") if (states.nil? and uid.nil?) jobs = @g5k_connection.get_json(api_uri("/sites/#{site}/jobs?#{parameters.join("&")}")).items jobs.map{ |j| @g5k_connection.get_json(j.rel_self)} # This request sometime is could take a little long when all jobs are requested # The API return by default 50 the limit was set to 25 (e.g., 23 seconds). end |
#get_metric(site, opts = {}) ⇒ Array
Returns information using the Metrology API.
Example
You can get detailed information of available metrics in a given site:
get_metric("rennes")
If you are only interested in the names of the available metrics:
get_metric("rennes").uids #=> ["cpu_nice", "boottime", "bytes_in", ...]
Then, you can get information about the probes available for a specific metric:
get_metric("rennes",:metric => "network_in")
Finally, you can query on a specific probe:
get_metric("rennes",:metric => "network_in",:query => {:from => 1450374553, :to => 1450374553, :only => "parasilo-11-eth0"})
781 782 783 784 785 786 787 788 789 |
# File 'lib/cute/g5k_api.rb', line 781 def get_metric(site,opts ={}) params = opts[:metric].nil? ? "" : "/#{opts[:metric]}/timeseries" if opts[:query] params+="?" opts[:query].each{ |k,v| params+="#{k}=#{v}&"} end info debug_cmd(api_uri("sites/#{site}/metrics#{params}"),"GET"), :debug @g5k_connection.get_json(api_uri("sites/#{site}/metrics#{params}")).items end |
#get_my_jobs(site, states = "running") ⇒ Array
Returns information of all my jobs submitted in a given site. By default it only shows the jobs in state running. You can specify another state like this:
Examples
get_my_jobs("nancy", "waiting")
Getting several states:
get_my_jobs("nancy", ["waiting","running"])
Valid states are specified in Grid’5000 API spec
803 804 805 806 807 808 809 810 811 |
# File 'lib/cute/g5k_api.rb', line 803 def get_my_jobs(site, states = "running") # raise ArgumentError,"States parameter should be an Array" unless states.is_a?(Array) jobs = get_jobs(site, g5k_user, states) deployments = get_deployments(site, g5k_user) # filtering deployments only the job in state running make sense jobs.map{ |j| j["deploy"] = deployments.select{ |d| d["created_at"] > j["started_at"]} if j["state"] == "running"} return jobs end |
#get_subnets(job) ⇒ Array
Returns an Array with all subnets reserved by a given job. Each element of the Array is a IPAddress::IPv4 object which we can interact with to obtain the details of our reserved subnets:
Example
require 'cute'
g5k = Cute::G5K::API.new()
job = g5k.reserve(:site => "lyon", :resources => "/slash_22=1+{virtual!='none'}/nodes=1")
subnet = g5k.get_subnets(job).first #=> we use 'first' because it is an array and we only reserved one subnet.
ips = subnet.map{ |ip| ip.to_s }
830 831 832 833 834 835 836 837 |
# File 'lib/cute/g5k_api.rb', line 830 def get_subnets(job) if job.resources["subnets"].nil? return nil else subnets = job.resources["subnets"] end subnets.map{|s| IPAddress::IPv4.new s } end |
#get_switch(site, name) ⇒ Hash
Returns information of a specific switch available in a given Grid’5000 site.
753 754 755 756 757 |
# File 'lib/cute/g5k_api.rb', line 753 def get_switch(site, name) s = get_switches(site).detect { |x| x.uid == name } raise "Unknown switch '#{name}'" if s.nil? return s end |
#get_switches(site) ⇒ Hash
Returns switches information available in a given Grid’5000 site.
732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 |
# File 'lib/cute/g5k_api.rb', line 732 def get_switches(site) info(debug_cmd(api_uri("/sites/#{site}/network_equipments"),"GET"), :debug) items = @g5k_connection.get_json(api_uri("/sites/#{site}/network_equipments")).items items = items.select { |x| x['kind'] == 'switch' } # extract nodes connected to those switches items.each { |switch| conns = switch['linecards'].detect { |c| c['kind'] == 'node' } next if conns.nil? # IB switches for example nodes = conns['ports'] \ .select { |x| x != {} } \ .map { |x| x['uid'] } \ .map { |x| "#{x}.#{site}.grid5000.fr"} switch['nodes'] = nodes } return items end |
#get_vlan_nodes(job) ⇒ Array
Returns all the nodes in the VLAN.
841 842 843 844 845 846 847 848 849 850 851 852 |
# File 'lib/cute/g5k_api.rb', line 841 def get_vlan_nodes(job) if job.resources["vlans"].nil? return nil else vlan_id = job.resources["vlans"].first end nodes = job["assigned_nodes"] reg = /^(\w+-\d+)(\..*)$/ nodes.map { |name| reg.match(name)[1]+"-kavlan-"+vlan_id.to_s+reg.match(name)[2] unless reg.match(name).nil? } end |
#nodes_status(site) ⇒ Hash
Returns the nodes state (e.g, free, busy, etc) that belong to a given Grid’5000 site.
667 668 669 670 671 672 673 674 675 |
# File 'lib/cute/g5k_api.rb', line 667 def nodes_status(site) nodes = {} site_status(site).nodes.each do |node| name = node[0] status = node[1]["soft"] nodes[name] = status end return nodes end |
#release(resource) ⇒ Object
Releases a resource, it can be a job or a deploy.
872 873 874 875 876 877 878 879 |
# File 'lib/cute/g5k_api.rb', line 872 def release(resource) raise ArgumentError, "parameter should be a G5KJSON data type" unless resource.is_a?(Cute::G5K::G5KJSON) begin return @g5k_connection.delete_json(resource.rel_self) rescue Cute::G5K::RequestFailed => e raise unless e.response.include?('already killed') end end |
#release_all(site) ⇒ Object
Releases all jobs on a site
856 857 858 859 860 861 862 863 864 865 866 867 868 |
# File 'lib/cute/g5k_api.rb', line 856 def release_all(site) raise ArgumentError, "parameter should be a string" unless site.is_a?(String) Timeout.timeout(20) do jobs = get_my_jobs(site,"running") + get_my_jobs(site,"waiting") break if jobs.empty? begin jobs.each { |j| release(j) } rescue Cute::G5K::RequestFailed => e raise unless e.response.include?('already killed') end end return true end |
#reserve(opts) ⇒ G5KJSON
Performs a reservation in Grid’5000.
Examples
By default this method blocks until the reservation is ready, if we want this method to return after creating the reservation we set the option :wait to false. Then, you can use the method wait_for_job to wait for the reservation.
job = g5k.reserve(:nodes => 25, :site => 'luxembourg', :walltime => '01:00:00', :wait => false)
job = g5k.wait_for_job(job, :wait_time => 100)
Reserving with properties
job = g5k.reserve(:site => 'lyon', :nodes => 2, :properties => "wattmeter='YES'")
job = g5k.reserve(:site => 'nancy', :nodes => 1, :properties => "switch='sgraphene1'")
job = g5k.reserve(:site => 'nancy', :nodes => 1, :properties => "cputype='Intel Xeon E5-2650'")
Subnet reservation
The example below reserves 2 nodes in the cluster chirloute located in Lille for 1 hour as well as 2 /22 subnets. We will get 2048 IP addresses that can be used, for example, in virtual machines. If walltime is not specified, 1 hour walltime will be assigned to the reservation.
job = g5k.reserve(:site => 'lille', :cluster => 'chirloute', :nodes => 2,
:env => 'wheezy-x64-xen', :keys => "path/to/my_ssh_jobkey",
:subnets => [22,2])
Multiple types
The option :type accepts an array for specifying multiple types. For example, if we want to submit a job deploy and destructive, we will type:
job = g5k.reserve(:site => "nancy", :nodes => 1, :walltime => "2:00:00", :type => [:deploy,:destructive])
Before using OAR hierarchy
In order to take advantage of this capability, SSH keys have to be specified at the moment of reserving resources. This has to be used whenever we perform a reservation with cpu and core hierarchy. Given that OAR needs access to both keys private and public users are encouraged to create a pair of SSH keys for managing jobs, for instance the following command can be used:
ssh-keygen -N "" -t rsa -f ~/my_ssh_jobkey
The reserved nodes can be accessed using “oarsh” or by configuring the SSH connection as shown in OAR2. You have to specify different keys per reservation if you want several jobs running at the same time in the same site. Example using the OAR hierarchy:
job = g5k.reserve(:site => "grenoble", :switches => 3, :nodes => 1, :cpus => 1, :cores => 1, :keys => "path/to/my_ssh_jobkey")
Remember that the path passed in the :keys parameter corresponds to the path in the Grid’5000 site frontend where you are submitting the job. This is because OAR needs to access to both keys (private and public) locally. For deployments the path corresponds to the local path from where the script is being executed (See deploy method).
Using OAR syntax
The parameter :resources can be used instead of parameters such as: :cluster, :nodes, :cpus, :walltime, :vlan, :subnets, :properties, etc, which are shortcuts for OAR syntax. These shortcuts are ignored if the the parameter :resources is used. Using the parameter :resources allows to express more flexible and complex reservations by using directly the OAR syntax. Therefore, the two examples shown below are equivalent:
job = g5k.reserve(:site => "grenoble", :switches => 3, :nodes => 1, :cpus => 1, :cores => 1, :keys => "~/my_ssh_jobkey")
job = g5k.reserve(:site => "grenoble", :resources => "/switch=3/nodes=1/cpu=1/core=1", :keys => "~/my_ssh_jobkey")
Combining OAR hierarchy with properties:
job = g5k.reserve(:site => "grenoble", :resources => "{ib10g='YES' and memnode=24160}/cluster=1/nodes=2/core=1", :keys => "~/my_ssh_jobkey")
If we want 2 nodes with the following constraints: 1) nodes on 2 different clusters of the same site, 2) nodes with virtualization capability enabled 3) 1 /22 subnet. The reservation will be like:
job = g5k.reserve(:site => "rennes", :resources => "/slash_22=1+{virtual!='none'}/cluster=2/nodes=1")
Another reservation for two clusters:
job = g5k.reserve(:site => "nancy", :resources => "{cluster='graphene'}/nodes=2+{cluster='griffon'}/nodes=3")
Reservation using a local VLAN
job = g5k.reserve(:site => 'nancy', :resources => "{type='kavlan-local'}/vlan=1,nodes=1", :env => 'wheezy-x64-xen')
982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 |
# File 'lib/cute/g5k_api.rb', line 982 def reserve(opts) # checking valid options valid_opts = [:site, :cluster, :switches, :cpus, :cores, :nodes, :walltime, :cmd, :type, :name, :subnets, :env, :vlan, :num_vlan,:properties, :resources, :reservation, :wait, :keys, :queue, :env_user] unre_opts = opts.keys - valid_opts raise ArgumentError, "Unrecognized option #{unre_opts}" unless unre_opts.empty? nodes = opts.fetch(:nodes, 1) walltime = opts.fetch(:walltime, '01:00:00') site = opts[:site] type = opts.fetch(:type, []) name = opts.fetch(:name, 'rubyCute job') command = opts[:cmd] opts[:wait] = true if opts[:wait].nil? cluster = opts[:cluster] switches = opts[:switches] cpus = opts[:cpus] cores = opts[:cores] subnets = opts[:subnets] properties = opts[:properties] reservation = opts[:reservation] resources = opts.fetch(:resources, "") type = [:deploy] if opts[:env] type = [type] if type.is_a?(Symbol) keys = opts[:keys] queue = opts[:queue] vlan = opts[:vlan] num_vlan = opts.fetch(:num_vlan, 1) available_vlans = nil if opts[:vlan] available_vlans = @g5k_connection.get_json(api_uri("sites/#{site}/vlans")).items.map{ |item| item["type"]}.uniq available_vlans.delete("NULL") end unless opts[:vlan].nil? raise ArgumentError, "VLAN type not available in site #{site}" unless available_vlans.include?(vlan) end raise 'At least nodes, time and site must be given' if [nodes, walltime, site].any? { |x| x.nil? } raise 'nodes should be an integer or a string containing either ALL or BEST' unless (nodes.is_a?(Fixnum) or ["ALL","BEST"].include?(nodes)) secs = walltime.to_secs walltime = walltime.to_time command = "sleep #{secs}" if command.nil? if resources == "" resources = "/switch=#{switches}" unless switches.nil? resources += "/nodes=#{nodes}" resources += "/cpu=#{cpus}" unless cpus.nil? resources += "/core=#{cores}" unless cores.nil? if cluster resources = (cluster.is_a?(Fixnum) ? "/cluster=#{cluster}" : "{cluster='#{cluster}'}") + resources end resources = "{type='#{vlan}'}/vlan=#{num_vlan}+" + resources unless vlan.nil? resources = "slash_#{subnets[0]}=#{subnets[1]}+" + resources unless subnets.nil? end resources += ",walltime=#{walltime}" unless resources.include?("walltime") payload = { 'resources' => resources, 'name' => name, 'command' => command } info "Reserving resources: #{resources} (type: #{type}) (in #{site})" payload['properties'] = properties unless properties.nil? payload['types'] = type.map{ |t| t.to_s} unless type.nil? type.map!{|t| t.to_sym} unless type.nil? payload['queue'] = queue if queue unless type.include?(:deploy) if opts[:keys] payload['import-job-key-from-file'] = [ File.(keys) ] end end if reservation payload['reservation'] = reservation info "Starting this reservation at #{reservation}" end begin info debug_cmd(api_uri("sites/#{site}/jobs"),"POST",payload.to_json), :debug r = @g5k_connection.post_json(api_uri("sites/#{site}/jobs"),payload) # This makes reference to the same class rescue Error => e info "Fail to submit job" info e. e.http_body.split("\\n").each{ |line| info line} raise end job = @g5k_connection.get_json(r.rel_self) job = wait_for_job(job) if opts[:wait] == true opts.delete(:nodes) # to not collapse with deploy options opts.delete(:vlan) opts.delete(:num_vlan) if opts[:env] if opts[:vlan] vlan_id = job.resources["vlans"].first deploy(job,opts.merge!({:vlan_id => vlan_id})) else deploy(job,opts) #type == :deploy end end return job end |
#rest ⇒ Object
It returns the RestClient::Resource object which provides you the get_json and post_json methods. This enables to perform low level REST requests. This method is intended to be used along with the api_uri method for generating valid URI.
Example:
require 'cute'
g5k = Cute::G5K::API.new()
g5k.rest.get_json(g5k.api_uri("/sites/grenoble/clusters")
610 611 612 |
# File 'lib/cute/g5k_api.rb', line 610 def rest @g5k_connection end |
#site ⇒ String
It returns the site name. Example:
site #=> "rennes"
This will only work when G5K::API is used within Grid’5000. In the other cases it will return nil
594 595 596 597 598 |
# File 'lib/cute/g5k_api.rb', line 594 def site p = `hostname`.chop res = /^.*\.(.*).*\.grid5000.fr/.match(p) res[1] unless res.nil? end |
#site_status(site) ⇒ Hash
Returns all the status information of a given Grid’5000 site.
660 661 662 663 |
# File 'lib/cute/g5k_api.rb', line 660 def site_status(site) info(debug_cmd(api_uri("sites/#{site}/status"),"GET"), :debug) @g5k_connection.get_json(api_uri("sites/#{site}/status")) end |
#site_uids ⇒ Array
Returns all sites identifiers
Example:
site_uids #=> ["grenoble", "lille", "luxembourg", "lyon",...]
625 626 627 |
# File 'lib/cute/g5k_api.rb', line 625 def site_uids return sites.uids end |
#sites ⇒ Array
Returns the description of all Grid’5000 sites.
678 679 680 |
# File 'lib/cute/g5k_api.rb', line 678 def sites @g5k_connection.get_json(api_uri("sites")).items end |
#wait_for_deploy(job, opts = {}) ⇒ Object
Blocks until deployments have terminated status
Examples
This method requires a job as a parameter and it will blocks by default until all deployments within the job pass form processing status to terminated status.
wait_for_deploy(job)
You can wait for specific deployments using the option :nodes. This can be useful when performing different deployments on the reserved resources.
wait_for_deploy(job, :nodes => ["adonis-10.grenoble.grid5000.fr"])
Another parameter you can specify is :wait_time that allows you to timeout the deployment (by default is 10h). The method will throw a Timeout exception that you can catch and react upon. This example illustrates how this can be used.
require 'cute'
g5k = Cute::G5K::API.new()
job = g5k.reserve(:nodes => 1, :site => 'lyon', :type => :deploy)
begin
g5k.deploy(job,:env => 'wheezy-x64-base')
g5k.wait_for_deploy(job,:wait_time => 100)
rescue Cute::G5K::EventTimeout
puts "We waited too long let's release the job"
g5k.release(job)
end
1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 |
# File 'lib/cute/g5k_api.rb', line 1313 def wait_for_deploy(job,opts = {}) raise "Deploy information not present in the given job" if job["deploy"].nil? opts.merge!({:wait_time => 36000}) if opts[:wait_time].nil? nodes = opts[:nodes] begin Timeout.timeout(opts[:wait_time]) do # it will ask just for processing status status = deploy_status(job,{:nodes => nodes, :status => "processing"}) until status.empty? do info "Waiting for #{status.length} deployment" sleep 4 status = deploy_status(job,{:nodes => nodes, :status => "processing"}) end info "Deployment finished" return job end rescue Timeout::Error raise EventTimeout.new("Timeout triggered") end end |
#wait_for_job(job, opts = {}) ⇒ Object
Blocks until job is in running state
Example
You can pass the parameter :wait_time that allows you to timeout the submission (by default is 10h). The method will throw a Timeout exception that you can catch and react upon. The following example shows how can be used, let’s suppose we want to find 5 nodes available for 3 hours. We can try in each site using the script below.
require 'cute'
g5k = Cute::G5K::API.new()
sites = g5k.site_uids
sites.each{ |site|
job = g5k.reserve(:site => site, :nodes => 5, :wait => false, :walltime => "03:00:00")
begin
job = g5k.wait_for_job(job, :wait_time => 60)
puts "Nodes assigned #{job['assigned_nodes']}"
break
rescue Cute::G5K::EventTimeout
puts "We waited too long in site #{site} let's release the job and try in another site"
g5k.release(job)
end
}
1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 |
# File 'lib/cute/g5k_api.rb', line 1131 def wait_for_job(job,opts={}) opts[:wait_time] = 36000 if opts[:wait_time].nil? jid = job['uid'] info "Waiting for reservation #{jid}" begin Timeout.timeout(opts[:wait_time]) do while true job = job.refresh(@g5k_connection) t = job['scheduled_at'] if !t.nil? t = Time.at(t) secs = [ t - Time.now, 0 ].max.to_i info "Reservation #{jid} should be available at #{t} (#{secs} s)" end break if job['state'] == 'running' raise "Job is finishing." if job['state'] == 'finishing' Kernel.sleep(5) end end rescue Timeout::Error raise EventTimeout.new("Event timeout") end info "Reservation #{jid} ready" return job end |