ambari-metrics-grafana/ambari-metrics/query_ctrl.ts

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

///<reference path="../../../headers/common.d.ts" />

import 'angular';
import _ from 'lodash';
import kbn from 'app/core/utils/kbn';
import {QueryCtrl} from 'app/plugins/sdk';

export class AmbariMetricsQueryCtrl extends QueryCtrl {

    static templateUrl = 'partials/query.editor.html';
    aggregators: any;
    aggregator: any;
    errors: any;
    precisions: any;
    transforms: any;
    transform: any;
    precisionInit: any;
    suggestMetrics: any;
    suggestApps: any;
    suggestHosts: any;
    seriesAggregators: any;

    /** @ngInject **/
    constructor($scope, $injector) {
        super($scope, $injector);
        this.errors = this.validateTarget(this.target);
        this.aggregators = ['none','avg', 'sum', 'min', 'max'];
        this.precisions = ['default','seconds', 'minutes', 'hours', 'days'];
        this.transforms = ['none','diff','rate'];
        this.seriesAggregators = ['none', 'avg', 'sum', 'min', 'max'];

        if (!this.target.aggregator) {
            this.target.aggregator = 'avg';
        }
        this.precisionInit = function () {
            if (typeof this.target.precision == 'undefined') {
                this.target.precision = "default";
            }
        };
        this.transform = function () {
            if (typeof this.target.transform == 'undefined') {
                this.target.transform = "none";
            }
        };
        this.seriesAggregator = function () {
            if (typeof $scope.target.seriesAggregator == 'undefined') {
                this.target.seriesAggregator = "none";
            }
        };
        $scope.$watch('target.app', function (newValue) {
            if (newValue === '') {
                this.target.metric = '';
                this.target.hosts = '';
                this.target.cluster = '';
            }
        });
        if (!this.target.downsampleAggregator) {
            this.target.downsampleAggregator = 'avg';
        }

        this.datasource.getAggregators().then(function(aggs) {
            this.aggregators = aggs;
        });

        this.suggestApps = (query, callback) => {
            this.datasource.suggestApps(query)
                .then(this.getTextValues)
                .then(callback);
        };

        this.suggestClusters = (query, callback) => {
            this.datasource.suggestClusters(this.target.app)
                .then(this.getTextValues)
                .then(callback);
        };

        this.suggestHosts = (query, callback) => {
            this.datasource.suggestHosts(this.target.app, this.target.cluster)
                .then(this.getTextValues)
                .then(callback);
        };

        this.suggestMetrics = (query, callback) => {
            this.datasource.suggestMetrics(query, this.target.app)
                .then(this.getTextValues)
                .then(callback);
        };

        this.suggestTagKeys = (query, callback) => {
            this.datasource.metricFindQuery('tag_names(' + this.target.metric + ')')
                .then(this.getTextValues)
                .then(callback);
        };

        this.suggestTagValues = (query, callback) => {
            this.datasource.metricFindQuery('tag_values(' + this.target.metric + ',' + this.target.currentTagKey + ')')
                .then(this.getTextValues)
                .then(callback);
        };

        this.getTextValues = (metricFindResult) => {
            return metricFindResult.map((value) => {return value.text });
        }
    }

    targetBlur () {
        this.target.errors = this.validateTarget(this.target);
        this.refresh();
    };
    

    addTag () {
        if (!this.addTagMode) {
            this.addTagMode = true;
            return;
        }

        if (!this.target.tags) {
            this.target.tags = {};
        }

        this.target.errors = this.validateTarget(this.target);

        if (!this.target.errors.tags) {
            this.target.tags[this.target.currentTagKey] = this.target.currentTagValue;
            this.target.currentTagKey = '';
            this.target.currentTagValue = '';
            this.targetBlur();
        }

        this.addTagMode = false;
    };

    removeTag (key) {
        delete this.target.tags[key];
        this.targetBlur();
    };

    getCollapsedText () {
        var text = this.target.metric + ' on ' + this.target.app;
        return text;
    };

    validateTarget (target) {
        var errs = {};

        if (target.tags && _.has(target.tags, target.currentTagKey)) {
            errs.tags = "Duplicate tag key '" + target.currentTagKey + "'.";
        }

        return errs;
    }
}