-
Notifications
You must be signed in to change notification settings - Fork 0
/
merge.go
86 lines (67 loc) · 2.23 KB
/
merge.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
package interfaces
import (
"github.com/control-monkey/terraform-provider-cm/internal/helpers"
"github.com/hashicorp/go-set/v2"
"reflect"
)
type OperationType string
const (
CreateOperation OperationType = "create"
UpdateOperation OperationType = "update"
DeleteOperation OperationType = "delete"
)
type MergedEntities[T MergeModel] struct {
EntitiesToCreate set.Collection[T]
EntitiesToUpdate set.Collection[T]
EntitiesToDelete set.Collection[T]
}
type MergeModel interface {
set.Hasher[string]
GetBlockIdentifier() string
}
func GetIdentifiers[T MergeModel](ms []T) []string {
knownEntities := filterOutUnknowns(ms)
retVal := mapToIdentifiers(knownEntities)
return retVal
}
func filterOutUnknowns[T MergeModel](ms []T) []T {
f := func(m T) bool {
return (m).GetBlockIdentifier() != ""
}
retVal := helpers.Filter(ms, f)
return retVal
}
func mapToIdentifiers[T MergeModel](ms []T) []string {
mapToIdentifier := func(m T) string {
return (m).GetBlockIdentifier()
}
return helpers.Map(ms, mapToIdentifier)
}
func MergeEntities[T MergeModel](plan []T, state []T) MergedEntities[T] {
planEntities := set.HashSetFrom[T, string](plan)
stateEntities := set.HashSetFrom[T, string](state)
entitiesToCreate := planEntities.Difference(stateEntities)
entitiesToDelete := stateEntities.Difference(planEntities)
idToModelToDelete := make(map[string]T, 0)
entitiesToRemoveFromCreate := make([]T, 0)
entitiesToRemoveFromDelete := make([]T, 0)
entitiesToUpdate := set.NewHashSet[T, string](0)
for _, p := range entitiesToDelete.Slice() {
idToModelToDelete[p.GetBlockIdentifier()] = p
}
for _, p := range entitiesToCreate.Slice() {
if entityToDelete := idToModelToDelete[p.GetBlockIdentifier()]; reflect.ValueOf(entityToDelete).IsNil() == false { // id in both add & delete
entitiesToRemoveFromCreate = append(entitiesToRemoveFromCreate, p)
entitiesToRemoveFromDelete = append(entitiesToRemoveFromDelete, entityToDelete)
entitiesToUpdate.Insert(p)
}
}
entitiesToCreate.RemoveSlice(entitiesToRemoveFromCreate)
entitiesToDelete.RemoveSlice(entitiesToRemoveFromDelete)
retVal := MergedEntities[T]{
EntitiesToCreate: entitiesToCreate,
EntitiesToUpdate: entitiesToUpdate,
EntitiesToDelete: entitiesToDelete,
}
return retVal
}