golang并发执行linux命令

wait.group并发执行linux命令

package main

import (
	"fmt"
	"os/exec"
	"sync"
	"time"
)

var imageURLs = []string{
	"myharbor.com/library/test1:v1",
	"myharbor.com/library/py36:ai",
	"myharbor.com/library/test:vl",
	"myharbor.com/library/1og:4.3.0",
}

type Message struct {
	image    string
	err      error
	out      string
	costTime float64
}

func RMI() {
	for _, imgURL := range imageURLs {
		cmd := exec.Command("docker", "rmi", imgURL)
		_, err2 := cmd.Output()
		if err2 != nil {
			fmt.Println("docker rmi错误:", err2)
		}
	}
}

func ExecPull() {
	Ch := make(chan Message, len(imageURLs))
	wg := sync.WaitGroup{}
	wg.Add(len(imageURLs))

	for _, imgURL := range imageURLs {
		go func(imgurL string) {
			defer wg.Done()
			msg0bj := Message{
				image: imgURL,
			}

			sTime := time.Now()
			cmd := exec.Command("docker", "pull", imgurL)
			out2, err2 := cmd.Output()
			if err2 != nil {
				msg0bj.err = err2
				fmt.Printf("docker pull error,%s:%+v \n", imgurL, err2)
			}
			eTime := time.Since(sTime).Seconds()
			msg0bj.out = string(out2)
			msg0bj.costTime = eTime
			Ch <- msg0bj
		}(imgURL)
	}
	wg.Wait()
	close(Ch)
	for item := range Ch {
		if item.err != nil {
			fmt.Printf("%s,success!,耗时:%f秒 \n", item.image, item.costTime)
		} else {
			fmt.Printf("%s,failed!,耗时:%f秒,输出:%s \n", item.image, item.costTime, item.out)
			RMI()
		}
	}
}

func main() {
	RMI()
	for i := 0; i < 10; i++ {
		ExecPull()
	}
}