mirror of https://github.com/status-im/consul.git
130 lines
4.0 KiB
Go
130 lines
4.0 KiB
Go
package godo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
// StorageActionsService is an interface for interfacing with the
|
|
// storage actions endpoints of the Digital Ocean API.
|
|
// See: https://developers.digitalocean.com/documentation/v2#storage-actions
|
|
type StorageActionsService interface {
|
|
Attach(ctx context.Context, volumeID string, dropletID int) (*Action, *Response, error)
|
|
DetachByDropletID(ctx context.Context, volumeID string, dropletID int) (*Action, *Response, error)
|
|
Get(ctx context.Context, volumeID string, actionID int) (*Action, *Response, error)
|
|
List(ctx context.Context, volumeID string, opt *ListOptions) ([]Action, *Response, error)
|
|
Resize(ctx context.Context, volumeID string, sizeGigabytes int, regionSlug string) (*Action, *Response, error)
|
|
}
|
|
|
|
// StorageActionsServiceOp handles communication with the storage volumes
|
|
// action related methods of the DigitalOcean API.
|
|
type StorageActionsServiceOp struct {
|
|
client *Client
|
|
}
|
|
|
|
// StorageAttachment represents the attachement of a block storage
|
|
// volume to a specific Droplet under the device name.
|
|
type StorageAttachment struct {
|
|
DropletID int `json:"droplet_id"`
|
|
}
|
|
|
|
// Attach a storage volume to a Droplet.
|
|
func (s *StorageActionsServiceOp) Attach(ctx context.Context, volumeID string, dropletID int) (*Action, *Response, error) {
|
|
request := &ActionRequest{
|
|
"type": "attach",
|
|
"droplet_id": dropletID,
|
|
}
|
|
return s.doAction(ctx, volumeID, request)
|
|
}
|
|
|
|
// DetachByDropletID a storage volume from a Droplet by Droplet ID.
|
|
func (s *StorageActionsServiceOp) DetachByDropletID(ctx context.Context, volumeID string, dropletID int) (*Action, *Response, error) {
|
|
request := &ActionRequest{
|
|
"type": "detach",
|
|
"droplet_id": dropletID,
|
|
}
|
|
return s.doAction(ctx, volumeID, request)
|
|
}
|
|
|
|
// Get an action for a particular storage volume by id.
|
|
func (s *StorageActionsServiceOp) Get(ctx context.Context, volumeID string, actionID int) (*Action, *Response, error) {
|
|
path := fmt.Sprintf("%s/%d", storageAllocationActionPath(volumeID), actionID)
|
|
return s.get(ctx, path)
|
|
}
|
|
|
|
// List the actions for a particular storage volume.
|
|
func (s *StorageActionsServiceOp) List(ctx context.Context, volumeID string, opt *ListOptions) ([]Action, *Response, error) {
|
|
path := storageAllocationActionPath(volumeID)
|
|
path, err := addOptions(path, opt)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
return s.list(ctx, path)
|
|
}
|
|
|
|
// Resize a storage volume.
|
|
func (s *StorageActionsServiceOp) Resize(ctx context.Context, volumeID string, sizeGigabytes int, regionSlug string) (*Action, *Response, error) {
|
|
request := &ActionRequest{
|
|
"type": "resize",
|
|
"size_gigabytes": sizeGigabytes,
|
|
"region": regionSlug,
|
|
}
|
|
return s.doAction(ctx, volumeID, request)
|
|
}
|
|
|
|
func (s *StorageActionsServiceOp) doAction(ctx context.Context, volumeID string, request *ActionRequest) (*Action, *Response, error) {
|
|
path := storageAllocationActionPath(volumeID)
|
|
|
|
req, err := s.client.NewRequest(ctx, http.MethodPost, path, request)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
root := new(actionRoot)
|
|
resp, err := s.client.Do(ctx, req, root)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
|
|
return root.Event, resp, err
|
|
}
|
|
|
|
func (s *StorageActionsServiceOp) get(ctx context.Context, path string) (*Action, *Response, error) {
|
|
req, err := s.client.NewRequest(ctx, http.MethodGet, path, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
root := new(actionRoot)
|
|
resp, err := s.client.Do(ctx, req, root)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
|
|
return root.Event, resp, err
|
|
}
|
|
|
|
func (s *StorageActionsServiceOp) list(ctx context.Context, path string) ([]Action, *Response, error) {
|
|
req, err := s.client.NewRequest(ctx, http.MethodGet, path, nil)
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
|
|
root := new(actionsRoot)
|
|
resp, err := s.client.Do(ctx, req, root)
|
|
if err != nil {
|
|
return nil, resp, err
|
|
}
|
|
if l := root.Links; l != nil {
|
|
resp.Links = l
|
|
}
|
|
|
|
return root.Actions, resp, err
|
|
}
|
|
|
|
func storageAllocationActionPath(volumeID string) string {
|
|
return fmt.Sprintf("%s/%s/actions", storageAllocPath, volumeID)
|
|
}
|