-
Notifications
You must be signed in to change notification settings - Fork 66
/
datafederation.go
114 lines (87 loc) · 3.63 KB
/
datafederation.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
102
103
104
105
106
107
108
109
110
111
112
113
114
package atlasdatafederation
import (
"net/http"
"go.mongodb.org/atlas/mongodbatlas"
"github.com/google/go-cmp/cmp"
"github.com/google/go-cmp/cmp/cmpopts"
"go.uber.org/zap"
"github.com/mongodb/mongodb-atlas-kubernetes/v2/internal/compat"
akov2 "github.com/mongodb/mongodb-atlas-kubernetes/v2/pkg/api/v1"
"github.com/mongodb/mongodb-atlas-kubernetes/v2/pkg/api/v1/common"
"github.com/mongodb/mongodb-atlas-kubernetes/v2/pkg/controller/workflow"
)
func (r *AtlasDataFederationReconciler) ensureDataFederation(ctx *workflow.Context, project *akov2.AtlasProject, dataFederation *akov2.AtlasDataFederation) workflow.Result {
log := ctx.Log
projectID := project.ID()
operatorSpec := &dataFederation.Spec
dataFederationToAtlas, err := dataFederation.ToAtlas()
if err != nil {
return workflow.Terminate(workflow.Internal, "can not convert DataFederation (operator -> atlas)")
}
atlasSpec, resp, err := ctx.Client.DataFederation.Get(ctx.Context, projectID, operatorSpec.Name)
if err != nil {
if resp == nil {
return workflow.Terminate(workflow.Internal, err.Error())
}
if resp.StatusCode != http.StatusNotFound {
return workflow.Terminate(workflow.DataFederationNotCreatedInAtlas, err.Error())
}
_, _, err = ctx.Client.DataFederation.Create(ctx.Context, projectID, dataFederationToAtlas)
if err != nil {
return workflow.Terminate(workflow.DataFederationNotCreatedInAtlas, err.Error())
}
return workflow.InProgress(workflow.DataFederationCreating, "Data Federation is being created")
}
dfFromAtlas, err := DataFederationFromAtlas(atlasSpec)
if err != nil {
return workflow.Terminate(workflow.Internal, "can not convert DataFederation (atlas -> operator)")
}
if areEqual, _ := dataFederationEqual(*dfFromAtlas, *operatorSpec, log); areEqual {
return workflow.OK()
}
_, _, err = ctx.Client.DataFederation.Update(ctx.Context, projectID, dataFederation.Spec.Name, dataFederationToAtlas, nil)
if err != nil {
return workflow.Terminate(workflow.DataFederationNotUpdatedInAtlas, err.Error())
}
return workflow.InProgress(workflow.DataFederationUpdating, "Data Federation is being updated")
}
func DataFederationFromAtlas(atlasDF *mongodbatlas.DataFederationInstance) (*akov2.DataFederationSpec, error) {
dfSpec := &akov2.DataFederationSpec{}
err := compat.JSONCopy(dfSpec, atlasDF)
return dfSpec, err
}
func dataFederationEqual(atlasSpec, operatorSpec akov2.DataFederationSpec, log *zap.SugaredLogger) (areEqual bool, diff string) {
mergedSpec, err := getMergedSpec(atlasSpec, operatorSpec)
if err != nil {
log.Errorf("failed to merge Data Federation specs: %s", err.Error())
return false, ""
}
d := cmp.Diff(atlasSpec, mergedSpec, cmpopts.EquateEmpty())
if d != "" {
log.Debugf("Data Federation diff: \n%s", d)
}
return d == "", d
}
func getMergedSpec(atlasSpec, operatorSpec akov2.DataFederationSpec) (akov2.DataFederationSpec, error) {
mergedSpec := akov2.DataFederationSpec{}
operatorSpec.PrivateEndpoints = []akov2.DataFederationPE{}
if err := compat.JSONCopy(&mergedSpec, atlasSpec); err != nil {
return mergedSpec, err
}
if err := compat.JSONCopy(&mergedSpec, operatorSpec); err != nil {
return mergedSpec, err
}
mergedSpec.Project = common.ResourceRefNamespaced{}
return mergedSpec, nil
}
func dataFederationMatchesSpec(log *zap.SugaredLogger, atlasSpec *mongodbatlas.DataFederationInstance, operatorSpec *akov2.AtlasDataFederation) (bool, error) {
newAtlasSpec, err := DataFederationFromAtlas(atlasSpec)
if err != nil {
return false, err
}
equal, diff := dataFederationEqual(*newAtlasSpec, operatorSpec.Spec, log)
if !equal {
log.Debugf("DataFederation differs from spec: %s", diff)
}
return equal, nil
}