forked from concourse/concourse
-
Notifications
You must be signed in to change notification settings - Fork 0
/
image_factory.go
60 lines (50 loc) · 1.34 KB
/
image_factory.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
package image
import (
"errors"
"code.cloudfoundry.org/lager"
"github.com/pf-qiu/concourse/v6/atc/worker"
)
var ErrUnsupportedResourceType = errors.New("unsupported resource type")
type imageFactory struct{}
func NewImageFactory() worker.ImageFactory {
return &imageFactory{}
}
func (f *imageFactory) GetImage(
logger lager.Logger,
worker worker.Worker,
volumeClient worker.VolumeClient,
imageSpec worker.ImageSpec,
teamID int,
) (worker.Image, error) {
if imageSpec.ImageArtifactSource != nil {
artifactVolume, existsOnWorker, err := imageSpec.ImageArtifactSource.ExistsOn(logger, worker)
if err != nil {
logger.Error("failed-to-check-if-volume-exists-on-worker", err)
return nil, err
}
if existsOnWorker {
return &imageProvidedByPreviousStepOnSameWorker{
artifactVolume: artifactVolume,
imageSpec: imageSpec,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
return &imageProvidedByPreviousStepOnDifferentWorker{
imageSpec: imageSpec,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
if imageSpec.ResourceType != "" {
return &imageFromBaseResourceType{
worker: worker,
resourceTypeName: imageSpec.ResourceType,
teamID: teamID,
volumeClient: volumeClient,
}, nil
}
return &imageFromRootfsURI{
url: imageSpec.ImageURL,
}, nil
}