kuma

1
package defaults
2

3
import (
4
	"context"
5
	"time"
6

7
	"github.com/go-logr/logr"
8
	"github.com/pkg/errors"
9
	"github.com/sethvargo/go-retry"
10

11
	"github.com/kumahq/kuma/pkg/core/resources/apis/system"
12
	"github.com/kumahq/kuma/pkg/core/resources/manager"
13
	"github.com/kumahq/kuma/pkg/core/resources/model"
14
	"github.com/kumahq/kuma/pkg/core/resources/store"
15
	"github.com/kumahq/kuma/pkg/core/runtime/component"
16
	"github.com/kumahq/kuma/pkg/core/user"
17
	kuma_log "github.com/kumahq/kuma/pkg/log"
18
)
19

20
type ZoneDefaultComponent struct {
21
	ResManager manager.ResourceManager
22
	Extensions context.Context
23
	ZoneName   string
24
}
25

26
var _ component.Component = &ZoneDefaultComponent{}
27

28
func (e *ZoneDefaultComponent) Start(stop <-chan struct{}) error {
29
	ctx, cancelFn := context.WithCancel(user.Ctx(context.Background(), user.ControlPlane))
30
	defer cancelFn()
31
	logger := kuma_log.AddFieldsFromCtx(log, ctx, e.Extensions)
32
	errChan := make(chan error)
33
	go func() {
34
		errChan <- retry.Do(ctx, retry.WithMaxDuration(10*time.Minute, retry.NewConstant(5*time.Second)), func(ctx context.Context) error {
35
			if err := EnsureOnlyOneZoneExists(ctx, e.ResManager, e.ZoneName, logger); err != nil {
36
				log.V(1).Info("could not ensure that Zone exists. Retrying.", "err", err)
37
				return retry.RetryableError(err)
38
			}
39
			return nil
40
		})
41
	}()
42
	select {
43
	case <-stop:
44
		return nil
45
	case err := <-errChan:
46
		return err
47
	}
48
}
49

50
func (e ZoneDefaultComponent) NeedLeaderElection() bool {
51
	return true
52
}
53

54
func EnsureOnlyOneZoneExists(
55
	ctx context.Context,
56
	resManager manager.ResourceManager,
57
	zoneName string,
58
	logger logr.Logger,
59
) error {
60
	logger.Info("ensuring Zone resource exists", "name", zoneName)
61
	zones := &system.ZoneResourceList{}
62
	if err := resManager.List(ctx, zones); err != nil {
63
		return errors.Wrap(err, "cannot list zones")
64
	}
65
	exists := false
66
	for _, zone := range zones.Items {
67
		if zone.GetMeta().GetName() == zoneName {
68
			exists = true
69
		} else {
70
			logger.Info("detected Zone resource with different name than Zone CP name. Deleting. This might happen if you change the name of the Zone CP", "name", zoneName)
71
			if err := resManager.Delete(ctx, zone, store.DeleteByKey(zone.GetMeta().GetName(), model.NoMesh)); err != nil {
72
				return errors.Wrap(err, "cannot delete old zone")
73
			}
74
		}
75
	}
76
	if !exists {
77
		logger.Info("creating Zone resource", "name", zoneName)
78
		zone := system.NewZoneResource()
79
		if err := resManager.Create(ctx, zone, store.CreateByKey(zoneName, model.NoMesh)); err != nil {
80
			return err
81
		}
82
	}
83
	return nil
84
}
85