-
Notifications
You must be signed in to change notification settings - Fork 277
/
installer.go
101 lines (87 loc) 路 2.62 KB
/
installer.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 eksd
import (
"context"
"fmt"
"time"
"github.com/aws/eks-anywhere/pkg/cluster"
"github.com/aws/eks-anywhere/pkg/constants"
"github.com/aws/eks-anywhere/pkg/logger"
"github.com/aws/eks-anywhere/pkg/retrier"
"github.com/aws/eks-anywhere/pkg/types"
)
const (
maxRetries = 5
backOffPeriod = 5 * time.Second
)
type EksdInstallerClient interface {
ApplyKubeSpecFromBytesWithNamespace(ctx context.Context, cluster *types.Cluster, data []byte, namespace string) error
}
type Reader interface {
ReadFile(url string) ([]byte, error)
}
type Installer struct {
client EksdInstallerClient
retrier *retrier.Retrier
reader Reader
}
func NewEksdInstaller(client EksdInstallerClient, reader Reader) *Installer {
return &Installer{
client: client,
retrier: retrier.NewWithMaxRetries(maxRetries, backOffPeriod),
reader: reader,
}
}
func (i *Installer) InstallEksdCRDs(ctx context.Context, clusterSpec *cluster.Spec, cluster *types.Cluster) error {
var eksdCRDs []byte
eksdCrds := map[string]struct{}{}
for _, vb := range clusterSpec.Bundles.Spec.VersionsBundles {
eksdCrds[vb.EksD.Components] = struct{}{}
}
for crd := range eksdCrds {
if err := i.retrier.Retry(
func() error {
var readerErr error
eksdCRDs, readerErr = i.reader.ReadFile(crd)
return readerErr
},
); err != nil {
return fmt.Errorf("loading manifest for eksd components: %v", err)
}
if err := i.retrier.Retry(
func() error {
return i.client.ApplyKubeSpecFromBytesWithNamespace(ctx, cluster, eksdCRDs, constants.EksaSystemNamespace)
},
); err != nil {
return fmt.Errorf("applying eksd release crd: %v", err)
}
}
return nil
}
// SetRetrier allows to modify the internal retrier
// For unit testing purposes only. It is not thread safe
func (i *Installer) SetRetrier(retrier *retrier.Retrier) {
i.retrier = retrier
}
func (i *Installer) InstallEksdManifest(ctx context.Context, clusterSpec *cluster.Spec, cluster *types.Cluster) error {
var eksdReleaseManifest []byte
for _, vb := range clusterSpec.Bundles.Spec.VersionsBundles {
if err := i.retrier.Retry(
func() error {
var readerErr error
eksdReleaseManifest, readerErr = i.reader.ReadFile(vb.EksD.EksDReleaseUrl)
return readerErr
},
); err != nil {
return fmt.Errorf("loading manifest for eksd components: %v", err)
}
logger.V(4).Info("Applying eksd manifest to cluster")
if err := i.retrier.Retry(
func() error {
return i.client.ApplyKubeSpecFromBytesWithNamespace(ctx, cluster, eksdReleaseManifest, constants.EksaSystemNamespace)
},
); err != nil {
return fmt.Errorf("applying eksd release manifest: %v", err)
}
}
return nil
}