feat: support to print the progress when oci download

Author: LinuxSuRen

pkg/downloader/oci.go

@@ -169,8 +169,11 @@ func (d *defaultOCIDownloader) getLatestTag(image, authToken string) (tag string
 	} else {
 		defer resp.Body.Close()
 
+		progressReader := NewProgressReader(resp.Body)
+		progressReader.SetLength(resp.ContentLength)
+
 		var data []byte
-		if data, err = io.ReadAll(resp.Body); err != nil {
+		if data, err = io.ReadAll(progressReader); err != nil {
 			return
 		}
 

pkg/downloader/progress.go

@@ -0,0 +1,51 @@
+package downloader
+
+import (
+	"fmt"
+	"io"
+	"os"
+)
+
+type ProgressReader interface {
+	io.Reader
+	Withoutput(io.Writer)
+	SetLength(int64)
+}
+
+type defaultProgressReader struct {
+	reader  io.Reader
+	w       io.Writer
+	total   int64
+	current int
+}
+
+var _ io.Reader = (*defaultProgressReader)(nil)
+
+func NewProgressReader(r io.Reader) ProgressReader {
+	return &defaultProgressReader{
+		reader: r,
+		w:      os.Stdout,
+	}
+}
+
+func (r *defaultProgressReader) Read(p []byte) (count int, err error) {
+	count, err = r.reader.Read(p)
+	if r.total > 0 {
+		if count > 0 {
+			r.current += count
+			fmt.Println(count, "==", r.total, "==", r.current)
+			fmt.Fprintf(r.w, "%d\n", int64(r.current*100)/r.total)
+		}
+	} else {
+		fmt.Fprintf(r.w, "%d\n", count)
+	}
+	return
+}
+
+func (r *defaultProgressReader) Withoutput(w io.Writer) {
+	r.w = w
+}
+
+func (r *defaultProgressReader) SetLength(len int64) {
+	r.total = len
+}

pkg/server/store_ext_manager.go

@@ -18,7 +18,6 @@ package server
 import (
 	"context"
 	"errors"
-	"github.com/linuxsuren/api-testing/pkg/util/home"
 	"io"
 	"net/http"
 	"os"
@@ -27,6 +26,8 @@ import (
 	"syscall"
 	"time"
 
+	"github.com/linuxsuren/api-testing/pkg/util/home"
+
 	"github.com/linuxsuren/api-testing/pkg/downloader"
 	"github.com/linuxsuren/api-testing/pkg/logging"
 
@@ -124,6 +125,8 @@ func (s *storeExtManager) Start(name, socket string) (err error) {
 
 func (s *storeExtManager) startPlugin(socketURL, plugin, pluginName string) (err error) {
 	socketFile := strings.TrimPrefix(socketURL, s.socketPrefix)
+	_ = os.RemoveAll(socketFile) // always deleting the socket file to avoid start failing
+
 	s.filesNeedToBeRemoved = append(s.filesNeedToBeRemoved, socketFile)
 	s.extStatusMap[pluginName] = true
 	if err = s.execer.RunCommandWithIO(plugin, "", os.Stdout, os.Stderr, s.processChan, "--socket", socketFile); err != nil {