Golang使用docker sdk管理docker

包括列出容器、创建容器、删除容器、进入容器、构建镜像等操作。

package dockertoolimport ("context""fmt""github.com/docker/docker/api/types""github.com/docker/docker/api/types/container""github.com/docker/docker/api/types/image""github.com/docker/docker/client""io""os"
)// 列出所有容器
func ListContainers() ([]types.Container, error) {apiClient, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())if err != nil {return nil, err}defer apiClient.Close()containers, err := apiClient.ContainerList(context.Background(), container.ListOptions{All: true})if err != nil {return nil, err}// TODO : for debugfor _, ctr := range containers {fmt.Printf("%s %s %q (status: %s)\n", ctr.ID, ctr.Image, ctr.Names, ctr.Status)}return containers, nil
}// 删除所有镜像和容器
func DeleteAllContainersAndImages() error {apiClient, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())if err != nil {return err}defer apiClient.Close()//停止、删除所有容器containers, err := apiClient.ContainerList(context.Background(), container.ListOptions{All: true})if err != nil {return err}for _, ctr := range containers {err = apiClient.ContainerStop(context.Background(), ctr.ID, container.StopOptions{})if err != nil {return err}err = apiClient.ContainerRemove(context.Background(), ctr.ID, container.RemoveOptions{})if err != nil {return err}}fmt.Println("All containers deleted")//删除所有镜像images, err := apiClient.ImageList(context.Background(), image.ListOptions{})if err != nil {return err}for _, im := range images {_, err := apiClient.ImageRemove(context.Background(), im.ID, image.RemoveOptions{})if err != nil {return err}}fmt.Println("All images deleted")return nil
}// 创建容器并运行
func BuildContainer(ContainerName string, ImageName string) error {cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())if err != nil {return err}defer cli.Close()config := &container.Config{Image: ImageName,}// 创建容器resp, err := cli.ContainerCreate(context.Background(), config, nil, nil, nil, ContainerName)if err != nil {panic(err)}fmt.Printf("Container created with ID: %s\n", resp.ID)// 启动容器if err := cli.ContainerStart(context.Background(), resp.ID, container.StartOptions{}); err != nil {panic(err)}fmt.Println("Container started")return nil
}// 本地创建镜像
func BuildImage(ImageName string, TarPath string, DockerfilePath string) error {cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())if err != nil {return err}defer cli.Close()// 设置构建上下文（这里使用文件路径作为示例）// 注意：这个路径应该包含 Dockerfile 和所有需要的构建文件buildContext, err := os.Open(TarPath)if err != nil {return err}defer buildContext.Close()// 设置构建选项options := types.ImageBuildOptions{Tags:           []string{ImageName},Dockerfile:     DockerfilePath,SuppressOutput: false, // 设置为 false 以获取构建日志NoCache:        false, // 不使用缓存Remove:         true,  // 构建完成后删除中间容器ForceRemove:    true,  // 强制删除中间容器PullParent:     true,  // 如果基础镜像不存在，则尝试拉取}// 触发构建response, err := cli.ImageBuild(context.Background(), buildContext, options)if err != nil {return err}fmt.Println("Built image " + ImageName)defer response.Body.Close()return nil
}// 进入容器并执行命令
func ContainerProcess(ContainerID string, ProcessID string) error {cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())if err != nil {return err}defer cli.Close()execConfig := types.ExecConfig{Cmd:          []string{"ps -f -p" + ProcessID}, // 使用bash执行ps命令，取决于容器内环境AttachStdin:  false,AttachStdout: true,AttachStderr: true,Tty:          false,}// 创建一个exec实例resp, err := cli.ContainerExecCreate(context.Background(), ContainerID, execConfig)if err != nil {return err}// 连接到exec实例的输出readCloser, err := cli.ContainerExecAttach(context.Background(), resp.ID, types.ExecStartCheck{})if err != nil {return err}defer readCloser.Close()_, err = io.Copy(os.Stdout, readCloser.Reader)if err != nil {return err}return nil
}