103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
package circonus
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/circonus-labs/circonus-gometrics/api/config"
|
|
"github.com/hashicorp/errwrap"
|
|
"github.com/hashicorp/terraform/helper/hashcode"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
)
|
|
|
|
const (
|
|
// circonus_check.mysql.* resource attribute names
|
|
checkMySQLDSNAttr = "dsn"
|
|
checkMySQLQueryAttr = "query"
|
|
)
|
|
|
|
var checkMySQLDescriptions = attrDescrs{
|
|
checkMySQLDSNAttr: "The connect DSN for the MySQL instance",
|
|
checkMySQLQueryAttr: "The SQL to use as the query",
|
|
}
|
|
|
|
var schemaCheckMySQL = &schema.Schema{
|
|
Type: schema.TypeSet,
|
|
Optional: true,
|
|
MaxItems: 1,
|
|
MinItems: 1,
|
|
Set: hashCheckMySQL,
|
|
Elem: &schema.Resource{
|
|
Schema: convertToHelperSchema(checkMySQLDescriptions, map[schemaAttr]*schema.Schema{
|
|
checkMySQLDSNAttr: &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
ValidateFunc: validateRegexp(checkMySQLDSNAttr, `^.+$`),
|
|
},
|
|
checkMySQLQueryAttr: &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Required: true,
|
|
StateFunc: func(v interface{}) string { return strings.TrimSpace(v.(string)) },
|
|
ValidateFunc: validateRegexp(checkMySQLQueryAttr, `.+`),
|
|
},
|
|
}),
|
|
},
|
|
}
|
|
|
|
// checkAPIToStateMySQL reads the Config data out of circonusCheck.CheckBundle into the
|
|
// statefile.
|
|
func checkAPIToStateMySQL(c *circonusCheck, d *schema.ResourceData) error {
|
|
MySQLConfig := make(map[string]interface{}, len(c.Config))
|
|
|
|
MySQLConfig[string(checkMySQLDSNAttr)] = c.Config[config.DSN]
|
|
MySQLConfig[string(checkMySQLQueryAttr)] = c.Config[config.SQL]
|
|
|
|
if err := d.Set(checkMySQLAttr, schema.NewSet(hashCheckMySQL, []interface{}{MySQLConfig})); err != nil {
|
|
return errwrap.Wrapf(fmt.Sprintf("Unable to store check %q attribute: {{err}}", checkMySQLAttr), err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// hashCheckMySQL creates a stable hash of the normalized values
|
|
func hashCheckMySQL(v interface{}) int {
|
|
m := v.(map[string]interface{})
|
|
b := &bytes.Buffer{}
|
|
b.Grow(defaultHashBufSize)
|
|
|
|
writeString := func(attrName schemaAttr) {
|
|
if v, ok := m[string(attrName)]; ok && v.(string) != "" {
|
|
fmt.Fprint(b, strings.TrimSpace(v.(string)))
|
|
}
|
|
}
|
|
|
|
// Order writes to the buffer using lexically sorted list for easy visual
|
|
// reconciliation with other lists.
|
|
writeString(checkMySQLDSNAttr)
|
|
writeString(checkMySQLQueryAttr)
|
|
|
|
s := b.String()
|
|
return hashcode.String(s)
|
|
}
|
|
|
|
func checkConfigToAPIMySQL(c *circonusCheck, l interfaceList) error {
|
|
c.Type = string(apiCheckTypeMySQL)
|
|
|
|
// Iterate over all `postgres` attributes, even though we have a max of 1 in
|
|
// the schema.
|
|
for _, mapRaw := range l {
|
|
mysqlConfig := newInterfaceMap(mapRaw)
|
|
|
|
if v, found := mysqlConfig[checkMySQLDSNAttr]; found {
|
|
c.Config[config.DSN] = v.(string)
|
|
}
|
|
|
|
if v, found := mysqlConfig[checkMySQLQueryAttr]; found {
|
|
c.Config[config.SQL] = v.(string)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|