From 1e745bdea1c67bb1eea37a97c383ce85f176dc8f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Motiejus=20Jak=C5=A1tys?= <motiejus@jakstys.lt>
Date: Mon, 24 May 2021 00:11:58 +0300
Subject: [PATCH] rename discarder to proxyWriter

---
 rootfs/rootfs.go | 27 ++++++++++++---------------
 1 file changed, 12 insertions(+), 15 deletions(-)

diff --git a/rootfs/rootfs.go b/rootfs/rootfs.go
index 4302944..7ae5f46 100644
--- a/rootfs/rootfs.go
+++ b/rootfs/rootfs.go
@@ -8,6 +8,7 @@ import (
 	"errors"
 	"fmt"
 	"io"
+	"io/ioutil"
 	"path/filepath"
 	"strings"
 
@@ -234,27 +235,23 @@ func whiteoutDirs(whreaddir map[string]int, nlayers int) []*tree {
 
 // openTargz creates a tar reader from a targzip or tar
 func openTargz(r io.Reader) (*tar.Reader, func() error) {
-	var hdrbuf bytes.Buffer
-	hdrw := &discarder{w: &hdrbuf}
+	hdrbuf := &bytes.Buffer{}
+	hdrw := &proxyWriter{w: hdrbuf}
 	gz, err := gzip.NewReader(io.TeeReader(r, hdrw))
 	if err == nil {
-		hdrw.w = nil
-		hdrbuf.Reset()
+		hdrw.w = ioutil.Discard
+		hdrbuf = nil
 		return tar.NewReader(gz), gz.Close
 	}
-	return tar.NewReader(io.MultiReader(&hdrbuf, r)), func() error { return nil }
+	return tar.NewReader(io.MultiReader(hdrbuf, r)), func() error { return nil }
 }
 
-// discarder is a pass-through writer until instructed to 'discard' its writes
-// by setting its writer to nil. Useful for proxying writes from a TeeReader
-// until it's known to be unnecessary.
-type discarder struct {
-	w       io.Writer
+// proxyWriter is a pass-through writer. Its underlying writer can be changed
+// on-the-fly.
+type proxyWriter struct {
+	w io.Writer
 }
 
-func (d *discarder) Write(p []byte) (int, error) {
-	if d.w == nil {
-		return len(p), nil
-	}
-	return d.w.Write(p)
+func (pw *proxyWriter) Write(p []byte) (int, error) {
+	return pw.w.Write(p)
 }