/
build.go
101 lines (84 loc) · 2.27 KB
/
build.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
package actions
import (
"context"
"strings"
"github.com/pkg/errors"
"github.com/spf13/cobra"
"golang.org/x/sync/errgroup"
"github.com/foomo/squadron"
)
func init() {
buildCmd.Flags().BoolVarP(&flagPush, "push", "p", false, "pushes built squadron units to the registry")
buildCmd.Flags().IntVar(&flagParallel, "parallel", 1, "run command in parallel")
buildCmd.Flags().StringVar(&flagBuildArgs, "build-args", "", "additional docker buildx build args")
buildCmd.Flags().StringVar(&flagPushArgs, "push-args", "", "additional docker push args")
}
var buildCmd = &cobra.Command{
Use: "build [UNIT...]",
Short: "build or rebuild squadron units",
Example: " squadron build frontend backend",
Args: cobra.MinimumNArgs(0),
RunE: func(cmd *cobra.Command, args []string) error {
return build(cmd.Context(), args, cwd, flagFiles, flagPush, flagParallel)
},
}
func build(ctx context.Context, args []string, cwd string, files []string, push bool, parallel int) error {
sq := squadron.New(cwd, "", files)
if err := sq.MergeConfigFiles(); err != nil {
return err
}
unitsNames, err := parseUnitNames(args, sq.GetConfig().Units)
if err != nil {
return err
}
if unitsNames != nil {
if err := sq.FilterConfig(unitsNames); err != nil {
return err
}
}
if err := sq.RenderConfig(ctx); err != nil {
return err
}
units, err := parseUnitArgs(args, sq.GetConfig().Units)
if err != nil {
return err
}
{
g, gctx := errgroup.WithContext(ctx)
g.SetLimit(parallel)
_ = squadron.Units(units).Iterate(func(n string, u *squadron.Unit) error {
name := n
unit := u
g.Go(func() error {
if out, err := unit.Build(gctx, sq.Name(), name, strings.Split(flagBuildArgs, " ")); err != nil {
return errors.Wrap(err, out)
}
return nil
})
return nil
})
err := g.Wait()
if err != nil {
return err
}
}
if push {
g, gctx := errgroup.WithContext(ctx)
g.SetLimit(parallel)
_ = squadron.Units(units).Iterate(func(n string, u *squadron.Unit) error {
name := n
unit := u
g.Go(func() error {
if out, err := unit.Push(gctx, sq.Name(), name, strings.Split(flagPushArgs, " ")); err != nil {
return errors.Wrap(err, out)
}
return nil
})
return nil
})
if err := g.Wait(); err != nil {
return err
}
}
return nil
}