-
Notifications
You must be signed in to change notification settings - Fork 300
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Introduce droplet autoscale godo methods
- Loading branch information
Showing
3 changed files
with
825 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,258 @@ | ||
| package godo | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "net/http" | ||
| "time" | ||
| ) | ||
|
|
||
| const ( | ||
| dropletAutoscaleBasePath = "/v2/droplets/autoscale" | ||
| ) | ||
|
|
||
| // DropletAutoscaleService defines an interface for managing droplet autoscale pools through DigitalOcean API | ||
| type DropletAutoscaleService interface { | ||
| Create(context.Context, *DropletAutoscalePoolRequest) (string, *Response, error) | ||
| Get(context.Context, string) (*DropletAutoscalePool, *Response, error) | ||
| List(context.Context, *ListOptions) ([]*DropletAutoscalePool, *Response, error) | ||
| ListMembers(context.Context, string, *ListOptions) ([]*DropletAutoscaleResource, *Response, error) | ||
| ListHistory(context.Context, string, *ListOptions) ([]*DropletAutoscaleHistoryEvent, *Response, error) | ||
| Update(context.Context, string, *DropletAutoscalePoolRequest) (string, *Response, error) | ||
| Delete(context.Context, string) (*Response, error) | ||
| DeleteDangerous(context.Context, string) (*Response, error) | ||
| } | ||
|
|
||
| // DropletAutoscalePool represents a DigitalOcean droplet autoscale pool | ||
| type DropletAutoscalePool struct { | ||
| ID string `json:"id"` | ||
| Name string `json:"name"` | ||
| Config *DropletAutoscaleConfiguration `json:"config"` | ||
| DropletTemplate *DropletAutoscaleResourceTemplate `json:"droplet_template"` | ||
| CreatedAt time.Time `json:"created_at"` | ||
| UpdatedAt time.Time `json:"updated_at"` | ||
| CurrentUtlization *DropletAutoscaleResourceUtilization `json:"current_utilization,omitempty"` | ||
| Status string `json:"status"` | ||
| } | ||
|
|
||
| // DropletAutoscaleConfiguration represents a DigitalOcean droplet autoscale pool configuration | ||
| type DropletAutoscaleConfiguration struct { | ||
| MinInstances uint64 `json:"min_instances,omitempty"` | ||
| MaxInstances uint64 `json:"max_instances,omitempty"` | ||
| TargetCPUUtilization float64 `json:"target_cpu_utilization,omitempty"` | ||
| TargetMemoryUtilization float64 `json:"target_memory_utilization,omitempty"` | ||
| CooldownMinutes uint32 `json:"cooldown_minutes,omitempty"` | ||
| TargetNumberInstances uint64 `json:"target_number_instances,omitempty"` | ||
| } | ||
|
|
||
| // DropletAutoscaleResourceTemplate represents a DigitalOcean droplet autoscale pool resource template | ||
| type DropletAutoscaleResourceTemplate struct { | ||
| SizeSlug string `json:"size"` | ||
| RegionSlug string `json:"region"` | ||
| Image string `json:"image"` | ||
| Tags []string `json:"tags"` | ||
| SSHKeys []string `json:"ssh_keys"` | ||
| VpcUUID string `json:"vpc_uuid"` | ||
| WithDropletAgent bool `json:"with_droplet_agent"` | ||
| ProjectID string `json:"project_id"` | ||
| IPV6 bool `json:"ipv6"` | ||
| UserData string `json:"user_data"` | ||
| } | ||
|
|
||
| // DropletAutoscaleResourceUtilization represents a DigitalOcean droplet autoscale pool resource utilization | ||
| type DropletAutoscaleResourceUtilization struct { | ||
| Memory float64 `json:"memory,omitempty"` | ||
| CPU float64 `json:"cpu,omitempty"` | ||
| } | ||
|
|
||
| // DropletAutoscaleResource represents a DigitalOcean droplet autoscale pool resource | ||
| type DropletAutoscaleResource struct { | ||
| DropletID uint64 `json:"droplet_id"` | ||
| CreatedAt time.Time `json:"created_at"` | ||
| UpdatedAt time.Time `json:"updated_at"` | ||
| HealthStatus string `json:"health_status"` | ||
| UnhealthyReason string `json:"unhealthy_reason,omitempty"` | ||
| Status string `json:"status"` | ||
| DropletAutoscaleResourceUtilization *DropletAutoscaleResourceUtilization `json:"current_utilization,omitempty"` | ||
| } | ||
|
|
||
| // DropletAutoscaleHistoryEvent represents a DigitalOcean droplet autoscale pool history event | ||
| type DropletAutoscaleHistoryEvent struct { | ||
| HistoryEventID string `json:"history_event_id"` | ||
| CurrentInstanceCount uint64 `json:"current_instance_count"` | ||
| DesiredInstanceCount uint64 `json:"desired_instance_count"` | ||
| Reason string `json:"reason"` | ||
| Status string `json:"status"` | ||
| ErrorReason string `json:"error_reason,omitempty"` | ||
| CreatedAt time.Time `json:"created_at"` | ||
| UpdatedAt time.Time `json:"updated_at"` | ||
| } | ||
|
|
||
| // DropletAutoscalePoolRequest represents a DigitalOcean droplet autoscale pool create/update request | ||
| type DropletAutoscalePoolRequest struct { | ||
| Name string `json:"name"` | ||
| AutoscalingConfig *DropletAutoscaleConfiguration `json:"config"` | ||
| DropletTemplate *DropletAutoscaleResourceTemplate `json:"droplet_template"` | ||
| } | ||
|
|
||
| type dropletAutoscalePoolRoot struct { | ||
| ID string `json:"id"` | ||
| } | ||
|
|
||
| type dropletAutoscalePoolsRoot struct { | ||
| AutoscalePools []*DropletAutoscalePool `json:"autoscale_pools"` | ||
| Links *Links `json:"links"` | ||
| Meta *Meta `json:"meta"` | ||
| } | ||
|
|
||
| type dropletAutoscaleMembersRoot struct { | ||
| Droplets []*DropletAutoscaleResource `json:"droplets"` | ||
| Links *Links `json:"links"` | ||
| Meta *Meta `json:"meta"` | ||
| } | ||
|
|
||
| type dropletAutoscaleHistoryEventsRoot struct { | ||
| History []*DropletAutoscaleHistoryEvent `json:"history"` | ||
| Links *Links `json:"links"` | ||
| Meta *Meta `json:"meta"` | ||
| } | ||
|
|
||
| // DropletAutoscaleServiceOp handles communication with droplet autoscale-related methods of the DigitalOcean API | ||
| type DropletAutoscaleServiceOp struct { | ||
| client *Client | ||
| } | ||
|
|
||
| var _ DropletAutoscaleService = &DropletAutoscaleServiceOp{} | ||
|
|
||
| // Create a new droplet autoscale pool | ||
| func (d *DropletAutoscaleServiceOp) Create(ctx context.Context, createReq *DropletAutoscalePoolRequest) (string, *Response, error) { | ||
| req, err := d.client.NewRequest(ctx, http.MethodPost, dropletAutoscaleBasePath, createReq) | ||
| if err != nil { | ||
| return "", nil, err | ||
| } | ||
| root := new(dropletAutoscalePoolRoot) | ||
| resp, err := d.client.Do(ctx, req, root) | ||
| if err != nil { | ||
| return "", nil, err | ||
| } | ||
| return root.ID, resp, nil | ||
| } | ||
|
|
||
| // Get an existing droplet autoscale pool | ||
| func (d *DropletAutoscaleServiceOp) Get(ctx context.Context, id string) (*DropletAutoscalePool, *Response, error) { | ||
| req, err := d.client.NewRequest(ctx, http.MethodGet, fmt.Sprintf("%s/%s", dropletAutoscaleBasePath, id), nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| root := new(DropletAutoscalePool) | ||
| resp, err := d.client.Do(ctx, req, root) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| return root, resp, err | ||
| } | ||
|
|
||
| // List all existing droplet autoscale pools | ||
| func (d *DropletAutoscaleServiceOp) List(ctx context.Context, opts *ListOptions) ([]*DropletAutoscalePool, *Response, error) { | ||
| path, err := addOptions(dropletAutoscaleBasePath, opts) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| req, err := d.client.NewRequest(ctx, http.MethodGet, path, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| root := new(dropletAutoscalePoolsRoot) | ||
| resp, err := d.client.Do(ctx, req, root) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| if root.Links != nil { | ||
| resp.Links = root.Links | ||
| } | ||
| if root.Meta != nil { | ||
| resp.Meta = root.Meta | ||
| } | ||
| return root.AutoscalePools, resp, err | ||
| } | ||
|
|
||
| // ListMembers all members for an existing droplet autoscale pool | ||
| func (d *DropletAutoscaleServiceOp) ListMembers(ctx context.Context, id string, opts *ListOptions) ([]*DropletAutoscaleResource, *Response, error) { | ||
| path, err := addOptions(fmt.Sprintf("%s/%s/members", dropletAutoscaleBasePath, id), opts) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| req, err := d.client.NewRequest(ctx, http.MethodGet, path, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| root := new(dropletAutoscaleMembersRoot) | ||
| resp, err := d.client.Do(ctx, req, root) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| if root.Links != nil { | ||
| resp.Links = root.Links | ||
| } | ||
| if root.Meta != nil { | ||
| resp.Meta = root.Meta | ||
| } | ||
| return root.Droplets, resp, err | ||
| } | ||
|
|
||
| // ListHistory all history events for an existing droplet autoscale pool | ||
| func (d *DropletAutoscaleServiceOp) ListHistory(ctx context.Context, id string, opts *ListOptions) ([]*DropletAutoscaleHistoryEvent, *Response, error) { | ||
| path, err := addOptions(fmt.Sprintf("%s/%s/history", dropletAutoscaleBasePath, id), opts) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| req, err := d.client.NewRequest(ctx, http.MethodGet, path, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| root := new(dropletAutoscaleHistoryEventsRoot) | ||
| resp, err := d.client.Do(ctx, req, root) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
| if root.Links != nil { | ||
| resp.Links = root.Links | ||
| } | ||
| if root.Meta != nil { | ||
| resp.Meta = root.Meta | ||
| } | ||
| return root.History, resp, err | ||
| } | ||
|
|
||
| // Update an existing autoscale pool | ||
| func (d *DropletAutoscaleServiceOp) Update(ctx context.Context, id string, updateReq *DropletAutoscalePoolRequest) (string, *Response, error) { | ||
| req, err := d.client.NewRequest(ctx, http.MethodPut, fmt.Sprintf("%s/%s", dropletAutoscaleBasePath, id), updateReq) | ||
| if err != nil { | ||
| return "", nil, err | ||
| } | ||
| root := new(dropletAutoscalePoolRoot) | ||
| resp, err := d.client.Do(ctx, req, root) | ||
| if err != nil { | ||
| return "", nil, err | ||
| } | ||
| return root.ID, resp, nil | ||
| } | ||
|
|
||
| // Delete an existing autoscale pool | ||
| func (d *DropletAutoscaleServiceOp) Delete(ctx context.Context, id string) (*Response, error) { | ||
| req, err := d.client.NewRequest(ctx, http.MethodDelete, fmt.Sprintf("%s/%s", dropletAutoscaleBasePath, id), nil) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return d.client.Do(ctx, req, nil) | ||
| } | ||
|
|
||
| // DeleteDangerous deletes an existing autoscale pool with all underlying resources | ||
| func (d *DropletAutoscaleServiceOp) DeleteDangerous(ctx context.Context, id string) (*Response, error) { | ||
| req, err := d.client.NewRequest(ctx, http.MethodDelete, fmt.Sprintf("%s/%s", dropletAutoscaleBasePath, id), nil) | ||
| req.Header.Set("X-Dangerous", "true") | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| return d.client.Do(ctx, req, nil) | ||
| } |
Oops, something went wrong.