This repository has been archived by the owner on Jul 6, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 100
/
provision.go
124 lines (106 loc) · 3.35 KB
/
provision.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 rediscache
import (
"context"
"encoding/json"
"fmt"
"github.com/Azure/open-service-broker-azure/pkg/service"
uuid "github.com/satori/go.uuid"
)
const enabled = "enabled"
func (s *serviceManager) GetProvisioner(
service.Plan,
) (service.Provisioner, error) {
return service.NewProvisioner(
service.NewProvisioningStep("preProvision", s.preProvision),
service.NewProvisioningStep("deployARMTemplate", s.deployARMTemplate),
)
}
func (s *serviceManager) preProvision(
context.Context,
service.Instance,
) (service.InstanceDetails, error) {
return &instanceDetails{
ARMDeploymentName: uuid.NewV4().String(),
ServerName: uuid.NewV4().String(),
}, nil
}
func (s *serviceManager) deployARMTemplate(
_ context.Context,
instance service.Instance,
) (service.InstanceDetails, error) {
dt := instance.Details.(*instanceDetails)
tagsObj := instance.ProvisioningParameters.GetObject("tags")
tags := make(map[string]string, len(tagsObj.Data))
for k := range tagsObj.Data {
tags[k] = tagsObj.GetString(k)
}
outputs, err := s.armDeployer.Deploy(
dt.ARMDeploymentName,
instance.ProvisioningParameters.GetString("resourceGroup"),
instance.ProvisioningParameters.GetString("location"),
armTemplateBytes,
buildGoTemplate(instance, provision),
map[string]interface{}{},
tags,
)
if err != nil {
return nil, fmt.Errorf("error deploying ARM template: %s", err)
}
var ok bool
dt.FullyQualifiedDomainName, ok = outputs["fullyQualifiedDomainName"].(string)
if !ok {
return nil, fmt.Errorf(
"error retrieving fully qualified domain name from deployment: %s",
err,
)
}
primaryKey, ok := outputs["primaryKey"].(string)
if !ok {
return nil, fmt.Errorf(
"error retrieving primary key from deployment: %s",
err,
)
}
dt.PrimaryKey = service.SecureString(primaryKey)
dt.NonSSLEnabled = (instance.ProvisioningParameters.GetString("enableNonSslPort") == enabled) // nolint: lll
return dt, err
}
const provision = "provision"
const update = "update"
func buildGoTemplate(
instance service.Instance,
mode string,
) map[string]interface{} {
var pp *service.ProvisioningParameters
if mode == provision {
pp = instance.ProvisioningParameters
} else if mode == update {
pp = instance.UpdatingParameters
}
dt := instance.Details.(*instanceDetails)
plan := instance.Plan
var enableNonSslPort string
if pp.GetString("enableNonSslPort") == enabled {
enableNonSslPort = "true"
} else {
enableNonSslPort = "false"
}
redisConfiguration := pp.GetObject("redisConfiguration").Data
if value, ok := redisConfiguration["rdb-backup-enabled"]; ok {
redisConfiguration["rdb-backup-enabled"] = (value == enabled)
}
redisConfigurationBytes, _ := json.Marshal(redisConfiguration)
subnetSettings := pp.GetObject("subnetSettings")
return map[string]interface{}{ // ARM template params
"location": pp.GetString("location"),
"serverName": dt.ServerName,
"enableNonSslPort": enableNonSslPort,
"redisCacheSKU": plan.GetProperties().Extended["redisCacheSKU"],
"redisCacheFamily": plan.GetProperties().Extended["redisCacheFamily"],
"redisCacheCapacity": pp.GetInt64("skuCapacity"),
"redisConfiguration": string(redisConfigurationBytes),
"shardCount": pp.GetInt64("shardCount"),
"subnetId": subnetSettings.GetString("subnetId"),
"staticIP": subnetSettings.GetString("staticIP"),
}
}