otto-de/marathon-api

# This class represents a Marathon Task.
# See https://mesosphere.github.io/marathon/docs/rest-api.html#get-/v2/tasks for full list of API's methods.
class Marathon::Task < Marathon::Base

  ACCESSORS = %w[ id appId host ports servicePorts version stagedAt startedAt ]

  # Create a new task object.
  # ++hash++: Hash including all attributes
  # ++marathon_instance++: MarathonInstance holding a connection to marathon
  def initialize(hash, marathon_instance = Marathon.singleton)
    super(hash, ACCESSORS)
    @marathon_instance = marathon_instance
  end

  # Kill the task that belongs to an application.
  # ++scale++: Scale the app down (i.e. decrement its instances setting by the number of tasks killed)
  #            after killing the specified tasks.
  def delete!(scale = false)
    new_task = self.class.delete(id, scale)
  end

  alias :kill! :delete!

  def to_s
    "Marathon::Task { :id => #{self.id} :appId => #{appId} :host => #{host} }"
  end

  # Returns a string for listing the task.
  def to_pretty_s
    %Q[
Task ID:    #{id}
App ID:     #{appId}
Host:       #{host}
Ports:      #{(ports || []).join(',')}
Staged at:  #{stagedAt}
Started at: #{startedAt}
Version:    #{version}
    ].strip
  end

  class << self

    # List tasks of all applications.
    # ++status++: Return only those tasks whose status matches this parameter.
    #             If not specified, all tasks are returned. Possible values: running, staging.
    def list(status = nil)
      Marathon.singleton.tasks.list(status)
    end

    # List all running tasks for application appId.
    # ++appId++: Application's id
    def get(appId)
      Marathon.singleton.tasks.get(appId)
    end

    # Kill the given list of tasks and scale apps if requested.
    # ++ids++: Id or list of ids with target tasks.
    # ++scale++: Scale the app down (i.e. decrement its instances setting by the number of tasks killed)
    #            after killing the specified tasks.
    def delete(ids, scale = false)
      Marathon.singleton.tasks.delete(ids, scale)
    end

    alias :remove :delete
    alias :kill :delete

    # Kill tasks that belong to the application appId.
    # ++appId++: Application's id
    # ++host++: Kill only those tasks running on host host.
    # ++scale++: Scale the app down (i.e. decrement its instances setting by the number of tasks killed)
    #            after killing the specified tasks.
    def delete_all(appId, host = nil, scale = false)
      Marathon.singleton.tasks.delete_all(appId, host, scale)
    end

    alias :remove_all :delete_all
    alias :kill_all :delete_all
  end
end

# This class represents a set of Tasks
class Marathon::Tasks
  def initialize(marathon_instance)
    @marathon_instance = marathon_instance
    @connection = marathon_instance.connection
  end

  # List tasks of all applications.
  # ++status++: Return only those tasks whose status matches this parameter.
  #             If not specified, all tasks are returned. Possible values: running, staging.
  def list(status = nil)
    query = {}
    Marathon::Util.add_choice(query, :status, status, %w[running staging])
    json = @connection.get('/v2/tasks', query)['tasks']
    json.map { |j| Marathon::Task.new(j, @marathon_instance) }
  end

  # List all running tasks for application appId.
  # ++appId++: Application's id
  def get(appId)
    json = @connection.get("/v2/apps/#{appId}/tasks")['tasks']
    json.map { |j| Marathon::Task.new(j, @marathon_instance) }
  end

  # Kill the given list of tasks and scale apps if requested.
  # ++ids++: Id or list of ids with target tasks.
  # ++scale++: Scale the app down (i.e. decrement its instances setting by the number of tasks killed)
  #            after killing the specified tasks.
  def delete(ids, scale = false)
    query = {}
    query[:scale] = true if scale
    ids = [ids] if ids.is_a?(String)
    @connection.post("/v2/tasks/delete", query, :body => {:ids => ids})
  end

  # Kill tasks that belong to the application appId.
  # ++appId++: Application's id
  # ++host++: Kill only those tasks running on host host.
  # ++scale++: Scale the app down (i.e. decrement its instances setting by the number of tasks killed)
  #            after killing the specified tasks.
  def delete_all(appId, host = nil, scale = false)
    query = {}
    query[:host] = host if host
    query[:scale] = true if scale
    json = @connection.delete("/v2/apps/#{appId}/tasks", query)['tasks']
    json.map { |j| Marathon::Task.new(j, @marathon_instance) }
  end

end