Skip to content

Commit

Permalink
Add draining logs
Browse files Browse the repository at this point in the history
  • Loading branch information
TwiN committed May 7, 2020
1 parent d000e72 commit 6fcaad5
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions k8s/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import (
"k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
"log"
"time"
)

Expand Down Expand Up @@ -84,7 +85,20 @@ func (k *KubernetesClient) Drain(nodeName string, ignoreDaemonSets, deleteLocalD
IgnoreDaemonsets: ignoreDaemonSets,
GracePeriodSeconds: -1,
Force: true,
Logger: &drainLogger{NodeName: nodeName},
DeleteLocalData: deleteLocalData,
Timeout: 3 * time.Minute,
})
}

type drainLogger struct {
NodeName string
}

func (l *drainLogger) Log(v ...interface{}) {
log.Println(fmt.Sprintf("[%s][DRAINER]", l.NodeName), fmt.Sprint(v...))
}

func (l *drainLogger) Logf(format string, v ...interface{}) {
log.Println(fmt.Sprintf("[%s][DRAINER]", l.NodeName), fmt.Sprintf(format, v...))
}

0 comments on commit 6fcaad5

Please sign in to comment.