mirror of
https://github.com/weaveworks/scope.git
synced 2026-07-16 20:11:09 +00:00
9e092f1a4a
This allows plugins to add controls to nodes that already have some controls set by other plugin. Previously only the last plugin that sets the controls in the node would have its controls visible. That was because of NodeControls' Merge function that actually weren't merging data from two inputs, but rather returning data that was newer and discarding the older one.
64 lines
1.6 KiB
Go
64 lines
1.6 KiB
Go
package kubernetes
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/weaveworks/scope/report"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
|
"k8s.io/kubernetes/pkg/apis/extensions"
|
|
"k8s.io/kubernetes/pkg/labels"
|
|
)
|
|
|
|
// These constants are keys used in node metadata
|
|
const (
|
|
FullyLabeledReplicas = "kubernetes_fully_labeled_replicas"
|
|
)
|
|
|
|
// ReplicaSet represents a Kubernetes replica set
|
|
type ReplicaSet interface {
|
|
Meta
|
|
Selector() labels.Selector
|
|
AddParent(topology, id string)
|
|
GetNode(probeID string) report.Node
|
|
}
|
|
|
|
type replicaSet struct {
|
|
*extensions.ReplicaSet
|
|
Meta
|
|
parents report.Sets
|
|
Node *api.Node
|
|
}
|
|
|
|
// NewReplicaSet creates a new ReplicaSet
|
|
func NewReplicaSet(r *extensions.ReplicaSet) ReplicaSet {
|
|
return &replicaSet{
|
|
ReplicaSet: r,
|
|
Meta: meta{r.ObjectMeta},
|
|
parents: report.MakeSets(),
|
|
}
|
|
}
|
|
|
|
func (r *replicaSet) Selector() labels.Selector {
|
|
selector, err := unversioned.LabelSelectorAsSelector(r.Spec.Selector)
|
|
if err != nil {
|
|
// TODO(paulbellamy): Remove the panic!
|
|
panic(err)
|
|
}
|
|
return selector
|
|
}
|
|
|
|
func (r *replicaSet) AddParent(topology, id string) {
|
|
r.parents = r.parents.Add(topology, report.MakeStringSet(id))
|
|
}
|
|
|
|
func (r *replicaSet) GetNode(probeID string) report.Node {
|
|
return r.MetaNode(report.MakeReplicaSetNodeID(r.UID())).WithLatests(map[string]string{
|
|
ObservedGeneration: fmt.Sprint(r.Status.ObservedGeneration),
|
|
Replicas: fmt.Sprint(r.Status.Replicas),
|
|
DesiredReplicas: fmt.Sprint(r.Spec.Replicas),
|
|
FullyLabeledReplicas: fmt.Sprint(r.Status.FullyLabeledReplicas),
|
|
report.ControlProbeID: probeID,
|
|
}).WithParents(r.parents).WithLatestActiveControls(ScaleUp, ScaleDown)
|
|
}
|