forked from vmware-archive/atc
-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
image_fetching_delegate.go
100 lines (81 loc) · 1.93 KB
/
image_fetching_delegate.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 engine
import (
"io"
"sync"
"unicode/utf8"
"github.com/concourse/atc"
"github.com/concourse/atc/db"
"github.com/concourse/atc/event"
"github.com/concourse/atc/exec"
)
type imageFetchingDelegate struct {
build db.Build
planID atc.PlanID
}
func (delegate *imageFetchingDelegate) ImageVersionDetermined(resourceCache *db.UsedResourceCache) error {
return delegate.build.SaveImageResourceVersion(resourceCache)
}
func (delegate *imageFetchingDelegate) Stdout() io.Writer {
return newDBEventWriter(
delegate.build,
event.Origin{
Source: event.OriginSourceStdout,
ID: event.OriginID(delegate.planID),
},
)
}
func (delegate *imageFetchingDelegate) Stderr() io.Writer {
return newDBEventWriter(
delegate.build,
event.Origin{
Source: event.OriginSourceStderr,
ID: event.OriginID(delegate.planID),
},
)
}
func newDBEventWriter(build db.Build, origin event.Origin) io.Writer {
return &dbEventWriter{
build: build,
origin: origin,
}
}
type dbEventWriter struct {
build db.Build
origin event.Origin
dangling []byte
}
func (writer *dbEventWriter) Write(data []byte) (int, error) {
text := append(writer.dangling, data...)
checkEncoding, _ := utf8.DecodeLastRune(text)
if checkEncoding == utf8.RuneError {
writer.dangling = text
return len(data), nil
}
writer.dangling = nil
err := writer.build.SaveEvent(event.Log{
Payload: string(text),
Origin: writer.origin,
})
if err != nil {
return 0, err
}
return len(data), nil
}
type implicitOutput struct {
resourceType string
info exec.VersionInfo
}
type implicitOutputsRepo struct {
outputs map[string]implicitOutput
lock *sync.Mutex
}
func (repo *implicitOutputsRepo) Register(resource string, output implicitOutput) {
repo.lock.Lock()
repo.outputs[resource] = output
repo.lock.Unlock()
}
func (repo *implicitOutputsRepo) Unregister(resource string) {
repo.lock.Lock()
delete(repo.outputs, resource)
repo.lock.Unlock()
}