From 1de2f3cc7dd83428f866888a224bfae86ea3ab0b Mon Sep 17 00:00:00 2001
From: Dingzhu Lurong <1015542478@qq.com>
Date: Fri, 29 Oct 2021 11:49:31 +0800
Subject: [PATCH] add writer error handler

---
 pkg/kubelet/prober/prober.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pkg/kubelet/prober/prober.go b/pkg/kubelet/prober/prober.go
index f766c132d28..53f71d9cb5a 100644
--- a/pkg/kubelet/prober/prober.go
+++ b/pkg/kubelet/prober/prober.go
@@ -297,7 +297,10 @@ func (eic *execInContainer) Stop() {
 func (eic *execInContainer) Start() error {
 	data, err := eic.run()
 	if eic.writer != nil {
-		eic.writer.Write(data)
+		// only record the write error, do not cover the command run error
+		if p, err := eic.writer.Write(data); err != nil {
+			klog.ErrorS(err, "Unable to write all bytes from execInContainer", "expectedBytes", len(data), "actualBytes", p)
+		}
 	}
 	return err
 }