-
Notifications
You must be signed in to change notification settings - Fork 199
/
stages_iterator.go
81 lines (66 loc) · 3.04 KB
/
stages_iterator.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
package build
import (
"context"
"fmt"
"github.com/werf/logboek"
build_image "github.com/werf/werf/pkg/build/image"
"github.com/werf/werf/pkg/build/stage"
)
type StagesIterator struct {
Conveyor *Conveyor
PrevStage stage.Interface
PrevNonEmptyStage stage.Interface
PrevBuiltStage stage.Interface
PrevNonEmptyStageImageSize int64
}
func NewStagesIterator(conveyor *Conveyor) *StagesIterator {
return &StagesIterator{Conveyor: conveyor}
}
func (iterator *StagesIterator) GetPrevImage(img *build_image.Image, stg stage.Interface) *stage.StageImage {
if stg.HasPrevStage() {
return iterator.PrevNonEmptyStage.GetStageImage()
} else if stg.IsStapelStage() && stg.Name() == "from" {
return img.GetBaseStageImage()
} else if img.IsDockerfileImage && img.DockerfileImageConfig.Staged {
return img.GetBaseStageImage()
}
return nil
}
func (iterator *StagesIterator) GetPrevBuiltImage(img *build_image.Image, stg stage.Interface) *stage.StageImage {
if stg.HasPrevStage() {
return iterator.PrevBuiltStage.GetStageImage()
} else if stg.IsStapelStage() && stg.Name() == "from" {
return img.GetBaseStageImage()
} else if img.IsDockerfileImage && img.DockerfileImageConfig.Staged {
return img.GetBaseStageImage()
}
return nil
}
func (iterator *StagesIterator) OnImageStage(ctx context.Context, img *build_image.Image, stg stage.Interface, onImageStageFunc func(img *build_image.Image, stg stage.Interface, isEmpty bool) error) error {
isEmpty, err := stg.IsEmpty(ctx, iterator.Conveyor, iterator.GetPrevBuiltImage(img, stg)) // FIXME(stapel-to-buildah): use StageImage
if err != nil {
return fmt.Errorf("error checking stage %s is empty: %w", stg.Name(), err)
}
logboek.Context(ctx).Debug().LogF("%s stage is empty: %v\n", stg.LogDetailedName(), isEmpty)
if stg.HasPrevStage() && iterator.PrevStage == nil {
panic(fmt.Sprintf("expected PrevStage to be set for image %q stage %s!", img.GetName(), stg.Name()))
}
if err := onImageStageFunc(img, stg, isEmpty); err != nil {
return err
}
iterator.PrevStage = stg
logboek.Context(ctx).Debug().LogF("Set prev stage = %q %s\n", iterator.PrevStage.Name(), iterator.PrevStage.GetDigest())
if !isEmpty {
iterator.PrevNonEmptyStage = stg
logboek.Context(ctx).Debug().LogF("Set prev non empty stage = %q %s\n", iterator.PrevNonEmptyStage.Name(), iterator.PrevNonEmptyStage.GetDigest())
if iterator.PrevNonEmptyStage.GetStageImage().Image.GetStageDescription() != nil {
iterator.PrevNonEmptyStageImageSize = iterator.PrevNonEmptyStage.GetStageImage().Image.GetStageDescription().Info.Size
logboek.Context(ctx).Debug().LogF("Set prev non empty stage image size = %d %q %s\n", iterator.PrevNonEmptyStageImageSize, iterator.PrevNonEmptyStage.Name(), iterator.PrevNonEmptyStage.GetDigest())
}
if stg.GetStageImage().Image.GetStageDescription() != nil {
iterator.PrevBuiltStage = stg
logboek.Context(ctx).Debug().LogF("Set prev built stage = %q (image %s)\n", iterator.PrevBuiltStage.Name(), iterator.PrevBuiltStage.GetStageImage().Image.Name())
}
}
return nil
}