Golang - Docker API - ImagePull 的解析结果

Question

我正在开发一个Go 脚本，该脚本使用Docker API 用于我的项目。登录到我的存储库后，我提取了我想要的 Docker 映像，但问题是 ImagePull 函数返回了一个 io.ReadCloser 的实例，我只能通过以下方式将其传递给系统输出：

io.Copy(os.Stdout, pullResp)

我可以看到响应很酷，但是我找不到一种体面的方法来解析它并根据它实现一个逻辑，如果下载了新版本的图像，它会做一些事情，以及其他如果图像是最新的。
如果您分享您的经验，如果您曾经遇到过这个问题，我会很高兴。

Answer 1

您可以导入github.com/docker/docker/pkg/jsonmessage 并同时使用JSONMessage 和JSONProgress 来解码流，但更容易调用 DisplayJSONMessagesToStream：它既解析流又将消息显示为文本。以下是使用 stderr 显示消息的方法：

    reader, err := cli.ImagePull(ctx, myImageRef, types.ImagePullOptions{})
    if err != nil {
            return err
    }
    defer reader.Close()

    termFd, isTerm := term.GetFdInfo(os.Stderr)
    jsonmessage.DisplayJSONMessagesStream(reader, os.Stderr, termFd, isTerm, nil)

好消息是它适应输出：如果这是一个 TTY（docker pull 的方式），它会更新行，但如果输出被重定向到文件，它不会。

Answer 2

@radoslav-stoyanov 在使用我的示例之前做

# docker rmi busybox

然后运行代码

package main

import (
    "encoding/json"
    "fmt"
    "github.com/docker/distribution/context"
    docker "github.com/docker/engine-api/client"
    "github.com/docker/engine-api/types"
    "io"
    "strings"
)

func main() {
    // DOCKER
    cli, err := docker.NewClient("unix:///var/run/docker.sock", "v1.28", nil, map[string]string{"User-Agent": "engine-api-cli-1.0"})
    if err != nil {
        panic(err)
    }

    imageName := "busybox:latest"

    events, err := cli.ImagePull(context.Background(), imageName, types.ImagePullOptions{})
    if err != nil {
        panic(err)
    }

    d := json.NewDecoder(events)

    type Event struct {
        Status         string `json:"status"`
        Error          string `json:"error"`
        Progress       string `json:"progress"`
        ProgressDetail struct {
            Current int `json:"current"`
            Total   int `json:"total"`
        } `json:"progressDetail"`
    }

    var event *Event
    for {
        if err := d.Decode(&event); err != nil {
            if err == io.EOF {
                break
            }

            panic(err)
        }

        fmt.Printf("EVENT: %+v\n", event)
    }

    // Latest event for new image
    // EVENT: {Status:Status: Downloaded newer image for busybox:latest Error: Progress:[==================================================>]  699.2kB/699.2kB ProgressDetail:{Current:699243 Total:699243}}
    // Latest event for up-to-date image
    // EVENT: {Status:Status: Image is up to date for busybox:latest Error: Progress: ProgressDetail:{Current:0 Total:0}}
    if event != nil {
        if strings.Contains(event.Status, fmt.Sprintf("Downloaded newer image for %s", imageName)) {
            // new
            fmt.Println("new")
        }

        if strings.Contains(event.Status, fmt.Sprintf("Image is up to date for %s", imageName)) {
            // up-to-date
            fmt.Println("up-to-date")
        }
    }
}

您可以查看 API 格式来创建您的结构（例如我的 Event）以在此处阅读它们 https://docs.docker.com/engine/api/v1.27/#operation/ImageCreate

希望能帮到你解决问题，谢谢

Answer 3

我为我的目的使用了类似的方法（不是 moby 客户端）。通常，读取流响应的想法是相同的。试一试并实施你的。

读取任何响应类型的流响应：

reader := bufio.NewReader(pullResp)
defer pullResp.Close()  // pullResp is io.ReadCloser
var resp bytes.Buffer
for {
    line, err := reader.ReadBytes('\n')
    if err != nil {
        // it could be EOF or read error
        // handle it
        break
    }
    resp.Write(line)
    resp.WriteByte('\n')
}

// print it
fmt.Println(resp.String())

---

但是，您在评论中的示例响应似乎是有效的 JSON 结构。 json.Decoder 是读取 JSON 流的最佳方式。这只是一个想法-

type ImagePullResponse struct {
   ID             string `json"id"`
   Status         string `json:"status"`
   ProgressDetail struct {
     Current int64 `json:"current"`
     Total   int64 `json:"total"`
   } `json:"progressDetail"`
   Progress string `json:"progress"`
}

然后做

d := json.NewDecoder(pullResp)
for {
   var pullResult ImagePullResponse
  if err := d.Decode(&pullResult); err != nil {
    // handle the error
    break
  }
  fmt.Println(pullResult)
}