157 lines
3.5 KiB
Go
157 lines
3.5 KiB
Go
package postgresql
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/hashicorp/errwrap"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
"github.com/lib/pq"
|
|
)
|
|
|
|
const (
|
|
extNameAttr = "name"
|
|
extSchemaAttr = "schema"
|
|
)
|
|
|
|
func resourcePostgreSQLExtension() *schema.Resource {
|
|
return &schema.Resource{
|
|
Create: resourcePostgreSQLExtensionCreate,
|
|
Read: resourcePostgreSQLExtensionRead,
|
|
Update: resourcePostgreSQLExtensionUpdate,
|
|
Delete: resourcePostgreSQLExtensionDelete,
|
|
Importer: &schema.ResourceImporter{
|
|
State: schema.ImportStatePassthrough,
|
|
},
|
|
|
|
Schema: map[string]*schema.Schema{
|
|
extNameAttr: {
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ForceNew: true,
|
|
},
|
|
extSchemaAttr: {
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
Computed: true,
|
|
Description: "Sets the schema of an extension",
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionCreate(d *schema.ResourceData, meta interface{}) error {
|
|
c := meta.(*Client)
|
|
conn, err := c.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
extName := d.Get(extNameAttr).(string)
|
|
|
|
b := bytes.NewBufferString("CREATE EXTENSION ")
|
|
fmt.Fprintf(b, pq.QuoteIdentifier(extName))
|
|
|
|
if v, ok := d.GetOk(extSchemaAttr); ok {
|
|
fmt.Fprint(b, " SCHEMA ", pq.QuoteIdentifier(v.(string)))
|
|
}
|
|
|
|
query := b.String()
|
|
_, err = conn.Query(query)
|
|
if err != nil {
|
|
return errwrap.Wrapf("Error creating extension: {{err}}", err)
|
|
}
|
|
|
|
d.SetId(extName)
|
|
|
|
return resourcePostgreSQLExtensionRead(d, meta)
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionRead(d *schema.ResourceData, meta interface{}) error {
|
|
c := meta.(*Client)
|
|
conn, err := c.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
extID := d.Id()
|
|
var extName, extSchema string
|
|
err = conn.QueryRow("SELECT e.extname, n.nspname FROM pg_catalog.pg_extension e, pg_catalog.pg_namespace n WHERE n.oid = e.extnamespace AND e.extname = $1", extID).Scan(&extName, &extSchema)
|
|
switch {
|
|
case err == sql.ErrNoRows:
|
|
log.Printf("[WARN] PostgreSQL extension (%s) not found", d.Id())
|
|
d.SetId("")
|
|
return nil
|
|
case err != nil:
|
|
return errwrap.Wrapf("Error reading extension: {{err}}", err)
|
|
default:
|
|
d.Set(extNameAttr, extName)
|
|
d.Set(extSchemaAttr, extSchema)
|
|
d.SetId(extName)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionDelete(d *schema.ResourceData, meta interface{}) error {
|
|
c := meta.(*Client)
|
|
conn, err := c.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
extID := d.Id()
|
|
|
|
query := fmt.Sprintf("DROP EXTENSION %s", pq.QuoteIdentifier(extID))
|
|
_, err = conn.Query(query)
|
|
if err != nil {
|
|
return errwrap.Wrapf("Error deleting extension: {{err}}", err)
|
|
}
|
|
|
|
d.SetId("")
|
|
|
|
return nil
|
|
}
|
|
|
|
func resourcePostgreSQLExtensionUpdate(d *schema.ResourceData, meta interface{}) error {
|
|
c := meta.(*Client)
|
|
conn, err := c.Connect()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer conn.Close()
|
|
|
|
// Can't rename a schema
|
|
|
|
if err := setExtSchema(conn, d); err != nil {
|
|
return err
|
|
}
|
|
|
|
return resourcePostgreSQLExtensionRead(d, meta)
|
|
}
|
|
|
|
func setExtSchema(conn *sql.DB, d *schema.ResourceData) error {
|
|
if !d.HasChange(extSchemaAttr) {
|
|
return nil
|
|
}
|
|
|
|
extID := d.Id()
|
|
_, nraw := d.GetChange(extSchemaAttr)
|
|
n := nraw.(string)
|
|
if n == "" {
|
|
return errors.New("Error setting extension name to an empty string")
|
|
}
|
|
|
|
query := fmt.Sprintf("ALTER EXTENSION %s SET SCHEMA %s", pq.QuoteIdentifier(extID), pq.QuoteIdentifier(n))
|
|
if _, err := conn.Query(query); err != nil {
|
|
return errwrap.Wrapf("Error updating extension SCHEMA: {{err}}", err)
|
|
}
|
|
|
|
return nil
|
|
}
|