terraform/vendor/github.com/terraform-providers/terraform-provider-openstack/openstack/resource_openstack_lb_pool_...

377 lines
9.9 KiB
Go

package openstack
import (
"fmt"
"log"
"time"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/helper/schema"
"github.com/gophercloud/gophercloud/openstack/networking/v2/extensions/lbaas_v2/listeners"
"github.com/gophercloud/gophercloud/openstack/networking/v2/extensions/lbaas_v2/pools"
)
func resourcePoolV2() *schema.Resource {
return &schema.Resource{
Create: resourcePoolV2Create,
Read: resourcePoolV2Read,
Update: resourcePoolV2Update,
Delete: resourcePoolV2Delete,
Importer: &schema.ResourceImporter{
State: resourcePoolV2Import,
},
Timeouts: &schema.ResourceTimeout{
Create: schema.DefaultTimeout(10 * time.Minute),
Update: schema.DefaultTimeout(10 * time.Minute),
Delete: schema.DefaultTimeout(10 * time.Minute),
},
Schema: map[string]*schema.Schema{
"region": {
Type: schema.TypeString,
Optional: true,
Computed: true,
ForceNew: true,
},
"tenant_id": {
Type: schema.TypeString,
Optional: true,
Computed: true,
ForceNew: true,
},
"name": {
Type: schema.TypeString,
Optional: true,
},
"description": {
Type: schema.TypeString,
Optional: true,
},
"protocol": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
ValidateFunc: func(v interface{}, k string) (ws []string, errors []error) {
value := v.(string)
if value != "TCP" && value != "HTTP" && value != "HTTPS" && value != "PROXY" {
errors = append(errors, fmt.Errorf(
"Only 'TCP', 'HTTP','HTTPS', and 'PROXY' are supported values for 'protocol'"))
}
return
},
},
// One of loadbalancer_id or listener_id must be provided
"loadbalancer_id": {
Type: schema.TypeString,
Optional: true,
ForceNew: true,
},
// One of loadbalancer_id or listener_id must be provided
"listener_id": {
Type: schema.TypeString,
Optional: true,
ForceNew: true,
},
"lb_method": {
Type: schema.TypeString,
Required: true,
ValidateFunc: func(v interface{}, k string) (ws []string, errors []error) {
value := v.(string)
if value != "ROUND_ROBIN" && value != "LEAST_CONNECTIONS" && value != "SOURCE_IP" {
errors = append(errors, fmt.Errorf(
"Only 'ROUND_ROBIN', 'LEAST_CONNECTIONS', and 'SOURCE_IP' are supported values for 'lb_method'"))
}
return
},
},
"persistence": {
Type: schema.TypeList,
Optional: true,
ForceNew: true,
Elem: &schema.Resource{
Schema: map[string]*schema.Schema{
"type": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
ValidateFunc: func(v interface{}, k string) (ws []string, errors []error) {
value := v.(string)
if value != "SOURCE_IP" && value != "HTTP_COOKIE" && value != "APP_COOKIE" {
errors = append(errors, fmt.Errorf(
"Only 'SOURCE_IP', 'HTTP_COOKIE', and 'APP_COOKIE' are supported values for 'persistence'"))
}
return
},
},
"cookie_name": {
Type: schema.TypeString,
Optional: true,
ForceNew: true,
},
},
},
},
"admin_state_up": {
Type: schema.TypeBool,
Default: true,
Optional: true,
},
},
}
}
func resourcePoolV2Create(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
lbClient, err := chooseLBV2Client(d, config)
if err != nil {
return fmt.Errorf("Error creating OpenStack networking client: %s", err)
}
adminStateUp := d.Get("admin_state_up").(bool)
lbID := d.Get("loadbalancer_id").(string)
listenerID := d.Get("listener_id").(string)
var persistence pools.SessionPersistence
if p, ok := d.GetOk("persistence"); ok {
pV := (p.([]interface{}))[0].(map[string]interface{})
persistence = pools.SessionPersistence{
Type: pV["type"].(string),
}
if persistence.Type == "APP_COOKIE" {
if pV["cookie_name"].(string) == "" {
return fmt.Errorf(
"Persistence cookie_name needs to be set if using 'APP_COOKIE' persistence type.")
} else {
persistence.CookieName = pV["cookie_name"].(string)
}
} else {
if pV["cookie_name"].(string) != "" {
return fmt.Errorf(
"Persistence cookie_name can only be set if using 'APP_COOKIE' persistence type.")
}
}
}
createOpts := pools.CreateOpts{
TenantID: d.Get("tenant_id").(string),
Name: d.Get("name").(string),
Description: d.Get("description").(string),
Protocol: pools.Protocol(d.Get("protocol").(string)),
LoadbalancerID: lbID,
ListenerID: listenerID,
LBMethod: pools.LBMethod(d.Get("lb_method").(string)),
AdminStateUp: &adminStateUp,
}
// Must omit if not set
if persistence != (pools.SessionPersistence{}) {
createOpts.Persistence = &persistence
}
log.Printf("[DEBUG] Create Options: %#v", createOpts)
timeout := d.Timeout(schema.TimeoutCreate)
// Wait for Listener or LoadBalancer to become active before continuing
if listenerID != "" {
listener, err := listeners.Get(lbClient, listenerID).Extract()
if err != nil {
return err
}
err = waitForLBV2Listener(lbClient, listener, "ACTIVE", lbPendingStatuses, timeout)
} else {
err = waitForLBV2LoadBalancer(lbClient, lbID, "ACTIVE", lbPendingStatuses, timeout)
}
if err != nil {
return err
}
log.Printf("[DEBUG] Attempting to create pool")
var pool *pools.Pool
err = resource.Retry(timeout, func() *resource.RetryError {
pool, err = pools.Create(lbClient, createOpts).Extract()
if err != nil {
return checkForRetryableError(err)
}
return nil
})
if err != nil {
return fmt.Errorf("Error creating pool: %s", err)
}
// Pool was successfully created
// Wait for pool to become active before continuing
err = waitForLBV2Pool(lbClient, pool, "ACTIVE", lbPendingStatuses, timeout)
if err != nil {
return err
}
d.SetId(pool.ID)
return resourcePoolV2Read(d, meta)
}
func resourcePoolV2Read(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
lbClient, err := chooseLBV2Client(d, config)
if err != nil {
return fmt.Errorf("Error creating OpenStack networking client: %s", err)
}
pool, err := pools.Get(lbClient, d.Id()).Extract()
if err != nil {
return CheckDeleted(d, err, "pool")
}
log.Printf("[DEBUG] Retrieved pool %s: %#v", d.Id(), pool)
d.Set("lb_method", pool.LBMethod)
d.Set("protocol", pool.Protocol)
d.Set("description", pool.Description)
d.Set("tenant_id", pool.TenantID)
d.Set("admin_state_up", pool.AdminStateUp)
d.Set("name", pool.Name)
d.Set("persistence", pool.Persistence)
d.Set("region", GetRegion(d, config))
return nil
}
func resourcePoolV2Update(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
lbClient, err := chooseLBV2Client(d, config)
if err != nil {
return fmt.Errorf("Error creating OpenStack networking client: %s", err)
}
var updateOpts pools.UpdateOpts
if d.HasChange("lb_method") {
updateOpts.LBMethod = pools.LBMethod(d.Get("lb_method").(string))
}
if d.HasChange("name") {
name := d.Get("name").(string)
updateOpts.Name = &name
}
if d.HasChange("description") {
description := d.Get("description").(string)
updateOpts.Description = &description
}
if d.HasChange("admin_state_up") {
asu := d.Get("admin_state_up").(bool)
updateOpts.AdminStateUp = &asu
}
timeout := d.Timeout(schema.TimeoutUpdate)
// Get a clean copy of the pool.
pool, err := pools.Get(lbClient, d.Id()).Extract()
if err != nil {
return fmt.Errorf("Unable to retrieve pool %s: %s", d.Id(), err)
}
// Wait for pool to become active before continuing
err = waitForLBV2Pool(lbClient, pool, "ACTIVE", lbPendingStatuses, timeout)
if err != nil {
return err
}
log.Printf("[DEBUG] Updating pool %s with options: %#v", d.Id(), updateOpts)
err = resource.Retry(timeout, func() *resource.RetryError {
_, err = pools.Update(lbClient, d.Id(), updateOpts).Extract()
if err != nil {
return checkForRetryableError(err)
}
return nil
})
if err != nil {
return fmt.Errorf("Unable to update pool %s: %s", d.Id(), err)
}
// Wait for pool to become active before continuing
err = waitForLBV2Pool(lbClient, pool, "ACTIVE", lbPendingStatuses, timeout)
if err != nil {
return err
}
return resourcePoolV2Read(d, meta)
}
func resourcePoolV2Delete(d *schema.ResourceData, meta interface{}) error {
config := meta.(*Config)
lbClient, err := chooseLBV2Client(d, config)
if err != nil {
return fmt.Errorf("Error creating OpenStack networking client: %s", err)
}
timeout := d.Timeout(schema.TimeoutDelete)
// Get a clean copy of the pool.
pool, err := pools.Get(lbClient, d.Id()).Extract()
if err != nil {
return CheckDeleted(d, err, "Unable to retrieve pool")
}
log.Printf("[DEBUG] Attempting to delete pool %s", d.Id())
err = resource.Retry(timeout, func() *resource.RetryError {
err = pools.Delete(lbClient, d.Id()).ExtractErr()
if err != nil {
return checkForRetryableError(err)
}
return nil
})
if err != nil {
return CheckDeleted(d, err, "Error deleting pool")
}
// Wait for Pool to delete
err = waitForLBV2Pool(lbClient, pool, "DELETED", lbPendingDeleteStatuses, timeout)
if err != nil {
return err
}
return nil
}
func resourcePoolV2Import(d *schema.ResourceData, meta interface{}) ([]*schema.ResourceData, error) {
config := meta.(*Config)
lbClient, err := chooseLBV2Client(d, config)
if err != nil {
return nil, fmt.Errorf("Error creating OpenStack networking client: %s", err)
}
pool, err := pools.Get(lbClient, d.Id()).Extract()
if err != nil {
return nil, CheckDeleted(d, err, "pool")
}
log.Printf("[DEBUG] Retrieved pool %s during the import: %#v", d.Id(), pool)
if len(pool.Listeners) > 0 && pool.Listeners[0].ID != "" {
d.Set("listener_id", pool.Listeners[0].ID)
} else if len(pool.Loadbalancers) > 0 && pool.Loadbalancers[0].ID != "" {
d.Set("loadbalancer_id", pool.Loadbalancers[0].ID)
} else {
return nil, fmt.Errorf("Unable to detect pool's Listener ID or Load Balancer ID")
}
return []*schema.ResourceData{d}, nil
}