IntenseWebs/opentofu
3
0
Fork 0
mirror of https://github.com/opentofu/opentofu.git synced 2025-02-25 18:45:20 -06:00
Code Issues Packages Projects Releases Wiki Activity

Enable import support for PostgreSQL's extensions.

Browse Source
This commit is contained in:
Sean Chittenden 2016-12-12 01:33:02 -08:00
parent e36827c5fe
commit 6deb61b0cf
No known key found for this signature in database
GPG Key ID: 4EBC9DC16C2E5E16
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
builtin/providers/postgresql/resource_postgresql_extension.go
View File

@ -3,6 +3,7 @@ package postgresql
import ( import (
"database/sql" "database/sql"
"fmt" "fmt"
"log"
"github.com/hashicorp/errwrap" "github.com/hashicorp/errwrap"
"github.com/hashicorp/terraform/helper/schema" "github.com/hashicorp/terraform/helper/schema"
@ -14,6 +15,9 @@ func resourcePostgreSQLExtension() *schema.Resource {
Create: resourcePostgreSQLExtensionCreate, Create: resourcePostgreSQLExtensionCreate,
Read: resourcePostgreSQLExtensionRead, Read: resourcePostgreSQLExtensionRead,
Delete: resourcePostgreSQLExtensionDelete, Delete: resourcePostgreSQLExtensionDelete,
Importer: &schema.ResourceImporter{
State: schema.ImportStatePassthrough,
},
Schema: map[string]*schema.Schema{ Schema: map[string]*schema.Schema{
"name": { "name": {
@ -26,8 +30,8 @@ func resourcePostgreSQLExtension() *schema.Resource {
} }
func resourcePostgreSQLExtensionCreate(d *schema.ResourceData, meta interface{}) error { func resourcePostgreSQLExtensionCreate(d *schema.ResourceData, meta interface{}) error {
client := meta.(*Client) c := meta.(*Client)
conn, err := client.Connect() conn, err := c.Connect()
if err != nil { if err != nil {
return err return err
} }
@ -47,32 +51,35 @@ func resourcePostgreSQLExtensionCreate(d *schema.ResourceData, meta interface{})
} }
func resourcePostgreSQLExtensionRead(d *schema.ResourceData, meta interface{}) error { func resourcePostgreSQLExtensionRead(d *schema.ResourceData, meta interface{}) error {
client := meta.(*Client) c := meta.(*Client)
conn, err := client.Connect() conn, err := c.Connect()
if err != nil { if err != nil {
return err return err
} }
defer conn.Close() defer conn.Close()
dbId := d.Id()
extensionName := d.Get("name").(string) extensionName := d.Get("name").(string)
var hasExtension bool var hasExtension bool
err = conn.QueryRow("SELECT 1 from pg_extension d WHERE extname=$1", extensionName).Scan(&hasExtension) err = conn.QueryRow("SELECT TRUE from pg_catalog.pg_extension d WHERE extname=$1", dbId).Scan(&hasExtension)
switch { switch {
case err == sql.ErrNoRows: case err == sql.ErrNoRows:
log.Printf("[WARN] PostgreSQL extension (%s) not found", d.Id())
d.SetId("") d.SetId("")
return nil return nil
case err != nil: case err != nil:
return errwrap.Wrapf("Error reading extension: {{err}}", err) return errwrap.Wrapf("Error reading extension: {{err}}", err)
default: default:
d.Set("extension", hasExtension) d.Set("extension", hasExtension)
d.SetId(extensionName)
return nil return nil
} }
} }
func resourcePostgreSQLExtensionDelete(d *schema.ResourceData, meta interface{}) error { func resourcePostgreSQLExtensionDelete(d *schema.ResourceData, meta interface{}) error {
client := meta.(*Client) c := meta.(*Client)
conn, err := client.Connect() conn, err := c.Connect()
if err != nil { if err != nil {
return err return err
} }