rancher/rke
1
0
Fork 0
mirror of https://github.com/rancher/rke.git synced 2025-05-12 10:26:20 +00:00
Code Issues Releases Wiki Activity

Check drainInput for nil to avoid panic

Browse Source
This commit is contained in:
rajashree 2020-06-16 15:51:54 -07:00
parent f33dc1da6c
commit fb17836063
1 changed files with 5 additions and 1 deletions

6
services/node_util.go
View File

@ -49,10 +49,14 @@ func cordonAndDrainNode(kubeClient *kubernetes.Clientset, host *hosts.Host, drai
}
func getDrainHelper(kubeClient *kubernetes.Clientset, upgradeStrategy v3.NodeUpgradeStrategy) drain.Helper {
var ignoreDaemonSets bool
if upgradeStrategy.DrainInput == nil || *upgradeStrategy.DrainInput.IgnoreDaemonSets {
ignoreDaemonSets = true
}
drainHelper := drain.Helper{
Client: kubeClient,
Force: upgradeStrategy.DrainInput.Force,
IgnoreAllDaemonSets: *upgradeStrategy.DrainInput.IgnoreDaemonSets,
IgnoreAllDaemonSets: ignoreDaemonSets,
DeleteLocalData: upgradeStrategy.DrainInput.DeleteLocalData,
GracePeriodSeconds: upgradeStrategy.DrainInput.GracePeriod,
Timeout: time.Second * time.Duration(upgradeStrategy.DrainInput.Timeout),