mirror of
https://github.com/grafana/grafana.git
synced 2025-02-16 18:34:52 -06:00
275 lines
8.0 KiB
Go
275 lines
8.0 KiB
Go
package cloudwatch
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"io/ioutil"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/aws/awsutil"
|
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
|
"github.com/aws/aws-sdk-go/aws/credentials/ec2rolecreds"
|
|
"github.com/aws/aws-sdk-go/aws/ec2metadata"
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
|
"github.com/aws/aws-sdk-go/service/cloudwatch"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/grafana/grafana/pkg/middleware"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
)
|
|
|
|
type actionHandler func(*cwRequest, *middleware.Context)
|
|
|
|
var actionHandlers map[string]actionHandler
|
|
|
|
type cwRequest struct {
|
|
Region string `json:"region"`
|
|
Action string `json:"action"`
|
|
Body []byte `json:"-"`
|
|
DataSource *m.DataSource
|
|
}
|
|
|
|
func init() {
|
|
actionHandlers = map[string]actionHandler{
|
|
"GetMetricStatistics": handleGetMetricStatistics,
|
|
"ListMetrics": handleListMetrics,
|
|
"DescribeAlarmsForMetric": handleDescribeAlarmsForMetric,
|
|
"DescribeAlarmHistory": handleDescribeAlarmHistory,
|
|
"DescribeInstances": handleDescribeInstances,
|
|
"__GetRegions": handleGetRegions,
|
|
"__GetNamespaces": handleGetNamespaces,
|
|
"__GetMetrics": handleGetMetrics,
|
|
"__GetDimensions": handleGetDimensions,
|
|
}
|
|
}
|
|
|
|
var awsCredentials map[string]*credentials.Credentials = make(map[string]*credentials.Credentials)
|
|
|
|
func getCredentials(profile string) *credentials.Credentials {
|
|
if _, ok := awsCredentials[profile]; ok {
|
|
return awsCredentials[profile]
|
|
}
|
|
|
|
sess := session.New()
|
|
creds := credentials.NewChainCredentials(
|
|
[]credentials.Provider{
|
|
&credentials.EnvProvider{},
|
|
&credentials.SharedCredentialsProvider{Filename: "", Profile: profile},
|
|
&ec2rolecreds.EC2RoleProvider{Client: ec2metadata.New(sess), ExpiryWindow: 5 * time.Minute},
|
|
})
|
|
awsCredentials[profile] = creds
|
|
|
|
return creds
|
|
}
|
|
|
|
func handleGetMetricStatistics(req *cwRequest, c *middleware.Context) {
|
|
cfg := &aws.Config{
|
|
Region: aws.String(req.Region),
|
|
Credentials: getCredentials(req.DataSource.Database),
|
|
}
|
|
|
|
svc := cloudwatch.New(session.New(cfg), cfg)
|
|
|
|
reqParam := &struct {
|
|
Parameters struct {
|
|
Namespace string `json:"namespace"`
|
|
MetricName string `json:"metricName"`
|
|
Dimensions []*cloudwatch.Dimension `json:"dimensions"`
|
|
Statistics []*string `json:"statistics"`
|
|
StartTime int64 `json:"startTime"`
|
|
EndTime int64 `json:"endTime"`
|
|
Period int64 `json:"period"`
|
|
} `json:"parameters"`
|
|
}{}
|
|
json.Unmarshal(req.Body, reqParam)
|
|
|
|
params := &cloudwatch.GetMetricStatisticsInput{
|
|
Namespace: aws.String(reqParam.Parameters.Namespace),
|
|
MetricName: aws.String(reqParam.Parameters.MetricName),
|
|
Dimensions: reqParam.Parameters.Dimensions,
|
|
Statistics: reqParam.Parameters.Statistics,
|
|
StartTime: aws.Time(time.Unix(reqParam.Parameters.StartTime, 0)),
|
|
EndTime: aws.Time(time.Unix(reqParam.Parameters.EndTime, 0)),
|
|
Period: aws.Int64(reqParam.Parameters.Period),
|
|
}
|
|
|
|
resp, err := svc.GetMetricStatistics(params)
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Unable to call AWS API", err)
|
|
return
|
|
}
|
|
|
|
c.JSON(200, resp)
|
|
}
|
|
|
|
func handleListMetrics(req *cwRequest, c *middleware.Context) {
|
|
cfg := &aws.Config{
|
|
Region: aws.String(req.Region),
|
|
Credentials: getCredentials(req.DataSource.Database),
|
|
}
|
|
|
|
svc := cloudwatch.New(session.New(cfg), cfg)
|
|
|
|
reqParam := &struct {
|
|
Parameters struct {
|
|
Namespace string `json:"namespace"`
|
|
MetricName string `json:"metricName"`
|
|
Dimensions []*cloudwatch.DimensionFilter `json:"dimensions"`
|
|
} `json:"parameters"`
|
|
}{}
|
|
json.Unmarshal(req.Body, reqParam)
|
|
|
|
params := &cloudwatch.ListMetricsInput{
|
|
Namespace: aws.String(reqParam.Parameters.Namespace),
|
|
MetricName: aws.String(reqParam.Parameters.MetricName),
|
|
Dimensions: reqParam.Parameters.Dimensions,
|
|
}
|
|
|
|
var resp cloudwatch.ListMetricsOutput
|
|
err := svc.ListMetricsPages(params,
|
|
func(page *cloudwatch.ListMetricsOutput, lastPage bool) bool {
|
|
metrics, _ := awsutil.ValuesAtPath(page, "Metrics")
|
|
for _, metric := range metrics {
|
|
resp.Metrics = append(resp.Metrics, metric.(*cloudwatch.Metric))
|
|
}
|
|
return !lastPage
|
|
})
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Unable to call AWS API", err)
|
|
return
|
|
}
|
|
|
|
c.JSON(200, resp)
|
|
}
|
|
|
|
func handleDescribeAlarmsForMetric(req *cwRequest, c *middleware.Context) {
|
|
cfg := &aws.Config{
|
|
Region: aws.String(req.Region),
|
|
Credentials: getCredentials(req.DataSource.Database),
|
|
}
|
|
|
|
svc := cloudwatch.New(session.New(cfg), cfg)
|
|
|
|
reqParam := &struct {
|
|
Parameters struct {
|
|
Namespace string `json:"namespace"`
|
|
MetricName string `json:"metricName"`
|
|
Dimensions []*cloudwatch.Dimension `json:"dimensions"`
|
|
Statistic string `json:"statistic"`
|
|
Period int64 `json:"period"`
|
|
} `json:"parameters"`
|
|
}{}
|
|
json.Unmarshal(req.Body, reqParam)
|
|
|
|
params := &cloudwatch.DescribeAlarmsForMetricInput{
|
|
Namespace: aws.String(reqParam.Parameters.Namespace),
|
|
MetricName: aws.String(reqParam.Parameters.MetricName),
|
|
Period: aws.Int64(reqParam.Parameters.Period),
|
|
}
|
|
if len(reqParam.Parameters.Dimensions) != 0 {
|
|
params.Dimensions = reqParam.Parameters.Dimensions
|
|
}
|
|
if reqParam.Parameters.Statistic != "" {
|
|
params.Statistic = aws.String(reqParam.Parameters.Statistic)
|
|
}
|
|
|
|
resp, err := svc.DescribeAlarmsForMetric(params)
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Unable to call AWS API", err)
|
|
return
|
|
}
|
|
|
|
c.JSON(200, resp)
|
|
}
|
|
|
|
func handleDescribeAlarmHistory(req *cwRequest, c *middleware.Context) {
|
|
cfg := &aws.Config{
|
|
Region: aws.String(req.Region),
|
|
Credentials: getCredentials(req.DataSource.Database),
|
|
}
|
|
|
|
svc := cloudwatch.New(session.New(cfg), cfg)
|
|
|
|
reqParam := &struct {
|
|
Parameters struct {
|
|
AlarmName string `json:"alarmName"`
|
|
HistoryItemType string `json:"historyItemType"`
|
|
StartDate int64 `json:"startDate"`
|
|
EndDate int64 `json:"endDate"`
|
|
} `json:"parameters"`
|
|
}{}
|
|
json.Unmarshal(req.Body, reqParam)
|
|
|
|
params := &cloudwatch.DescribeAlarmHistoryInput{
|
|
AlarmName: aws.String(reqParam.Parameters.AlarmName),
|
|
StartDate: aws.Time(time.Unix(reqParam.Parameters.StartDate, 0)),
|
|
EndDate: aws.Time(time.Unix(reqParam.Parameters.EndDate, 0)),
|
|
}
|
|
if reqParam.Parameters.HistoryItemType != "" {
|
|
params.HistoryItemType = aws.String(reqParam.Parameters.HistoryItemType)
|
|
}
|
|
|
|
resp, err := svc.DescribeAlarmHistory(params)
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Unable to call AWS API", err)
|
|
return
|
|
}
|
|
|
|
c.JSON(200, resp)
|
|
}
|
|
|
|
func handleDescribeInstances(req *cwRequest, c *middleware.Context) {
|
|
cfg := &aws.Config{
|
|
Region: aws.String(req.Region),
|
|
Credentials: getCredentials(req.DataSource.Database),
|
|
}
|
|
|
|
svc := ec2.New(session.New(cfg), cfg)
|
|
|
|
reqParam := &struct {
|
|
Parameters struct {
|
|
Filters []*ec2.Filter `json:"filters"`
|
|
InstanceIds []*string `json:"instanceIds"`
|
|
} `json:"parameters"`
|
|
}{}
|
|
json.Unmarshal(req.Body, reqParam)
|
|
|
|
params := &ec2.DescribeInstancesInput{}
|
|
if len(reqParam.Parameters.Filters) > 0 {
|
|
params.Filters = reqParam.Parameters.Filters
|
|
}
|
|
if len(reqParam.Parameters.InstanceIds) > 0 {
|
|
params.InstanceIds = reqParam.Parameters.InstanceIds
|
|
}
|
|
|
|
var resp ec2.DescribeInstancesOutput
|
|
err := svc.DescribeInstancesPages(params,
|
|
func(page *ec2.DescribeInstancesOutput, lastPage bool) bool {
|
|
reservations, _ := awsutil.ValuesAtPath(page, "Reservations")
|
|
for _, reservation := range reservations {
|
|
resp.Reservations = append(resp.Reservations, reservation.(*ec2.Reservation))
|
|
}
|
|
return !lastPage
|
|
})
|
|
if err != nil {
|
|
c.JsonApiErr(500, "Unable to call AWS API", err)
|
|
return
|
|
}
|
|
|
|
c.JSON(200, resp)
|
|
}
|
|
|
|
func HandleRequest(c *middleware.Context, ds *m.DataSource) {
|
|
var req cwRequest
|
|
req.Body, _ = ioutil.ReadAll(c.Req.Request.Body)
|
|
req.DataSource = ds
|
|
json.Unmarshal(req.Body, &req)
|
|
|
|
if handler, found := actionHandlers[req.Action]; !found {
|
|
c.JsonApiErr(500, "Unexpected AWS Action", errors.New(req.Action))
|
|
return
|
|
} else {
|
|
handler(&req, c)
|
|
}
|
|
}
|