terraform/builtin/providers/consul/query_options.go

126 lines
3.0 KiB
Go

package consul
import (
"time"
consulapi "github.com/hashicorp/consul/api"
"github.com/hashicorp/terraform/helper/schema"
)
const (
queryOptAllowStale = "allow_stale"
queryOptDatacenter = "datacenter"
queryOptNear = "near"
queryOptNodeMeta = "node_meta"
queryOptRequireConsistent = "require_consistent"
queryOptToken = "token"
queryOptWaitIndex = "wait_index"
queryOptWaitTime = "wait_time"
)
var schemaQueryOpts = &schema.Schema{
Optional: true,
Type: schema.TypeSet,
Elem: &schema.Resource{
Schema: map[string]*schema.Schema{
queryOptAllowStale: &schema.Schema{
Optional: true,
Default: true,
Type: schema.TypeBool,
},
queryOptDatacenter: &schema.Schema{
// Optional because we'll pull the default from the local agent if it's
// not specified, but we can query remote data centers as a result.
Optional: true,
Type: schema.TypeString,
},
queryOptNear: &schema.Schema{
Optional: true,
Type: schema.TypeString,
},
queryOptNodeMeta: &schema.Schema{
Optional: true,
Type: schema.TypeMap,
},
queryOptRequireConsistent: &schema.Schema{
Optional: true,
Default: false,
Type: schema.TypeBool,
},
queryOptToken: &schema.Schema{
Optional: true,
Default: true,
Type: schema.TypeString,
},
queryOptWaitIndex: &schema.Schema{
Optional: true,
Default: true,
Type: schema.TypeInt,
ValidateFunc: makeValidationFunc(queryOptWaitIndex, []interface{}{
validateIntMin(0),
}),
},
queryOptWaitTime: &schema.Schema{
Optional: true,
Default: true,
Type: schema.TypeString,
ValidateFunc: makeValidationFunc(queryOptWaitTime, []interface{}{
validateDurationMin("0ns"),
}),
},
},
},
}
func getQueryOpts(d *schema.ResourceData, client *consulapi.Client) (*consulapi.QueryOptions, error) {
queryOpts := &consulapi.QueryOptions{}
if v, ok := d.GetOk(queryOptAllowStale); ok {
queryOpts.AllowStale = v.(bool)
}
if v, ok := d.GetOk(queryOptDatacenter); ok {
queryOpts.Datacenter = v.(string)
}
if queryOpts.Datacenter == "" {
dc, err := getDC(d, client)
if err != nil {
return nil, err
}
queryOpts.Datacenter = dc
}
if v, ok := d.GetOk(queryOptNear); ok {
queryOpts.Near = v.(string)
}
if v, ok := d.GetOk(queryOptRequireConsistent); ok {
queryOpts.RequireConsistent = v.(bool)
}
if v, ok := d.GetOk(queryOptNodeMeta); ok {
m := v.(map[string]interface{})
nodeMetaMap := make(map[string]string, len(queryOptNodeMeta))
for s, t := range m {
nodeMetaMap[s] = t.(string)
}
queryOpts.NodeMeta = nodeMetaMap
}
if v, ok := d.GetOk(queryOptToken); ok {
queryOpts.Token = v.(string)
}
if v, ok := d.GetOk(queryOptWaitIndex); ok {
queryOpts.WaitIndex = uint64(v.(int))
}
if v, ok := d.GetOk(queryOptWaitTime); ok {
d, _ := time.ParseDuration(v.(string))
queryOpts.WaitTime = d
}
return queryOpts, nil
}