forked from cloudfoundry-attic/bosh-init
-
Notifications
You must be signed in to change notification settings - Fork 0
/
disk.go
76 lines (62 loc) · 1.68 KB
/
disk.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
package disk
import (
"reflect"
bosherr "github.com/cloudfoundry/bosh-agent/errors"
bicloud "github.com/cloudfoundry/bosh-init/cloud"
biproperty "github.com/cloudfoundry/bosh-init/common/property"
biconfig "github.com/cloudfoundry/bosh-init/config"
)
type Disk interface {
CID() string
NeedsMigration(newSize int, newCloudProperties biproperty.Map) bool
Delete() error
}
type disk struct {
cid string
size int
cloudProperties biproperty.Map
cloud bicloud.Cloud
repo biconfig.DiskRepo
}
func NewDisk(
diskRecord biconfig.DiskRecord,
cloud bicloud.Cloud,
repo biconfig.DiskRepo,
) Disk {
return &disk{
cid: diskRecord.CID,
size: diskRecord.Size,
cloudProperties: diskRecord.CloudProperties,
cloud: cloud,
repo: repo,
}
}
func (d *disk) CID() string {
return d.cid
}
func (d *disk) NeedsMigration(newSize int, newCloudProperties biproperty.Map) bool {
return d.size != newSize || !reflect.DeepEqual(d.cloudProperties, newCloudProperties)
}
func (d *disk) Delete() error {
deleteErr := d.cloud.DeleteDisk(d.cid)
if deleteErr != nil {
// allow DiskNotFoundError for idempotency
cloudErr, ok := deleteErr.(bicloud.Error)
if !ok || cloudErr.Type() != bicloud.DiskNotFoundError {
return bosherr.WrapError(deleteErr, "Deleting disk in the cloud")
}
}
diskRecord, found, err := d.repo.Find(d.cid)
if err != nil {
return bosherr.WrapErrorf(err, "Finding disk record (cid=%s)", d.cid)
}
if !found {
return nil
}
err = d.repo.Delete(diskRecord)
if err != nil {
return bosherr.WrapError(err, "Deleting disk record")
}
// returns bicloud.Error only if it is a DiskNotFoundError
return deleteErr
}