This repository has been archived by the owner on Jan 21, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1
/
migrator.go
124 lines (96 loc) · 3.04 KB
/
migrator.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
115
116
117
118
119
120
121
122
123
124
package migrator
import (
"io"
"log"
"sync"
"code.cloudfoundry.org/cc-to-perm-migrator/migrator/models"
"code.cloudfoundry.org/lager"
)
//go:generate counterfeiter . Retriever
type Retriever interface {
FetchResources(logger lager.Logger, progressLogger *log.Logger, orgs chan<- models.Organization, spaces chan<- models.Space, errs chan<- error)
}
//go:generate counterfeiter . Populator
type Populator interface {
PopulateOrganization(logger lager.Logger, org models.Organization, namespace string) []error
PopulateSpace(logger lager.Logger, space models.Space, namespace string) []error
}
//go:generate counterfeiter . Reporter
type Reporter interface {
GenerateReport(w io.Writer, orgs []models.Organization, spaces []models.Space, errs []error)
}
type Migrator struct {
retriever Retriever
populator Populator
reporter Reporter
namespace string
}
func NewMigrator(retriever Retriever, populator Populator, reporter Reporter, namespace string) *Migrator {
return &Migrator{
retriever: retriever,
populator: populator,
reporter: reporter,
namespace: namespace,
}
}
func (m *Migrator) Migrate(logger lager.Logger, progressLogger *log.Logger, writer io.Writer, dryRun bool) {
orgChan := make(chan models.Organization)
spaceChan := make(chan models.Space)
errChan := make(chan error)
var (
orgs []models.Organization
spaces []models.Space
retrieveErrs []error
populateOrgErrs []error
populateSpaceErrs []error
wg sync.WaitGroup
)
if dryRun {
progressLogger.Print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
progressLogger.Print("@ DRY-RUN; ROLE ASSIGNMENTS WILL NOT BE MIGRATED @")
progressLogger.Print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
defer func() {
progressLogger.Print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
progressLogger.Print("@ DRY-RUN; ROLE ASSIGNMENTS WERE NOT MIGRATED @")
progressLogger.Print("@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@")
}()
}
wg.Add(3)
go func() {
defer close(orgChan)
defer close(spaceChan)
defer close(errChan)
m.retriever.FetchResources(logger, progressLogger, orgChan, spaceChan, errChan)
}()
go func() {
defer wg.Done()
orgLogger := logger.Session("populate-organizations")
for org := range orgChan {
orgs = append(orgs, org)
if !dryRun {
errs := m.populator.PopulateOrganization(orgLogger, org, m.namespace)
populateOrgErrs = append(populateOrgErrs, errs...)
}
}
}()
go func() {
defer wg.Done()
spaceLogger := logger.Session("populate-spaces")
for space := range spaceChan {
spaces = append(spaces, space)
if !dryRun {
errs := m.populator.PopulateSpace(spaceLogger, space, m.namespace)
populateSpaceErrs = append(populateSpaceErrs, errs...)
}
}
}()
go func() {
defer wg.Done()
for err := range errChan {
retrieveErrs = append(retrieveErrs, err)
}
}()
wg.Wait()
errs := append(retrieveErrs, append(populateOrgErrs, populateSpaceErrs...)...)
m.reporter.GenerateReport(writer, orgs, spaces, errs)
}