mirror of
https://github.com/letic/terraform-provider-google.git
synced 2024-07-03 08:42:39 +00:00
135 lines
2.8 KiB
Go
135 lines
2.8 KiB
Go
![]() |
package google
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/hashicorp/terraform/helper/schema"
|
||
|
|
||
|
"golang.org/x/net/context"
|
||
|
)
|
||
|
|
||
|
func resourceBigtableTable() *schema.Resource {
|
||
|
return &schema.Resource{
|
||
|
Create: resourceBigtableTableCreate,
|
||
|
Read: resourceBigtableTableRead,
|
||
|
Delete: resourceBigtableTableDestroy,
|
||
|
|
||
|
Schema: map[string]*schema.Schema{
|
||
|
"name": {
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
|
||
|
"instance_name": {
|
||
|
Type: schema.TypeString,
|
||
|
Required: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
|
||
|
"split_keys": {
|
||
|
Type: schema.TypeList,
|
||
|
Optional: true,
|
||
|
ForceNew: true,
|
||
|
Elem: &schema.Schema{Type: schema.TypeString},
|
||
|
},
|
||
|
|
||
|
"project": {
|
||
|
Type: schema.TypeString,
|
||
|
Optional: true,
|
||
|
ForceNew: true,
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func resourceBigtableTableCreate(d *schema.ResourceData, meta interface{}) error {
|
||
|
config := meta.(*Config)
|
||
|
ctx := context.Background()
|
||
|
|
||
|
project, err := getProject(d, config)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
instanceName := d.Get("instance_name").(string)
|
||
|
c, err := config.bigtableClientFactory.NewAdminClient(project, instanceName)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error starting admin client. %s", err)
|
||
|
}
|
||
|
|
||
|
defer c.Close()
|
||
|
|
||
|
name := d.Get("name").(string)
|
||
|
if v, ok := d.GetOk("split_keys"); ok {
|
||
|
splitKeys := convertSchemaArrayToStringArray(v.([]interface{}))
|
||
|
err = c.CreatePresplitTable(ctx, name, splitKeys)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error creating presplit table. %s", err)
|
||
|
}
|
||
|
} else {
|
||
|
err = c.CreateTable(ctx, name)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error creating table. %s", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
d.SetId(name)
|
||
|
|
||
|
return resourceBigtableTableRead(d, meta)
|
||
|
}
|
||
|
|
||
|
func resourceBigtableTableRead(d *schema.ResourceData, meta interface{}) error {
|
||
|
config := meta.(*Config)
|
||
|
ctx := context.Background()
|
||
|
|
||
|
project, err := getProject(d, config)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
instanceName := d.Get("instance_name").(string)
|
||
|
c, err := config.bigtableClientFactory.NewAdminClient(project, instanceName)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error starting admin client. %s", err)
|
||
|
}
|
||
|
|
||
|
defer c.Close()
|
||
|
|
||
|
name := d.Id()
|
||
|
_, err = c.TableInfo(ctx, name)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error retrieving table. Could not find %s in %s.", name, instanceName)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func resourceBigtableTableDestroy(d *schema.ResourceData, meta interface{}) error {
|
||
|
config := meta.(*Config)
|
||
|
ctx := context.Background()
|
||
|
|
||
|
project, err := getProject(d, config)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
instanceName := d.Get("instance_name").(string)
|
||
|
c, err := config.bigtableClientFactory.NewAdminClient(project, instanceName)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error starting admin client. %s", err)
|
||
|
}
|
||
|
|
||
|
defer c.Close()
|
||
|
|
||
|
name := d.Get("name").(string)
|
||
|
err = c.DeleteTable(ctx, name)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Error deleting instance. %s", err)
|
||
|
}
|
||
|
|
||
|
d.SetId("")
|
||
|
|
||
|
return nil
|
||
|
}
|