From 98fd88f9cda1052bc30612c38a26d958612693ed Mon Sep 17 00:00:00 2001
From: Alexander Zobnin <alexanderzobnin@gmail.com>
Date: Fri, 22 Jan 2016 00:19:40 +0300
Subject: [PATCH] Creates query builder service.

---
 plugins/datasource-zabbix/datasource.js   |   3 +-
 plugins/datasource-zabbix/queryBuilder.js | 157 ++++++++++++++++++++++
 2 files changed, 158 insertions(+), 2 deletions(-)
 create mode 100644 plugins/datasource-zabbix/queryBuilder.js

diff --git a/plugins/datasource-zabbix/datasource.js b/plugins/datasource-zabbix/datasource.js
index 0691a80..54181a9 100644
--- a/plugins/datasource-zabbix/datasource.js
+++ b/plugins/datasource-zabbix/datasource.js
@@ -13,8 +13,7 @@ function (angular, _, dateMath) {
   'use strict';
 
   /** @ngInject */
-  function ZabbixAPIDatasource(instanceSettings, $q, backendSrv, templateSrv, alertSrv,
-                                ZabbixAPI, Utils, zabbixHelperSrv, ZabbixCache) {
+  function ZabbixAPIDatasource(instanceSettings, $q, backendSrv, templateSrv, alertSrv, ZabbixAPI, Utils, zabbixHelperSrv, ZabbixCache) {
 
     // General data source settings
     this.name             = instanceSettings.name;
diff --git a/plugins/datasource-zabbix/queryBuilder.js b/plugins/datasource-zabbix/queryBuilder.js
new file mode 100644
index 0000000..5293409
--- /dev/null
+++ b/plugins/datasource-zabbix/queryBuilder.js
@@ -0,0 +1,157 @@
+define([
+  'angular',
+  'lodash',
+  './zabbixCacheSrv',
+  './utils'
+],
+function (angular, _) {
+  'use strict';
+
+  var module = angular.module('grafana.services');
+
+  module.service('QueryBuilderSrv', function(ZabbixCache, Utils) {
+
+    this.build = function (groupFilter, hostFilter, appFilter, itemFilter) {
+
+      // Find items by item names and perform queries
+      var groups = [];
+      var hosts = [];
+      var apps = [];
+      var items = [];
+
+      if (Utils.isRegex(hostFilter)) {
+
+        // Filter groups
+        if (Utils.isRegex(groupFilter)) {
+          var groupPattern = Utils.buildRegex(groupFilter);
+          groups = _.filter(ZabbixCache.getGroups(), function (groupObj) {
+            return groupPattern.test(groupObj.name);
+          });
+        } else {
+          var findedGroup = _.find(ZabbixCache.getGroups(), {'name': groupFilter});
+          if (findedGroup) {
+            groups.push(findedGroup);
+          } else {
+            groups = undefined;
+          }
+        }
+        if (groups) {
+          var groupids = _.map(groups, 'groupid');
+          hosts = _.filter(ZabbixCache.getHosts(), function (hostObj) {
+            return _.intersection(groupids, hostObj.groups).length;
+          });
+        } else {
+          // No groups finded
+          return [];
+        }
+
+        // Filter hosts
+        var hostPattern = Utils.buildRegex(hostFilter);
+        hosts = _.filter(hosts, function (hostObj) {
+          return hostPattern.test(hostObj.name);
+        });
+      } else {
+        var findedHost = _.find(ZabbixCache.getHosts(), {'name': hostFilter});
+        if (findedHost) {
+          hosts.push(findedHost);
+        } else {
+          // No hosts finded
+          return [];
+        }
+      }
+
+      // Find items belongs to selected hosts
+      items = _.filter(ZabbixCache.getItems(), function (itemObj) {
+        return _.contains(_.map(hosts, 'hostid'), itemObj.hostid);
+      });
+
+      if (Utils.isRegex(itemFilter)) {
+
+        // Filter applications
+        if (Utils.isRegex(appFilter)) {
+          var appPattern = Utils.buildRegex(appFilter);
+          apps = _.filter(ZabbixCache.getApplications(), function (appObj) {
+            return appPattern.test(appObj.name);
+          });
+        }
+        // Don't use application filter if it empty
+        else if (appFilter === "") {
+          apps = undefined;
+        }
+        else {
+          var findedApp = _.find(ZabbixCache.getApplications(), {'name': appFilter});
+          if (findedApp) {
+            apps.push(findedApp);
+          } else {
+            // No applications finded
+            return [];
+          }
+        }
+
+        // Find items belongs to selected applications
+        if (apps) {
+          var appids = _.flatten(_.map(apps, 'applicationids'));
+          items = _.filter(items, function (itemObj) {
+            return _.intersection(appids, itemObj.applications).length;
+          });
+        }
+
+        if (items) {
+          var itemPattern = Utils.buildRegex(itemFilter);
+          items = _.filter(items, function (itemObj) {
+            return itemPattern.test(itemObj.name);
+          });
+        } else {
+          // No items finded
+          return [];
+        }
+      } else {
+        items = _.filter(items, {'name': hostFilter});
+        if (!items.length) {
+          // No items finded
+          return [];
+        }
+      }
+
+      // Set host as host name for each item
+      items = _.each(items, function (itemObj) {
+        itemObj.host = _.find(hosts, {'hostid': itemObj.hostid}).name;
+      });
+
+      // Use alias only for single metric, otherwise use item names
+      var alias;
+      if (items.length === 1) {
+        alias = templateSrv.replace(alias, options.scopedVars);
+      }
+
+      var history;
+      if ((from < useTrendsFrom) && self.trends) {
+        // Use trends
+        var points = downsampleFunction ? downsampleFunction.value : "avg";
+        history = self.zabbixAPI.getTrends(items, from, to)
+          .then(_.bind(zabbixHelperSrv.handleTrendResponse, zabbixHelperSrv, items, alias, scale, points));
+      } else {
+        // Use history
+        history = self.zabbixAPI.getHistory(items, from, to)
+          .then(_.bind(zabbixHelperSrv.handleHistoryResponse, zabbixHelperSrv, items, alias, scale));
+      }
+
+      return history.then(function (timeseries) {
+        var timeseries_data = _.flatten(timeseries);
+        return _.map(timeseries_data, function (timeseries) {
+
+          // Series downsampling
+          if (timeseries.datapoints.length > options.maxDataPoints) {
+            var ms_interval = Math.floor((to - from) / options.maxDataPoints) * 1000;
+            var downsampleFunc = downsampleFunction ? downsampleFunction.value : "avg";
+            timeseries.datapoints = zabbixHelperSrv.downsampleSeries(timeseries.datapoints, to, ms_interval, downsampleFunc);
+          }
+          return timeseries;
+        });
+      });
+
+      return itemFilter;
+    };
+
+  });
+});
\ No newline at end of file