open-nomad/command/agent/stats_endpoint.go

package agent

import (
	"fmt"
	"net/http"
	"strconv"

	"github.com/hashicorp/nomad/client"
)

func (s *HTTPServer) StatsRequest(resp http.ResponseWriter, req *http.Request) (interface{}, error) {
	if s.agent.client == nil {
		return nil, clientNotRunning
	}

	var tsRequest bool

	// Check if the user has requested time series
	if qp := req.URL.Query().Get("ts"); qp != "" {
		if tsReq, err := strconv.ParseBool(qp); err == nil {
			tsRequest = tsReq
		}
	}

	clientStats := s.agent.client.StatsReporter()

	// Return the host stats if alloc ID is not present
	var allocID, task string
	if allocID = req.URL.Query().Get("allocation"); allocID == "" {
		return clientStats.HostStats(), nil
	}

	// Check if the allocation is running on the node
	allocStats, ok := clientStats.AllocStats()[allocID]
	if !ok {
		return nil, fmt.Errorf("alloc %q is not running on this client", allocID)
	}

	// Return the resource usage of the task if the task name is specified
	if task = req.URL.Query().Get("task"); task != "" {
		taskStats, err := allocStats.TaskStats(task)
		if err != nil {
			return nil, err
		}
		return s.getStats(tsRequest, taskStats), nil
	}

	// Return the resource usage of all the tasks in an allocation if task name
	// is not specified
	res := make(map[string]interface{})
	for task, taskStats := range allocStats.AllocStats() {
		res[task] = s.getStats(tsRequest, taskStats)
	}
	return res, nil
}

func (s *HTTPServer) getStats(tsRequest bool, taskStats client.TaskStatsReporter) interface{} {
	if tsRequest {
		return taskStats.ResourceUsageTS()
	}
	return taskStats.ResourceUsage()
}