mirror of https://github.com/status-im/consul.git
96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
// Copyright 2016 Circonus, Inc. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// CheckBundleMetrics API support - Fetch, Create*, Update, and Delete**
|
|
// See: https://login.circonus.com/resources/api/calls/check_bundle_metrics
|
|
// * : create metrics by adding to array with a status of 'active'
|
|
// ** : delete (distable collection of) metrics by changing status from 'active' to 'available'
|
|
|
|
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"regexp"
|
|
|
|
"github.com/circonus-labs/circonus-gometrics/api/config"
|
|
)
|
|
|
|
// CheckBundleMetrics defines metrics for a specific check bundle. See https://login.circonus.com/resources/api/calls/check_bundle_metrics for more information.
|
|
type CheckBundleMetrics struct {
|
|
CID string `json:"_cid,omitempty"` // string
|
|
Metrics []CheckBundleMetric `json:"metrics"` // See check_bundle.go for CheckBundleMetric definition
|
|
}
|
|
|
|
// FetchCheckBundleMetrics retrieves metrics for the check bundle with passed cid.
|
|
func (a *API) FetchCheckBundleMetrics(cid CIDType) (*CheckBundleMetrics, error) {
|
|
if cid == nil || *cid == "" {
|
|
return nil, fmt.Errorf("Invalid check bundle metrics CID [none]")
|
|
}
|
|
|
|
metricsCID := string(*cid)
|
|
|
|
matched, err := regexp.MatchString(config.CheckBundleMetricsCIDRegex, metricsCID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !matched {
|
|
return nil, fmt.Errorf("Invalid check bundle metrics CID [%s]", metricsCID)
|
|
}
|
|
|
|
result, err := a.Get(metricsCID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if a.Debug {
|
|
a.Log.Printf("[DEBUG] fetch check bundle metrics, received JSON: %s", string(result))
|
|
}
|
|
|
|
metrics := &CheckBundleMetrics{}
|
|
if err := json.Unmarshal(result, metrics); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return metrics, nil
|
|
}
|
|
|
|
// UpdateCheckBundleMetrics updates passed metrics.
|
|
func (a *API) UpdateCheckBundleMetrics(cfg *CheckBundleMetrics) (*CheckBundleMetrics, error) {
|
|
if cfg == nil {
|
|
return nil, fmt.Errorf("Invalid check bundle metrics config [nil]")
|
|
}
|
|
|
|
metricsCID := string(cfg.CID)
|
|
|
|
matched, err := regexp.MatchString(config.CheckBundleMetricsCIDRegex, metricsCID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !matched {
|
|
return nil, fmt.Errorf("Invalid check bundle metrics CID [%s]", metricsCID)
|
|
}
|
|
|
|
jsonCfg, err := json.Marshal(cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if a.Debug {
|
|
a.Log.Printf("[DEBUG] update check bundle metrics, sending JSON: %s", string(jsonCfg))
|
|
}
|
|
|
|
result, err := a.Put(metricsCID, jsonCfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
metrics := &CheckBundleMetrics{}
|
|
if err := json.Unmarshal(result, metrics); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return metrics, nil
|
|
}
|