126 lines
3.0 KiB
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
|
|
}
|