golang docker sdk 多协程调用ImagePull方法下载镜像同时显示镜像下载进度

要在Golang中使用Docker SDK进行多协程调用ImagePull方法下载镜像，并同时显示镜像下载进度，可以使用以下代码：

package main

import (
	"context"
	"fmt"
	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client"
	"io"
	"os"
	"sync"
)

func main() {
	// 创建Docker客户端
	cli, err := client.NewClientWithOpts(client.FromEnv)
	if err != nil {
		panic(err)
	}

	// 镜像名称和标签
	imageName := "nginx"
	imageTag := "latest"

	// 创建一个等待组，用于等待所有协程完成
	var wg sync.WaitGroup

	// 获取镜像的总大小
	imageSize, err := getImageSize(cli, imageName, imageTag)
	if err != nil {
		panic(err)
	}

	// 创建一个通道，用于接收下载进度
	progressCh := make(chan int64)

	// 开启一个协程来显示下载进度
	go func() {
		for {
			downloadedSize := <-progressCh
			fmt.Printf("Downloaded: %d/%d\n", downloadedSize, imageSize)
		}
	}()

	// 开启多个协程进行镜像下载
	for i := 0; i < 5; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()

			// 下载镜像
			err := pullImage(cli, imageName, imageTag, progressCh)
			if err != nil {
				fmt.Printf("Error downloading image: %s\n", err.Error())
			}
		}()
	}

	// 等待所有协程完成
	wg.Wait()

	// 关闭通道
	close(progressCh)
}

// 获取镜像的总大小
func getImageSize(cli *client.Client, imageName, imageTag string) (int64, error) {
	imageRef := fmt.Sprintf("%s:%s", imageName, imageTag)

	// 查询镜像信息
	imageInspect, _, err := cli.ImageInspectWithRaw(context.Background(), imageRef)
	if err != nil {
		return 0, err
	}

	// 获取镜像大小
	return imageInspect.Size, nil
}

// 下载镜像
func pullImage(cli *client.Client, imageName, imageTag string, progressCh chan<- int64) error {
	imageRef := fmt.Sprintf("%s:%s", imageName, imageTag)

	// 配置镜像下载选项
	options := types.ImagePullOptions{}

	// 发起镜像下载请求
	response, err := cli.ImagePull(context.Background(), imageRef, options)
	if err != nil {
		return err
	}

	// 读取镜像下载流
	defer response.Close()
	_, err = io.Copy(io.MultiWriter(os.Stdout, progressWriter(progressCh)), response)
	if err != nil && err != io.EOF {
		return err
	}

	return nil
}

// 进度写入器
func progressWriter(progressCh chan<- int64) io.Writer {
	var downloadedSize int64

	return &writerWithProgress{
		progressCh:      progressCh,
		downloadedSize:  &downloadedSize,
	}
}

// 带进度的写入器
type writerWithProgress struct {
	progressCh      chan<- int64
	downloadedSize  *int64
}

func (w *writerWithProgress) Write(p []byte) (n int, err error) {
	n = len(p)
	*w.downloadedSize += int64(n)
	w.progressCh <- *w.downloadedSize
	return
}

此代码使用Docker SDK创建了一个Docker客户端，并使用多个协程并行地下载镜像。通过一个通道来接收下载进度，并在另一个协程中显示下载进度。通过getImageSize函数获取镜像的总大小，并通过pullImage函数下载镜像。progressWriter函数返回一个实现了io.Writer接口的进度写入器，用于将下载进度写入通道。writerWithProgress结构体实现了io.Writer接口的Write方法，用于更新下载进度并将其写入通道。