This repository has been archived by the owner on Mar 24, 2022. It is now read-only.
/
client.go
78 lines (65 loc) · 1.69 KB
/
client.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
package cliaas
import (
"github.com/aws/aws-sdk-go/service/ec2"
"github.com/pivotal-cf/cliaas/iaas"
"github.com/pivotal-cf/cliaas/iaas/aws"
)
type Client interface {
Delete(vmIdentifier string) error
Replace(vmIdentifier string, imageIdentifier string, diskSizeGB int64) error
GetDisk(vmIdentifier string) (iaas.Disk, error)
}
func NewAWSAPIClient(client aws.AWSClient) Client {
return &awsAPIClient{
client: client,
}
}
type awsAPIClient struct {
client aws.AWSClient
}
func (c *awsAPIClient) Delete(identifier string) error {
return c.client.DeleteVM(identifier)
}
func (c *awsAPIClient) Replace(identifier string, ami string, diskSizeGB int64) error {
vmInfo, err := c.client.GetVMInfo(identifier + "*")
if err != nil {
return err
}
err = c.client.StopVM(vmInfo.InstanceID)
if err != nil {
_ = c.client.StartVM(vmInfo.InstanceID)
return err
}
err = c.client.WaitForStatus(vmInfo.InstanceID, ec2.InstanceStateNameStopped)
if err != nil {
_ = c.client.StartVM(vmInfo.InstanceID)
return err
}
instanceID, err := c.client.CreateVM(
ami,
identifier,
vmInfo,
)
if err != nil {
_ = c.client.StartVM(vmInfo.InstanceID)
return err
}
err = c.client.WaitForStatus(instanceID, ec2.InstanceStateNameRunning)
if err != nil {
_ = c.client.DeleteVM(instanceID)
return err
}
if vmInfo.PublicIP != "" {
err = c.client.AssignPublicIP(instanceID, vmInfo.PublicIP)
if err != nil {
_ = c.client.DeleteVM(instanceID)
_ = c.client.AssignPublicIP(vmInfo.InstanceID, vmInfo.PublicIP)
_ = c.client.StartVM(vmInfo.InstanceID)
return err
}
}
return nil
}
func (c *awsAPIClient) GetDisk(identifier string) (iaas.Disk, error) {
return iaas.Disk{SizeGB: int64(0)}, nil
}