-
Notifications
You must be signed in to change notification settings - Fork 12
/
image_build_response.go
131 lines (108 loc) · 3.17 KB
/
image_build_response.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
package docker
import (
"encoding/json"
"fmt"
"io"
"github.com/docker/docker/api/types"
"github.com/docker/docker/pkg/jsonmessage"
)
const (
checkingStartCode = iota
processingStartCode
processingDataAndCheckingStopCode
processingStopCode
)
var (
artifactsTarStartReadCode = []byte("1EA01F53E0277546E1B17267F29A60B3CD4DC12744C2FA2BF0897065DC3749F3")
artifactsTarStopReadCode = []byte("A2F00DB0DEE3540E246B75B872D64773DF67BC51C5D36D50FA6978E2FFDA7D43")
)
func DisplayFromImageBuildResponse(w io.Writer, response types.ImageBuildResponse) error {
dec := json.NewDecoder(response.Body)
for {
var jm jsonmessage.JSONMessage
if err := dec.Decode(&jm); err != nil {
if err == io.EOF {
return nil
}
return fmt.Errorf("unable to decode message from docker daemon: %w", err)
}
if err := jm.Display(w, false); err != nil {
return err
}
}
}
func ReadTarFromImageBuildResponse(tarWriter, buildLogWriter io.Writer, response types.ImageBuildResponse) error {
dec := json.NewDecoder(response.Body)
currentState := checkingStartCode
var codeCursor int
var bufferedData []byte
for {
var jm jsonmessage.JSONMessage
if err := dec.Decode(&jm); err != nil {
if err == io.EOF {
return nil
}
return fmt.Errorf("unable to decode message from docker daemon: %w", err)
}
if jm.Error != nil {
return jm.Error
}
var startReadCodeSuspectedBytes []byte
msg := jm.Stream
if msg != "" {
for _, b := range []byte(msg) {
switch currentState {
case checkingStartCode:
if b == artifactsTarStartReadCode[0] {
currentState = processingStartCode
codeCursor++
startReadCodeSuspectedBytes = append(startReadCodeSuspectedBytes, b)
} else if _, err := buildLogWriter.Write([]byte{b}); err != nil {
return fmt.Errorf("build log writer failed: %w", err)
}
case processingStartCode:
if b == artifactsTarStartReadCode[codeCursor] {
if len(artifactsTarStartReadCode) > codeCursor+1 {
codeCursor++
startReadCodeSuspectedBytes = append(startReadCodeSuspectedBytes, b)
} else {
currentState = processingDataAndCheckingStopCode
codeCursor = 0
startReadCodeSuspectedBytes = nil
}
} else {
currentState = checkingStartCode
codeCursor = 0
if _, err := buildLogWriter.Write(append(startReadCodeSuspectedBytes, b)); err != nil {
return fmt.Errorf("build log writer failed: %w", err)
}
startReadCodeSuspectedBytes = nil
}
case processingDataAndCheckingStopCode:
bufferedData = append(bufferedData, b)
if b == artifactsTarStopReadCode[0] {
currentState = processingStopCode
codeCursor++
continue
}
if _, err := tarWriter.Write(bufferedData); err != nil {
return fmt.Errorf("tar writer failed: %w", err)
}
bufferedData = nil
case processingStopCode:
bufferedData = append(bufferedData, b)
if b == artifactsTarStopReadCode[codeCursor] {
if len(artifactsTarStopReadCode) > codeCursor+1 {
codeCursor++
} else {
return nil
}
} else {
currentState = processingDataAndCheckingStopCode
codeCursor = 0
}
}
}
}
}
}