Skip to content

Commit 50d9c38

Browse files
authored
CLOUDP-212108: Fix Net Peerings part 2 (#1266)
Signed-off-by: Jose Vazquez <jose.vazquez@mongodb.com>
1 parent 871e5cb commit 50d9c38

File tree

3 files changed

+13
-7
lines changed

3 files changed

+13
-7
lines changed

tools/clean/atlas/atlas.go

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,12 @@ const (
1818
CloudProviderAZURE = "AZURE"
1919
)
2020

21+
var (
22+
SupportedProviders = []string{CloudProviderAWS, CloudProviderGCP, CloudProviderAZURE}
23+
24+
GovSupportedProviders = []string{CloudProviderAWS}
25+
)
26+
2127
type Cleaner struct {
2228
client *admin.APIClient
2329
aws *provider.AWS

tools/clean/atlas/network_peering.go

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,9 @@ import (
88
"go.mongodb.org/atlas-sdk/v20231001002/admin"
99
)
1010

11-
var (
12-
SupportedProviders = []string{"AWS", "AZURE", "GCP"}
13-
)
14-
15-
func (c *Cleaner) listNetworkPeering(ctx context.Context, projectID string) []admin.BaseNetworkPeeringConnectionSettings {
11+
func (c *Cleaner) listNetworkPeering(ctx context.Context, projectID string, providers []string) []admin.BaseNetworkPeeringConnectionSettings {
1612
peers := []admin.BaseNetworkPeeringConnectionSettings{}
17-
for _, providerName := range SupportedProviders {
13+
for _, providerName := range providers {
1814
peers = append(peers, c.listNetworkPeeringForProvider(ctx, projectID, providerName)...)
1915
}
2016
if len(peers) == 0 {

tools/clean/atlas/projects.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,11 @@ func (c *Cleaner) GetProjectDependencies(ctx context.Context, projectID string,
4949
go func() {
5050
defer wg.Done()
5151

52-
deps.NetworkPeering = c.listNetworkPeering(ctx, projectID)
52+
providers := SupportedProviders
53+
if isGov {
54+
providers = GovSupportedProviders
55+
}
56+
deps.NetworkPeering = c.listNetworkPeering(ctx, projectID, providers)
5357
}()
5458

5559
go func() {

0 commit comments

Comments
 (0)