93 lines
2.0 KiB
Go
93 lines
2.0 KiB
Go
package postgresql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/hashicorp/errwrap"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
"github.com/lib/pq"
|
|
)
|
|
|
|
func resourcePostgreSQLExtension() *schema.Resource {
|
|
return &schema.Resource{
|
|
Create: resourcePostgreSQLExtensionCreate,
|
|
Read: resourcePostgreSQLExtensionRead,
|
|
Delete: resourcePostgreSQLExtensionDelete,
|
|
|
|
Schema: map[string]*schema.Schema{
|
|
"name": {
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ForceNew: true,
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionCreate(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Client)
|
|
conn, err := client.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
extensionName := d.Get("name").(string)
|
|
|
|
query := fmt.Sprintf("CREATE EXTENSION %s", pq.QuoteIdentifier(extensionName))
|
|
_, err = conn.Query(query)
|
|
if err != nil {
|
|
return errwrap.Wrapf("Error creating extension: {{err}}", err)
|
|
}
|
|
|
|
d.SetId(extensionName)
|
|
|
|
return resourcePostgreSQLExtensionRead(d, meta)
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionRead(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Client)
|
|
conn, err := client.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
extensionName := d.Get("name").(string)
|
|
|
|
var hasExtension bool
|
|
err = conn.QueryRow("SELECT 1 from pg_extension d WHERE extname=$1", extensionName).Scan(&hasExtension)
|
|
switch {
|
|
case err == sql.ErrNoRows:
|
|
d.SetId("")
|
|
return nil
|
|
case err != nil:
|
|
return errwrap.Wrapf("Error reading extension: {{err}}", err)
|
|
default:
|
|
d.Set("extension", hasExtension)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionDelete(d *schema.ResourceData, meta interface{}) error {
|
|
client := meta.(*Client)
|
|
conn, err := client.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
extensionName := d.Get("name").(string)
|
|
|
|
query := fmt.Sprintf("DROP EXTENSION %s", pq.QuoteIdentifier(extensionName))
|
|
_, err = conn.Query(query)
|
|
if err != nil {
|
|
return errwrap.Wrapf("Error deleting extension: {{err}}", err)
|
|
}
|
|
|
|
d.SetId("")
|
|
|
|
return nil
|
|
}
|