initial version

main.go

@@ -0,0 +1,99 @@
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"log"
+	"os"
+	"os/exec"
+	"strconv"
+	"strings"
+	"sync"
+	"time"
+
+	"github.com/sethgrid/multibar"
+
+	"gopkg.in/yaml.v2"
+)
+
+func poll(cmd string) (int, int) {
+	out, err := exec.Command("sh", "-c", cmd).Output()
+	if err != nil {
+		fmt.Printf("[OUTUT] %s\n", out)
+		log.Fatal(err)
+	}
+
+	parts := strings.Split(string(out), "/")
+	if len(parts) != 2 {
+		log.Fatalf("Command should have returned <actual>/<total> instead it was:", out)
+	}
+
+	act, err := strconv.Atoi(strings.TrimSpace(parts[0]))
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	summ, err := strconv.Atoi(strings.TrimSpace(parts[1]))
+	if err != nil {
+		log.Fatal(err)
+	}
+	return act, summ
+}
+
+func main() {
+	log.Println("poll progress ...")
+	if len(os.Args) < 2 {
+		log.Fatal("yaml file is required")
+	}
+
+	data, err := ioutil.ReadFile(os.Args[1])
+	if err != nil {
+		log.Fatalf("Cloudn't read file: %v", err)
+	}
+	var obj map[string]string
+	err = yaml.Unmarshal([]byte(data), &obj)
+
+	progressBars, _ := multibar.New()
+	progressBars.Println("Azure blob copy progress:")
+
+	wg := &sync.WaitGroup{}
+	wg.Add(len(obj))
+
+	totals := map[string]int{}
+	for task, cmd := range obj {
+		go func(task, cmd string) {
+			_, summ := poll(cmd)
+			totals[task] = summ
+			wg.Done()
+		}(task, cmd)
+	}
+	wg.Wait()
+
+	wg.Add(len(obj))
+	for task, cmd := range obj {
+		p := progressBars.MakeBar(totals[task], fmt.Sprintf("%-30s", task))
+		go func(task, cmd string, progressFn multibar.ProgressFunc) {
+
+			act, summ := poll(cmd)
+			progressFn(act)
+			for act < summ {
+				act, _ = poll(cmd)
+				progressFn(act)
+			}
+
+			wg.Done()
+		}(task, cmd, p)
+	}
+
+	go progressBars.Listen()
+	//time.Sleep(time.Second * 3)
+
+	for _, b := range progressBars.Bars {
+		b.Update(0)
+	}
+
+	wg.Wait()
+	time.Sleep(time.Second * 1)
+
+	fmt.Println("DONE")
+}