weave-scope/app/origin_host.go

package main

import (
	"net/http"
	"strings"

	"github.com/gorilla/mux"

	"github.com/weaveworks/scope/probe/host"
	"github.com/weaveworks/scope/report"
)

// OriginHost represents a host that runs a probe, i.e. the origin host of
// some data in the system. The struct is returned by the /api/origin/{id}
// handler.
type OriginHost struct {
	Hostname string   `json:"hostname"`
	OS       string   `json:"os"`
	Networks []string `json:"networks"`
	Load     string   `json:"load"`
}

func getOriginHost(t report.Topology, nodeID string) (OriginHost, bool) {
	h, ok := t.NodeMetadatas[nodeID]
	if !ok {
		return OriginHost{}, false
	}

	return OriginHost{
		Hostname: h[host.HostName],
		OS:       h[host.OS],
		Networks: strings.Split(h[host.LocalNetworks], " "),
		Load:     h[host.Load],
	}, true
}

// makeOriginHostHandler makes the /api/origin/* handler.
func makeOriginHostHandler(rep Reporter) http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		var (
			vars   = mux.Vars(r)
			nodeID = vars["id"]
		)
		origin, ok := getOriginHost(rep.Report().Host, nodeID)
		if !ok {
			http.NotFound(w, r)
			return
		}
		respondWith(w, http.StatusOK, origin)
	}
}