76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
/*
|
|
Copyright 2019 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package drain
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
corev1 "k8s.io/api/core/v1"
|
|
utilerrors "k8s.io/apimachinery/pkg/util/errors"
|
|
)
|
|
|
|
// This file contains default implementations of how to
|
|
// drain/cordon/uncordon nodes. These functions may be called
|
|
// directly, or their functionality copied into your own code, for
|
|
// example if you want different output behaviour.
|
|
|
|
// RunNodeDrain shows the canonical way to drain a node.
|
|
// You should first cordon the node, e.g. using RunCordonOrUncordon
|
|
func RunNodeDrain(drainer *Helper, nodeName string) error {
|
|
// TODO(justinsb): Ensure we have adequate e2e coverage of this function in library consumers
|
|
list, errs := drainer.GetPodsForDeletion(nodeName)
|
|
if errs != nil {
|
|
return utilerrors.NewAggregate(errs)
|
|
}
|
|
if warnings := list.Warnings(); warnings != "" {
|
|
fmt.Fprintf(drainer.ErrOut, "WARNING: %s\n", warnings)
|
|
}
|
|
|
|
if err := drainer.DeleteOrEvictPods(list.Pods()); err != nil {
|
|
// Maybe warn about non-deleted pods here
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// RunCordonOrUncordon demonstrates the canonical way to cordon or uncordon a Node
|
|
func RunCordonOrUncordon(drainer *Helper, node *corev1.Node, desired bool) error {
|
|
if drainer.Ctx == nil {
|
|
return fmt.Errorf("RunCordonOrUncordon error: drainer.Ctx can't be nil")
|
|
}
|
|
if drainer.Client == nil {
|
|
return fmt.Errorf("RunCordonOrUncordon error: drainer.Client can't be nil")
|
|
}
|
|
// TODO(justinsb): Ensure we have adequate e2e coverage of this function in library consumers
|
|
c := NewCordonHelper(node)
|
|
|
|
if updateRequired := c.UpdateIfRequired(desired); !updateRequired {
|
|
// Already done
|
|
return nil
|
|
}
|
|
|
|
err, patchErr := c.PatchOrReplaceWithContext(drainer.Ctx, drainer.Client, false)
|
|
if err != nil {
|
|
if patchErr != nil {
|
|
return fmt.Errorf("cordon error: %s; merge patch error: %w", err.Error(), patchErr)
|
|
}
|
|
return fmt.Errorf("cordon error: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|