Add worker pool API to sdk

api/container/containerv1/api_service.go

@@ -18,6 +18,7 @@ const ErrCodeAPICreation = "APICreationError"
 type ContainerServiceAPI interface {
 	Clusters() Clusters
 	Workers() Workers
+	WorkerPools() WorkerPool
 	WebHooks() Webhooks
 	Subnets() Subnets
 	KubeVersions() KubeVersions
@@ -76,6 +77,11 @@ func (c *csService) Workers() Workers {
 	return newWorkerAPI(c.Client)
 }
 
+//WorkerPools implements Cluster WorkerPools API
+func (c *csService) WorkerPools() WorkerPool {
+	return newWorkerPoolAPI(c.Client)
+}
+
 //Subnets implements Cluster Subnets API
 func (c *csService) Subnets() Subnets {
 	return newSubnetAPI(c.Client)

api/container/containerv1/worker_pool.go

@@ -0,0 +1,175 @@
+package containerv1
+
+import (
+	"fmt"
+
+	"github.com/IBM-Cloud/bluemix-go/client"
+)
+
+// WorkerPool common worker pool data
+type WorkerPoolConfig struct {
+	Name        string            `json:"name" binding:"required"`
+	Size        int               `json:"sizePerZone" binding:"required"`
+	MachineType string            `json:"machineType" binding:"required"`
+	Isolation   string            `json:"isolation"`
+	Labels      map[string]string `json:"labels"`
+}
+
+// WorkerPoolRequest provides worker pool data
+// swagger:model
+type WorkerPoolRequest struct {
+	WorkerPoolConfig
+	DiskEncryption bool             `json:"diskEncryption" description:"true or false to use encryption for the secondary disk"`
+	Zones          []WorkerPoolZone `json:"zones"`
+}
+
+// WorkerPoolPatchRequest provides attributes to patch update worker pool
+// swagger:model
+type WorkerPoolPatchRequest struct {
+	Size            int    `json:"sizePerZone"`
+	ReasonForResize string `json:"reasonForResize"`
+	State           string `json:"state"`
+}
+
+// WorkerPoolResponse provides worker pool data
+// swagger:model
+type WorkerPoolResponse struct {
+	WorkerPoolConfig
+	ID              string                  `json:"id" binding:"required"`
+	Region          string                  `json:"region" binding:"required"`
+	State           string                  `json:"state"`
+	WorkerVersion   string                  `json:"kubeVersion"`
+	TargetVersion   string                  `json:"targetVersion"`
+	EOS             string                  `json:"versionEOS"`
+	MasterEOS       string                  `json:"masterVersionEOS"`
+	ReasonForDelete string                  `json:"reasonForDelete"`
+	IsBalanced      bool                    `json:"isBalanced"`
+	Zones           WorkerPoolZoneResponses `json:"zones"`
+}
+
+// WorkerPoolResponses sorts WorkerPoolResponse by ID.
+// swagger:model
+type WorkerPoolResponses []WorkerPoolResponse
+
+// WorkerPoolZoneNetwork holds network configuration for a zone
+type WorkerPoolZoneNetwork struct {
+	PrivateVLAN string `json:"privateVlan" binding:"required"`
+	PublicVLAN  string `json:"publicVlan"`
+}
+
+// WorkerPoolZone provides zone data
+// swagger:model
+type WorkerPoolZone struct {
+	WorkerPoolZoneNetwork
+	ID string `json:"id" binding:"required"`
+}
+
+// WorkerPoolZonePatchRequest updates worker pool zone data
+// swagger:model
+type WorkerPoolZonePatchRequest struct {
+	WorkerPoolZoneNetwork
+}
+
+// WorkerPoolZoneResponse response contents for zone
+// swagger:model
+type WorkerPoolZoneResponse struct {
+	WorkerPoolZone
+	WorkerCount int `json:"workerCount"`
+}
+
+// WorkerPoolZoneResponses sorts WorkerPoolZoneResponse by ID.
+// swagger:model
+type WorkerPoolZoneResponses []WorkerPoolZoneResponse
+
+//Workers ...
+type WorkerPool interface {
+	CreateWorkerPool(clusterNameOrID string, workerPoolReq WorkerPoolRequest) (WorkerPoolResponse, error)
+	ResizeWorkerPool(clusterNameOrID, workerPoolNameOrID string, size int) error
+	PatchWorkerPool(clusterNameOrID, workerPoolNameOrID, state string) error
+	DeleteWorkerPool(clusterNameOrID string, workerPoolNameOrID string) error
+	ListWorkerPools(clusterNameOrID string) ([]WorkerPoolResponse, error)
+	GetWorkerPool(clusterNameOrID, workerPoolNameOrID string) (WorkerPoolResponse, error)
+	AddZone(clusterNameOrID string, poolID string, workerPoolZone WorkerPoolZone) error
+	RemoveZone(clusterNameOrID, zone, poolID string) error
+	UpdateZoneNetwork(clusterNameOrID, zone, poolID, privateVlan, publicVlan string) error
+}
+
+type workerpool struct {
+	client *client.Client
+}
+
+func newWorkerPoolAPI(c *client.Client) WorkerPool {
+	return &workerpool{
+		client: c,
+	}
+}
+
+// CreateWorkerPool calls the API to create a worker pool
+func (w *workerpool) CreateWorkerPool(clusterNameOrID string, workerPoolReq WorkerPoolRequest) (WorkerPoolResponse, error) {
+	var successV WorkerPoolResponse
+	_, err := w.client.Post(fmt.Sprintf("/v1/clusters/%s/workerpools", clusterNameOrID), workerPoolReq, &successV)
+	return successV, err
+}
+
+// ResizeWorkerPool calls the API to resize a worker
+func (w *workerpool) PatchWorkerPool(clusterNameOrID, workerPoolNameOrID, state string) error {
+	requestBody := WorkerPoolPatchRequest{
+		State: state,
+	}
+	_, err := w.client.Patch(fmt.Sprintf("/v1/clusters/%s/workerpools/%s", clusterNameOrID, workerPoolNameOrID), requestBody, nil)
+	return err
+}
+
+// ResizeWorkerPool calls the API to resize a worker
+func (w *workerpool) ResizeWorkerPool(clusterNameOrID, workerPoolNameOrID string, size int) error {
+	requestBody := WorkerPoolPatchRequest{
+		State: "resizing",
+		Size:  size,
+	}
+	_, err := w.client.Patch(fmt.Sprintf("/v1/clusters/%s/workerpools/%s", clusterNameOrID, workerPoolNameOrID), requestBody, nil)
+	return err
+}
+
+// DeleteWorkerPool calls the API to remove a worker pool
+func (w *workerpool) DeleteWorkerPool(clusterNameOrID string, workerPoolNameOrID string) error {
+	// Make the request, don't care about return value
+	_, err := w.client.Delete(fmt.Sprintf("/v1/clusters/%s/workerpools/%s", clusterNameOrID, workerPoolNameOrID))
+	return err
+}
+
+// ListWorkerPools calls the API to list all worker pools for a cluster
+func (w *workerpool) ListWorkerPools(clusterNameOrID string) ([]WorkerPoolResponse, error) {
+	var successV []WorkerPoolResponse
+	_, err := w.client.Get(fmt.Sprintf("/v1/clusters/%s/workerpools", clusterNameOrID), &successV)
+	return successV, err
+}
+
+// GetWorkerPool calls the API to get a worker pool
+func (w *workerpool) GetWorkerPool(clusterNameOrID, workerPoolNameOrID string) (WorkerPoolResponse, error) {
+	var successV WorkerPoolResponse
+	_, err := w.client.Get(fmt.Sprintf("/v1/clusters/%s/workerpools/%s", clusterNameOrID, workerPoolNameOrID), &successV)
+	return successV, err
+}
+
+// AddZone calls the API to add a zone to a cluster and worker pool
+func (w *workerpool) AddZone(clusterNameOrID string, poolID string, workerPoolZone WorkerPoolZone) error {
+	// Make the request, don't care about return value
+	_, err := w.client.Post(fmt.Sprintf("/v1/clusters/%s/workerpools/%s/zones", clusterNameOrID, poolID), workerPoolZone, nil)
+	return err
+}
+
+// RemoveZone calls the API to remove a zone from a worker pool in a cluster
+func (w *workerpool) RemoveZone(clusterNameOrID, zone, poolID string) error {
+	_, err := w.client.Delete(fmt.Sprintf("/v1/clusters/%s/workerpools/%s/zones/%s", clusterNameOrID, poolID, zone))
+	return err
+}
+
+// UpdateZoneNetwork calls the API to update a zone's network
+func (w *workerpool) UpdateZoneNetwork(clusterNameOrID, zone, poolID, privateVlan, publicVlan string) error {
+	body := WorkerPoolZoneNetwork{
+		PrivateVLAN: privateVlan,
+		PublicVLAN:  publicVlan,
+	}
+	_, err := w.client.Patch(fmt.Sprintf("/v1/clusters/%s/workerpools/%s/zones/%s", clusterNameOrID, poolID, zone), body, nil)
+	return err
+}