forked from cloudfoundry/bosh-cli
-
Notifications
You must be signed in to change notification settings - Fork 0
/
vm.go
225 lines (189 loc) · 5.96 KB
/
vm.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
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
package vm
import (
"time"
biagentclient "github.com/cloudfoundry/bosh-agent/agentclient"
bias "github.com/cloudfoundry/bosh-agent/agentclient/applyspec"
bicloud "github.com/cloudfoundry/bosh-cli/cloud"
biconfig "github.com/cloudfoundry/bosh-cli/config"
bidisk "github.com/cloudfoundry/bosh-cli/deployment/disk"
bideplmanifest "github.com/cloudfoundry/bosh-cli/deployment/manifest"
biui "github.com/cloudfoundry/bosh-cli/ui"
bosherr "github.com/cloudfoundry/bosh-utils/errors"
boshlog "github.com/cloudfoundry/bosh-utils/logger"
boshretry "github.com/cloudfoundry/bosh-utils/retrystrategy"
boshsys "github.com/cloudfoundry/bosh-utils/system"
"github.com/pivotal-golang/clock"
)
type VM interface {
CID() string
Exists() (bool, error)
AgentClient() biagentclient.AgentClient
WaitUntilReady(timeout time.Duration, delay time.Duration) error
Start() error
Stop() error
Apply(bias.ApplySpec) error
UpdateDisks(bideplmanifest.DiskPool, biui.Stage) ([]bidisk.Disk, error)
WaitToBeRunning(maxAttempts int, delay time.Duration) error
AttachDisk(bidisk.Disk) error
DetachDisk(bidisk.Disk) error
Disks() ([]bidisk.Disk, error)
UnmountDisk(bidisk.Disk) error
MigrateDisk() error
RunScript(script string, options map[string]interface{}) error
Delete() error
GetState() (biagentclient.AgentState, error)
}
type vm struct {
cid string
vmRepo biconfig.VMRepo
stemcellRepo biconfig.StemcellRepo
diskDeployer DiskDeployer
agentClient biagentclient.AgentClient
cloud bicloud.Cloud
fs boshsys.FileSystem
logger boshlog.Logger
logTag string
}
func NewVM(
cid string,
vmRepo biconfig.VMRepo,
stemcellRepo biconfig.StemcellRepo,
diskDeployer DiskDeployer,
agentClient biagentclient.AgentClient,
cloud bicloud.Cloud,
fs boshsys.FileSystem,
logger boshlog.Logger,
) VM {
return &vm{
cid: cid,
vmRepo: vmRepo,
stemcellRepo: stemcellRepo,
diskDeployer: diskDeployer,
agentClient: agentClient,
cloud: cloud,
fs: fs,
logger: logger,
logTag: "vm",
}
}
func (vm *vm) CID() string {
return vm.cid
}
func (vm *vm) Exists() (bool, error) {
exists, err := vm.cloud.HasVM(vm.cid)
if err != nil {
return false, bosherr.WrapErrorf(err, "Checking existance of VM '%s'", vm.cid)
}
return exists, nil
}
func (vm *vm) AgentClient() biagentclient.AgentClient {
return vm.agentClient
}
func (vm *vm) WaitUntilReady(timeout time.Duration, delay time.Duration) error {
agentPingRetryable := biagentclient.NewPingRetryable(vm.agentClient)
timeService := clock.NewClock() //TODO: inject timeService
agentPingRetryStrategy := boshretry.NewTimeoutRetryStrategy(timeout, delay, agentPingRetryable, timeService, vm.logger)
return agentPingRetryStrategy.Try()
}
func (vm *vm) Start() error {
vm.logger.Debug(vm.logTag, "Starting agent")
err := vm.agentClient.Start()
if err != nil {
return bosherr.WrapError(err, "Starting agent")
}
return nil
}
func (vm *vm) Stop() error {
vm.logger.Debug(vm.logTag, "Stopping agent")
err := vm.agentClient.Stop()
if err != nil {
return bosherr.WrapError(err, "Stopping agent")
}
return nil
}
func (vm *vm) Apply(newState bias.ApplySpec) error {
vm.logger.Debug(vm.logTag, "Sending apply message to the agent with '%#v'", newState)
err := vm.agentClient.Apply(newState)
if err != nil {
return bosherr.WrapError(err, "Sending apply spec to agent")
}
return nil
}
func (vm *vm) UpdateDisks(diskPool bideplmanifest.DiskPool, eventLoggerStage biui.Stage) ([]bidisk.Disk, error) {
disks, err := vm.diskDeployer.Deploy(diskPool, vm.cloud, vm, eventLoggerStage)
if err != nil {
return disks, bosherr.WrapError(err, "Deploying disk")
}
return disks, nil
}
func (vm *vm) WaitToBeRunning(maxAttempts int, delay time.Duration) error {
agentGetStateRetryable := biagentclient.NewGetStateRetryable(vm.agentClient)
agentGetStateRetryStrategy := boshretry.NewAttemptRetryStrategy(maxAttempts, delay, agentGetStateRetryable, vm.logger)
return agentGetStateRetryStrategy.Try()
}
func (vm *vm) AttachDisk(disk bidisk.Disk) error {
err := vm.cloud.AttachDisk(vm.cid, disk.CID())
if err != nil {
return bosherr.WrapError(err, "Attaching disk in the cloud")
}
err = vm.agentClient.MountDisk(disk.CID())
if err != nil {
return bosherr.WrapError(err, "Mounting disk")
}
return nil
}
func (vm *vm) DetachDisk(disk bidisk.Disk) error {
err := vm.cloud.DetachDisk(vm.cid, disk.CID())
if err != nil {
return bosherr.WrapError(err, "Detaching disk in the cloud")
}
return nil
}
func (vm *vm) Disks() ([]bidisk.Disk, error) {
result := []bidisk.Disk{}
disks, err := vm.agentClient.ListDisk()
if err != nil {
return result, bosherr.WrapError(err, "Listing vm disks")
}
for _, diskCID := range disks {
disk := bidisk.NewDisk(biconfig.DiskRecord{CID: diskCID}, nil, nil)
result = append(result, disk)
}
return result, nil
}
func (vm *vm) UnmountDisk(disk bidisk.Disk) error {
return vm.agentClient.UnmountDisk(disk.CID())
}
func (vm *vm) MigrateDisk() error {
return vm.agentClient.MigrateDisk()
}
func (vm *vm) RunScript(script string, options map[string]interface{}) error {
return vm.agentClient.RunScript(script, options)
}
func (vm *vm) Delete() error {
deleteErr := vm.cloud.DeleteVM(vm.cid)
if deleteErr != nil {
// allow VMNotFoundError for idempotency
cloudErr, ok := deleteErr.(bicloud.Error)
if !ok || cloudErr.Type() != bicloud.VMNotFoundError {
return bosherr.WrapError(deleteErr, "Deleting vm in the cloud")
}
}
err := vm.vmRepo.ClearCurrent()
if err != nil {
return bosherr.WrapError(err, "Deleting vm from vm repo")
}
err = vm.stemcellRepo.ClearCurrent()
if err != nil {
return bosherr.WrapError(err, "Clearing current stemcell from stemcell repo")
}
// returns bicloud.Error only if it is a VMNotFoundError
return deleteErr
}
func (vm *vm) GetState() (biagentclient.AgentState, error) {
agentState, err := vm.agentClient.GetState()
if err != nil {
return agentState, bosherr.WrapError(err, "Getting vm state")
}
return agentState, nil
}