mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-04 02:30:45 +00:00
3dd59c8b9b
NewNodeMetadata -> MakeNodeMetadata. It doesn't return a pointer, so Make is more idiomatic. Invoke MakeNodeMetadata when necessary. The zero value for a NodeMetadata is no longer valid. Split MakeNodeMetadata to two constructors. MakeNodeMetadata when you don't have anything to prepopulate; MakeNodeMetadataWith when you do. Also, a fix to the tests in app. We unmarshal a RenderableNode struct, which has a JSON-ignored NodeMetadata field. The zero value is invalid, so we need to fix that before performing comparisons.
28 lines
860 B
Go
28 lines
860 B
Go
package host
|
|
|
|
import (
|
|
"github.com/weaveworks/scope/report"
|
|
)
|
|
|
|
// Tagger tags each node in each topology of a report with the origin host
|
|
// node ID of this (probe) host. Effectively, a foreign key linking every node
|
|
// in every topology to an origin host node in the host topology.
|
|
type Tagger struct{ hostNodeID string }
|
|
|
|
// NewTagger tags each node with a foreign key linking it to its origin host
|
|
// in the host topology.
|
|
func NewTagger(hostID string) Tagger {
|
|
return Tagger{hostNodeID: report.MakeHostNodeID(hostID)}
|
|
}
|
|
|
|
// Tag implements Tagger.
|
|
func (t Tagger) Tag(r report.Report) (report.Report, error) {
|
|
md := report.MakeNodeMetadataWith(map[string]string{report.HostNodeID: t.hostNodeID})
|
|
for _, topology := range r.Topologies() {
|
|
for nodeID := range topology.NodeMetadatas {
|
|
topology.NodeMetadatas[nodeID].Merge(md)
|
|
}
|
|
}
|
|
return r, nil
|
|
}
|