func parseNodeSelectorWithPrompt(
oldSelector klabels.Selector,
newSelectorString string,
applicator labels.Applicator,
) (klabels.Selector, error) {
newSelector, err := parseNodeSelector(newSelectorString)
if err != nil {
return newSelector, err
}
if oldSelector.String() == newSelector.String() {
return newSelector, nil
}
newNodeLabels, err := applicator.GetMatches(newSelector, labels.NODE, false)
if err != nil {
return newSelector, util.Errorf("Error getting matching labels: %v", err)
}
oldNodeLabels, err := applicator.GetMatches(oldSelector, labels.NODE, false)
if err != nil {
return newSelector, util.Errorf("Error getting matching labels: %v", err)
}
toRemove, toAdd := makeNodeChanges(oldNodeLabels, newNodeLabels)
fmt.Printf("Changing deployment from '%v' to '%v':\n", oldSelector.String(), newSelectorString)
fmt.Printf("Removing:%9s hosts %s\n", fmt.Sprintf("-%v", len(toRemove)), toRemove)
fmt.Printf("Adding: %9s hosts %s\n", fmt.Sprintf("+%v", len(toAdd)), toAdd)
fmt.Println("Continue?")
if !confirm() {
return newSelector, util.Errorf("User cancelled")
}
return newSelector, nil
}
func confirmMinheathForSelector(minHealth int, selector klabels.Selector, applicator labels.Applicator) error {
matches, err := applicator.GetMatches(selector, labels.NODE, false)
if err != nil {
return err
}
if len(matches) < minHealth {
fmt.Printf("Your selector matches %d nodes but your minhealth is set to only %d, this daemon set will not replicate. Continue?\n", len(matches), minHealth)
if !confirm() {
return util.Errorf("User cancelled")
}
}
return nil
}
func scheduledPods(t *testing.T, pods labels.Applicator) []labels.Labeled {
podSelector := klabels.Everything().Add("podTest", klabels.EqualsOperator, []string{"successful"})
labeled, err := pods.GetMatches(podSelector, labels.POD)
Assert(t).IsNil(err, "expected no error matching pods")
return labeled
}