resource_flexibleengine_elb_health.go

package flexibleengine

import (
	"fmt"
	"log"
	"time"

	"github.com/hashicorp/terraform-plugin-sdk/helper/schema"

	"github.com/huaweicloud/golangsdk/openstack/networking/v2/extensions/elbaas/healthcheck"
)

func resourceHealth() *schema.Resource {
	return &schema.Resource{
		Create: resourceHealthCreate,
		Read:   resourceHealthRead,
		Update: resourceHealthUpdate,
		Delete: resourceHealthDelete,

		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,
			},

			"listener_id": {
				Type:     schema.TypeString,
				Required: true,
				ForceNew: true,
			},
			"healthcheck_protocol": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"healthcheck_uri": {
				Type:     schema.TypeString,
				Optional: true,
			},
			"healthcheck_connect_port": {
				Type:     schema.TypeInt,
				Optional: true,
			},
			"healthy_threshold": {
				Type:     schema.TypeInt,
				Optional: true,
			},
			"unhealthy_threshold": {
				Type:     schema.TypeInt,
				Optional: true,
				Computed: true,
			},
			"healthcheck_timeout": {
				Type:     schema.TypeInt,
				Optional: true,
			},
			"healthcheck_interval": {
				Type:     schema.TypeInt,
				Optional: true,
			},
		},
	}
}

func resourceHealthCreate(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)
	client, err := config.otcV1Client(GetRegion(d, config))
	if err != nil {
		return fmt.Errorf("Error creating FlexibleEngine networking client: %s", err)
	}

	createOpts := healthcheck.CreateOpts{
		ListenerID:             d.Get("listener_id").(string),
		HealthcheckProtocol:    d.Get("healthcheck_protocol").(string),
		HealthcheckUri:         d.Get("healthcheck_uri").(string),
		HealthcheckConnectPort: d.Get("healthcheck_connect_port").(int),
		HealthyThreshold:       d.Get("healthy_threshold").(int),
		UnhealthyThreshold:     d.Get("unhealthy_threshold").(int),
		HealthcheckTimeout:     d.Get("healthcheck_timeout").(int),
		HealthcheckInterval:    d.Get("healthcheck_interval").(int),
	}

	health, err := healthcheck.Create(client, createOpts).Extract()
	if err != nil {
		return err
	}
	d.SetId(health.ID)

	log.Printf("Successfully created healthcheck %s.", health.ID)

	return resourceHealthRead(d, meta)
}

func resourceHealthRead(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)
	networkingClient, err := config.otcV1Client(GetRegion(d, config))
	if err != nil {
		return fmt.Errorf("Error creating FlexibleEngine networking client: %s", err)
	}

	health, err := healthcheck.Get(networkingClient, d.Id()).Extract()
	if err != nil {
		return CheckDeleted(d, err, "health")
	}

	log.Printf("[DEBUG] Retrieved health %s: %#v", d.Id(), health)

	d.SetId(health.ID)
	d.Set("listener_id", health.ListenerID)
	d.Set("healthcheck_protocol", health.HealthcheckProtocol)
	d.Set("healthcheck_uri", health.HealthcheckUri)
	d.Set("healtcheck_connect_port", health.HealthcheckConnectPort)
	d.Set("healthy_threshold", health.HealthyThreshold)
	d.Set("unhealthy_threshold", health.UnhealthyThreshold)
	d.Set("healthcheck_timeout", health.HealthcheckTimeout)
	d.Set("healthcheck_interval", health.HealthcheckInterval)

	d.Set("region", GetRegion(d, config))

	return nil
}

func resourceHealthUpdate(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)
	networkingClient, err := config.otcV1Client(GetRegion(d, config))
	if err != nil {
		return fmt.Errorf("Error creating FlexibleEngine networking client: %s", err)
	}

	var updateOpts healthcheck.UpdateOpts
	if d.HasChange("healthcheck_protocol") {
		updateOpts.HealthcheckProtocol = d.Get("healthcheck_protocol").(string)
	}
	if d.HasChange("healthcheck_uri") {
		updateOpts.HealthcheckUri = d.Get("healthcheck_uri").(string)
	}
	if d.HasChange("healthcheck_connect_port") {
		updateOpts.HealthyThreshold = d.Get("healthcheck_connect_port").(int)
	}
	if d.HasChange("healthy_threshold") {
		updateOpts.HealthyThreshold = d.Get("healthy_threshold").(int)
	}
	if d.HasChange("unhealthy_threshold") {
		updateOpts.UnhealthyThreshold = d.Get("unhealthy_threshold").(int)
	}
	if d.HasChange("healthcheck_timeout") {
		updateOpts.HealthcheckTimeout = d.Get("healthcheck_timeout").(int)
	}
	if d.HasChange("healthcheck_interval") {
		updateOpts.HealthcheckInterval = d.Get("healthcheck_interval").(int)
	}

	log.Printf("[DEBUG] Updating health %s with options: %#v", d.Id(), updateOpts)

	_, err = healthcheck.Update(networkingClient, d.Id(), updateOpts).Extract()
	if err != nil {
		return err
	}

	return resourceHealthRead(d, meta)
}

func resourceHealthDelete(d *schema.ResourceData, meta interface{}) error {
	config := meta.(*Config)
	client, err := config.otcV1Client(GetRegion(d, config))
	if err != nil {
		return fmt.Errorf("Error creating FlexibleEngine networking client: %s", err)
	}

	id := d.Id()
	log.Printf("[DEBUG] Deleting health %s", id)

	if err := healthcheck.Delete(client, id).ExtractErr(); err != nil {
		return err
	}

	log.Printf("Successfully deleted health %s", id)
	return nil
}