|
|
@@ -16,6 +16,7 @@ import (
|
|
|
"time"
|
|
|
|
|
|
"golang.org/x/net/context/ctxhttp"
|
|
|
+ "golang.org/x/oauth2/google"
|
|
|
|
|
|
"github.com/grafana/grafana/pkg/api/pluginproxy"
|
|
|
"github.com/grafana/grafana/pkg/components/null"
|
|
|
@@ -34,6 +35,11 @@ var (
|
|
|
metricNameFormat *regexp.Regexp
|
|
|
)
|
|
|
|
|
|
+const (
|
|
|
+ gceAuthentication string = "gce"
|
|
|
+ jwtAuthentication string = "jwt"
|
|
|
+)
|
|
|
+
|
|
|
// StackdriverExecutor executes queries for the Stackdriver datasource
|
|
|
type StackdriverExecutor struct {
|
|
|
httpClient *http.Client
|
|
|
@@ -71,6 +77,8 @@ func (e *StackdriverExecutor) Query(ctx context.Context, dsInfo *models.DataSour
|
|
|
switch queryType {
|
|
|
case "annotationQuery":
|
|
|
result, err = e.executeAnnotationQuery(ctx, tsdbQuery)
|
|
|
+ case "ensureDefaultProjectQuery":
|
|
|
+ result, err = e.ensureDefaultProject(ctx, tsdbQuery)
|
|
|
case "timeSeriesQuery":
|
|
|
fallthrough
|
|
|
default:
|
|
|
@@ -85,6 +93,16 @@ func (e *StackdriverExecutor) executeTimeSeriesQuery(ctx context.Context, tsdbQu
|
|
|
Results: make(map[string]*tsdb.QueryResult),
|
|
|
}
|
|
|
|
|
|
+ authenticationType := e.dsInfo.JsonData.Get("authenticationType").MustString(jwtAuthentication)
|
|
|
+ if authenticationType == gceAuthentication {
|
|
|
+ defaultProject, err := e.getDefaultProject(ctx)
|
|
|
+ if err != nil {
|
|
|
+ return nil, fmt.Errorf("Failed to retrieve default project from GCE metadata server. error: %v", err)
|
|
|
+ }
|
|
|
+
|
|
|
+ e.dsInfo.JsonData.Set("defaultProject", defaultProject)
|
|
|
+ }
|
|
|
+
|
|
|
queries, err := e.buildQueries(tsdbQuery)
|
|
|
if err != nil {
|
|
|
return nil, err
|
|
|
@@ -168,8 +186,7 @@ func reverse(s string) string {
|
|
|
}
|
|
|
|
|
|
func interpolateFilterWildcards(value string) string {
|
|
|
- re := regexp.MustCompile("[*]")
|
|
|
- matches := len(re.FindAllStringIndex(value, -1))
|
|
|
+ matches := strings.Count(value, "*")
|
|
|
if matches == 2 && strings.HasSuffix(value, "*") && strings.HasPrefix(value, "*") {
|
|
|
value = strings.Replace(value, "*", "", -1)
|
|
|
value = fmt.Sprintf(`has_substring("%s")`, value)
|
|
|
@@ -337,11 +354,21 @@ func (e *StackdriverExecutor) unmarshalResponse(res *http.Response) (Stackdriver
|
|
|
func (e *StackdriverExecutor) parseResponse(queryRes *tsdb.QueryResult, data StackdriverResponse, query *StackdriverQuery) error {
|
|
|
metricLabels := make(map[string][]string)
|
|
|
resourceLabels := make(map[string][]string)
|
|
|
+ var resourceTypes []string
|
|
|
+
|
|
|
+ for _, series := range data.TimeSeries {
|
|
|
+ if !containsLabel(resourceTypes, series.Resource.Type) {
|
|
|
+ resourceTypes = append(resourceTypes, series.Resource.Type)
|
|
|
+ }
|
|
|
+ }
|
|
|
|
|
|
for _, series := range data.TimeSeries {
|
|
|
points := make([]tsdb.TimePoint, 0)
|
|
|
|
|
|
defaultMetricName := series.Metric.Type
|
|
|
+ if len(resourceTypes) > 1 {
|
|
|
+ defaultMetricName += " " + series.Resource.Type
|
|
|
+ }
|
|
|
|
|
|
for key, value := range series.Metric.Labels {
|
|
|
if !containsLabel(metricLabels[key], value) {
|
|
|
@@ -385,7 +412,7 @@ func (e *StackdriverExecutor) parseResponse(queryRes *tsdb.QueryResult, data Sta
|
|
|
points = append(points, tsdb.NewTimePoint(null.FloatFrom(value), float64((point.Interval.EndTime).Unix())*1000))
|
|
|
}
|
|
|
|
|
|
- metricName := formatLegendKeys(series.Metric.Type, defaultMetricName, series.Metric.Labels, series.Resource.Labels, make(map[string]string), query)
|
|
|
+ metricName := formatLegendKeys(series.Metric.Type, defaultMetricName, series.Resource.Type, series.Metric.Labels, series.Resource.Labels, make(map[string]string), query)
|
|
|
|
|
|
queryRes.Series = append(queryRes.Series, &tsdb.TimeSeries{
|
|
|
Name: metricName,
|
|
|
@@ -411,7 +438,7 @@ func (e *StackdriverExecutor) parseResponse(queryRes *tsdb.QueryResult, data Sta
|
|
|
bucketBound := calcBucketBound(point.Value.DistributionValue.BucketOptions, i)
|
|
|
additionalLabels := map[string]string{"bucket": bucketBound}
|
|
|
buckets[i] = &tsdb.TimeSeries{
|
|
|
- Name: formatLegendKeys(series.Metric.Type, defaultMetricName, series.Metric.Labels, series.Resource.Labels, additionalLabels, query),
|
|
|
+ Name: formatLegendKeys(series.Metric.Type, defaultMetricName, series.Resource.Type, series.Metric.Labels, series.Resource.Labels, additionalLabels, query),
|
|
|
Points: make([]tsdb.TimePoint, 0),
|
|
|
}
|
|
|
if maxKey < i {
|
|
|
@@ -427,7 +454,7 @@ func (e *StackdriverExecutor) parseResponse(queryRes *tsdb.QueryResult, data Sta
|
|
|
bucketBound := calcBucketBound(point.Value.DistributionValue.BucketOptions, i)
|
|
|
additionalLabels := map[string]string{"bucket": bucketBound}
|
|
|
buckets[i] = &tsdb.TimeSeries{
|
|
|
- Name: formatLegendKeys(series.Metric.Type, defaultMetricName, series.Metric.Labels, series.Resource.Labels, additionalLabels, query),
|
|
|
+ Name: formatLegendKeys(series.Metric.Type, defaultMetricName, series.Resource.Type, series.Metric.Labels, series.Resource.Labels, additionalLabels, query),
|
|
|
Points: make([]tsdb.TimePoint, 0),
|
|
|
}
|
|
|
}
|
|
|
@@ -442,6 +469,7 @@ func (e *StackdriverExecutor) parseResponse(queryRes *tsdb.QueryResult, data Sta
|
|
|
queryRes.Meta.Set("resourceLabels", resourceLabels)
|
|
|
queryRes.Meta.Set("metricLabels", metricLabels)
|
|
|
queryRes.Meta.Set("groupBys", query.GroupBys)
|
|
|
+ queryRes.Meta.Set("resourceTypes", resourceTypes)
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
@@ -455,7 +483,7 @@ func containsLabel(labels []string, newLabel string) bool {
|
|
|
return false
|
|
|
}
|
|
|
|
|
|
-func formatLegendKeys(metricType string, defaultMetricName string, metricLabels map[string]string, resourceLabels map[string]string, additionalLabels map[string]string, query *StackdriverQuery) string {
|
|
|
+func formatLegendKeys(metricType string, defaultMetricName string, resourceType string, metricLabels map[string]string, resourceLabels map[string]string, additionalLabels map[string]string, query *StackdriverQuery) string {
|
|
|
if query.AliasBy == "" {
|
|
|
return defaultMetricName
|
|
|
}
|
|
|
@@ -469,6 +497,10 @@ func formatLegendKeys(metricType string, defaultMetricName string, metricLabels
|
|
|
return []byte(metricType)
|
|
|
}
|
|
|
|
|
|
+ if metaPartName == "resource.type" && resourceType != "" {
|
|
|
+ return []byte(resourceType)
|
|
|
+ }
|
|
|
+
|
|
|
metricPart := replaceWithMetricPart(metaPartName, metricType)
|
|
|
|
|
|
if metricPart != nil {
|
|
|
@@ -550,8 +582,6 @@ func (e *StackdriverExecutor) createRequest(ctx context.Context, dsInfo *models.
|
|
|
if !ok {
|
|
|
return nil, errors.New("Unable to find datasource plugin Stackdriver")
|
|
|
}
|
|
|
- projectName := dsInfo.JsonData.Get("defaultProject").MustString()
|
|
|
- proxyPass := fmt.Sprintf("stackdriver%s", "v3/projects/"+projectName+"/timeSeries")
|
|
|
|
|
|
var stackdriverRoute *plugins.AppPluginRoute
|
|
|
for _, route := range plugin.Routes {
|
|
|
@@ -561,7 +591,22 @@ func (e *StackdriverExecutor) createRequest(ctx context.Context, dsInfo *models.
|
|
|
}
|
|
|
}
|
|
|
|
|
|
+ projectName := dsInfo.JsonData.Get("defaultProject").MustString()
|
|
|
+ proxyPass := fmt.Sprintf("stackdriver%s", "v3/projects/"+projectName+"/timeSeries")
|
|
|
+
|
|
|
pluginproxy.ApplyRoute(ctx, req, proxyPass, stackdriverRoute, dsInfo)
|
|
|
|
|
|
return req, nil
|
|
|
}
|
|
|
+
|
|
|
+func (e *StackdriverExecutor) getDefaultProject(ctx context.Context) (string, error) {
|
|
|
+ authenticationType := e.dsInfo.JsonData.Get("authenticationType").MustString(jwtAuthentication)
|
|
|
+ if authenticationType == gceAuthentication {
|
|
|
+ defaultCredentials, err := google.FindDefaultCredentials(ctx, "https://www.googleapis.com/auth/monitoring.read")
|
|
|
+ if err != nil {
|
|
|
+ return "", fmt.Errorf("Failed to retrieve default project from GCE metadata server. error: %v", err)
|
|
|
+ }
|
|
|
+ return defaultCredentials.ProjectID, nil
|
|
|
+ }
|
|
|
+ return e.dsInfo.JsonData.Get("defaultProject").MustString(), nil
|
|
|
+}
|