/
VNetworkHandler.go
204 lines (171 loc) · 5.63 KB
/
VNetworkHandler.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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
// Proof of Concepts of CB-Spider.
// The CB-Spider is a sub-Framework of the Cloud-Barista Multi-Cloud Project.
// The CB-Spider Mission is to connect all the clouds with a single interface.
//
// * Cloud-Barista: https://github.com/cloud-barista
//
// This is a Cloud Driver Example for PoC Test.
//
// program by ysjeon@mz.co.kr, 2019.07.
// modify by devunet@mz.co.kr, 2019.11.
package resources
import (
"context"
"errors"
"fmt"
"strconv"
compute "google.golang.org/api/compute/v1"
"time"
idrv "github.com/cloud-barista/cb-spider/cloud-control-manager/cloud-driver/interfaces"
irs "github.com/cloud-barista/cb-spider/cloud-control-manager/cloud-driver/interfaces/resources"
"github.com/davecgh/go-spew/spew"
)
type GCPVNetworkHandler struct {
Region idrv.RegionInfo
Ctx context.Context
Client *compute.Service
Credential idrv.CredentialInfo
}
func (vNetworkHandler *GCPVNetworkHandler) CreateVNetwork(vNetworkReqInfo irs.VNetworkReqInfo) (irs.VNetworkInfo, error) {
projectID := vNetworkHandler.Credential.ProjectID
region := vNetworkHandler.Region.Region
name := GetCBDefaultVNetName()
vNetInfo, errVnet := vNetworkHandler.Client.Networks.Get(projectID, name).Do()
var cnt string
spew.Dump(vNetInfo)
if errVnet != nil {
network := &compute.Network{
Name: name,
//Name: GetCBDefaultVNetName(),
AutoCreateSubnetworks: true, // subnet 자동으로 생성됨
}
_, err := vNetworkHandler.Client.Networks.Insert(projectID, network).Do()
if err != nil {
cblogger.Error(err)
}
before_time := time.Now()
time.Sleep(time.Second * 20)
max_time := 120
// loop --> 생성 check --> 생성 되었으면, break; 안됐으면 sleep 5초 -->
// if(total sleep 120sec?) error
for {
newvNetInfo, errVnet := vNetworkHandler.Client.Networks.Get(projectID, name).Do()
if errVnet != nil {
time.Sleep(time.Second * 5)
after_time := time.Now()
diff := after_time.Sub(before_time)
if int(diff.Seconds()) > max_time {
cblogger.Error("max time 동안 vNet 정보가 조회가 안되어서 에러처리함")
cblogger.Error(errVnet)
return irs.VNetworkInfo{}, errVnet
}
} else {
cnt = strconv.Itoa(len(newvNetInfo.Subnetworks) + 1)
break
}
}
} else {
cnt = strconv.Itoa(len(vNetInfo.Subnetworks) + 1)
}
fmt.Println("CNT : ", cnt)
subnetInfo, errSubnet := vNetworkHandler.GetVNetwork(vNetworkReqInfo.Name)
if errSubnet == nil {
spew.Dump(subnetInfo)
cblogger.Error(errSubnet)
return irs.VNetworkInfo{}, errors.New("Already Exist")
}
// vNetResult, _ := vNetworkHandler.ListVNetwork()
networkUrl := "https://www.googleapis.com/compute/v1/projects/" + projectID + "/global/networks/" + name
subnetWork := &compute.Subnetwork{
Name: vNetworkReqInfo.Name,
IpCidrRange: "192.168." + cnt + ".0/24",
Network: networkUrl,
}
res, err := vNetworkHandler.Client.Subnetworks.Insert(projectID, region, subnetWork).Do()
if err != nil {
cblogger.Error(err)
return irs.VNetworkInfo{}, err
}
cblogger.Info(res)
//생성되는데 시간이 필요 함. 약 20초정도?
info, err2 := vNetworkHandler.Client.Subnetworks.Get(projectID, region, vNetworkReqInfo.Name).Do()
if err2 != nil {
cblogger.Error(err2)
return irs.VNetworkInfo{}, err2
}
networkInfo := irs.VNetworkInfo{
Name: info.Name,
Id: strconv.FormatUint(info.Id, 10),
AddressPrefix: info.IpCidrRange,
KeyValueList: []irs.KeyValue{
{"SubnetId", info.Name},
{"Region", info.Region},
{"GatewayAddress", info.GatewayAddress},
{"SelfLink", info.SelfLink},
},
}
return networkInfo, nil
}
func (vNetworkHandler *GCPVNetworkHandler) ListVNetwork() ([]*irs.VNetworkInfo, error) {
projectID := vNetworkHandler.Credential.ProjectID
region := vNetworkHandler.Region.Region
vNetworkList, err := vNetworkHandler.Client.Subnetworks.List(projectID, region).Do()
if err != nil {
return nil, err
}
var vNetworkInfo []*irs.VNetworkInfo
for _, item := range vNetworkList.Items {
networkInfo := irs.VNetworkInfo{
Name: item.Name,
Id: strconv.FormatUint(item.Id, 10),
AddressPrefix: item.IpCidrRange,
KeyValueList: []irs.KeyValue{
{"SubnetId", item.Name},
{"Region", item.Region},
{"GatewayAddress", item.GatewayAddress},
{"SelfLink", item.SelfLink},
},
}
vNetworkInfo = append(vNetworkInfo, &networkInfo)
}
return vNetworkInfo, nil
}
func (vNetworkHandler *GCPVNetworkHandler) GetVNetwork(vNetworkID string) (irs.VNetworkInfo, error) {
projectID := vNetworkHandler.Credential.ProjectID
region := vNetworkHandler.Region.Region
//name := vNetworkID
name := GetCBDefaultVNetName()
cblogger.Infof("Name : [%s]", name)
info, err := vNetworkHandler.Client.Subnetworks.Get(projectID, region, vNetworkID).Do()
if err != nil {
cblogger.Error(err)
return irs.VNetworkInfo{}, err
}
networkInfo := irs.VNetworkInfo{
Name: info.Name,
Id: strconv.FormatUint(info.Id, 10),
AddressPrefix: info.IpCidrRange,
KeyValueList: []irs.KeyValue{
{"SubnetId", info.Name},
{"Region", info.Region},
{"GatewayAddress", info.GatewayAddress},
{"SelfLink", info.SelfLink},
},
}
return networkInfo, nil
}
func (vNetworkHandler *GCPVNetworkHandler) DeleteVNetwork(vNetworkID string) (bool, error) {
projectID := vNetworkHandler.Credential.ProjectID
region := vNetworkHandler.Region.Region
//name := vNetworkID
name := GetCBDefaultVNetName()
cblogger.Infof("Name : [%s]", name)
info, err := vNetworkHandler.Client.Subnetworks.Delete(projectID, region, vNetworkID).Do()
cblogger.Info(info)
if err != nil {
cblogger.Error(err)
return false, err
}
//fmt.Println(info)
return true, nil
}