-
Notifications
You must be signed in to change notification settings - Fork 503
/
main.go
100 lines (82 loc) · 2.2 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package main
import (
"context"
"encoding/json"
"fmt"
"os"
"sync"
"text/template"
"github.com/spf13/pflag"
"github.com/togettoyou/hub-mirror/pkg"
)
var (
content = pflag.StringP("content", "", "", "原始镜像,格式为:{ \"platform\": \"\", \"hub-mirror\": [] }")
maxContent = pflag.IntP("maxContent", "", 11, "原始镜像个数限制")
repository = pflag.StringP("repository", "", "", "推送仓库地址,为空默认为 hub.docker.com")
username = pflag.StringP("username", "", "", "仓库用户名")
password = pflag.StringP("password", "", "", "仓库密码")
outputPath = pflag.StringP("outputPath", "", "output.md", "结果输出路径")
)
func main() {
pflag.Parse()
fmt.Println("验证原始镜像内容")
var hubMirrors struct {
Content []string `json:"hub-mirror"`
Platform string `json:"platform"`
}
err := json.Unmarshal([]byte(*content), &hubMirrors)
if err != nil {
panic(err)
}
if len(hubMirrors.Content) > *maxContent {
panic("提交的原始镜像个数超出了最大限制")
}
fmt.Printf("mirrors: %+v, platform: %+v\n", hubMirrors.Content, hubMirrors.Platform)
fmt.Println("初始化 Docker 客户端")
cli, err := pkg.NewCli(context.Background(), *repository, *username, *password, os.Stdout)
if err != nil {
panic(err)
}
outputs := make([]*pkg.Output, 0)
mu := sync.Mutex{}
wg := sync.WaitGroup{}
for _, source := range hubMirrors.Content {
source := source
if source == "" {
continue
}
fmt.Println("开始转换镜像", source)
wg.Add(1)
go func() {
defer wg.Done()
output, err := cli.PullTagPushImage(context.Background(), source, hubMirrors.Platform)
if err != nil {
fmt.Println(source, "转换异常: ", err)
return
}
mu.Lock()
defer mu.Unlock()
outputs = append(outputs, output)
}()
}
wg.Wait()
if len(outputs) == 0 {
panic("没有转换成功的镜像")
}
tmpl, err := template.ParseFiles("output.tmpl")
if err != nil {
panic(err)
}
outputFile, err := os.Create(*outputPath)
if err != nil {
panic(err)
}
defer outputFile.Close()
err = tmpl.Execute(outputFile, map[string]interface{}{
"Outputs": outputs,
"Repository": *repository,
})
if err != nil {
panic(err)
}
}