-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
cloudsql.go
99 lines (86 loc) · 2.51 KB
/
cloudsql.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
// Copyright 2018 The Terraformer Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package gcp
import (
"context"
"github.com/GoogleCloudPlatform/terraformer/terraformutils"
sqladmin "google.golang.org/api/sqladmin/v1beta4"
)
var cloudSQLAllowEmptyValues = []string{}
var cloudSQLAdditionalFields = map[string]interface{}{}
type CloudSQLGenerator struct {
GCPService
}
func (g *CloudSQLGenerator) loadDBInstances(svc *sqladmin.Service, project string) error {
dbInstances, err := svc.Instances.List(project).Do()
if err != nil {
return err
}
for _, dbInstance := range dbInstances.Items {
g.Resources = append(g.Resources, terraformutils.NewResource(
dbInstance.Name,
dbInstance.Name,
"google_sql_database_instance",
g.ProviderName,
map[string]string{
"project": project,
"name": dbInstance.Name,
},
cloudSQLAllowEmptyValues,
cloudSQLAdditionalFields,
))
err := g.loadDBs(svc, dbInstance.Name, project)
if err != nil {
return err
}
}
return nil
}
func (g *CloudSQLGenerator) loadDBs(svc *sqladmin.Service, instanceName, project string) error {
DBs, err := svc.Databases.List(project, instanceName).Do()
if err != nil {
return err
}
for _, db := range DBs.Items {
g.Resources = append(g.Resources, terraformutils.NewResource(
instanceName+":"+db.Name,
instanceName+"-"+db.Name,
"google_sql_database",
g.ProviderName,
map[string]string{
"instance": instanceName,
"project": project,
"name": db.Name,
},
cloudSQLAllowEmptyValues,
cloudSQLAdditionalFields,
))
}
return nil
}
// Generate TerraformResources from GCP API,
// from each databases create many TerraformResource(dbinstance + databases)
// Need dbinstance name as ID for terraform resource
func (g *CloudSQLGenerator) InitResources() error {
project := g.GetArgs()["project"].(string)
ctx := context.Background()
svc, err := sqladmin.NewService(ctx)
if err != nil {
return err
}
if err := g.loadDBInstances(svc, project); err != nil {
return err
}
return nil
}