feat(middleware/gzip_utils): Add progress bar to Gzip middleware

sikalabs · Oct 2, 2021 · aa006a6 · aa006a6
1 parent 9af949f
commit aa006a6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/backup/middleware/gzip/gzip.go b/backup/middleware/gzip/gzip.go
@@ -2,7 +2,9 @@ package gzip
 
 import (
 	"io"
+	"os"
 
+	"github.com/cheggaaa/pb/v3"
 	"github.com/sikalabs/tergum/utils/gzip_utils"
 )
 
@@ -13,5 +15,17 @@ func (m GzipMiddleware) Validate() error {
 }
 
 func (m GzipMiddleware) Process(data io.ReadSeeker) (io.ReadSeeker, error) {
-	return gzip_utils.GzipIO(data)
+	var err error
+
+	size, _ := data.Seek(0, os.SEEK_END)
+	data.Seek(0, 0)
+
+	bar := pb.Full.Start64(size)
+
+	// create proxy reader
+	barReader := bar.NewProxyReader(data)
+	out, err := gzip_utils.GzipIO(barReader)
+	bar.Finish()
+
+	return out, err
 }
diff --git a/utils/gzip_utils/gzip_utils.go b/utils/gzip_utils/gzip_utils.go
@@ -42,7 +42,7 @@ func GzipBytes(data []byte) ([]byte, error) {
 	return b.Bytes(), nil
 }
 
-func GzipIO(src io.ReadSeeker) (io.ReadSeeker, error) {
+func GzipIO(src io.Reader) (io.ReadSeeker, error) {
 	var err error
 
 	outputFile, err := os.CreateTemp("", "tergum-gzip-")