From b7c7030a4681e3c5c8d3e565588a02e9e8a2e9db Mon Sep 17 00:00:00 2001
From: Sven Klemm <sven.klemm@sql-ag.de>
Date: Tue, 13 Mar 2018 23:06:39 +0100
Subject: [PATCH] add regex operators

---
 .../datasource/postgres/query_builder.ts       |  8 ++++++++
 .../plugins/datasource/postgres/query_ctrl.ts  | 18 ++++++++++++++++--
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/public/app/plugins/datasource/postgres/query_builder.ts b/public/app/plugins/datasource/postgres/query_builder.ts
index 7a227f33b93..23691830b17 100644
--- a/public/app/plugins/datasource/postgres/query_builder.ts
+++ b/public/app/plugins/datasource/postgres/query_builder.ts
@@ -47,4 +47,12 @@ export class PostgresQueryBuilder {
     return query;
   }
 
+  buildDatatypeQuery(column: string) {
+    var query = "SELECT data_type FROM information_schema.columns WHERE ";
+    query += " table_schema = " + this.queryModel.quoteLiteral(this.target.schema);
+    query += " AND table_name = " + this.queryModel.quoteLiteral(this.target.table);
+    query += " AND column_name = " + this.queryModel.quoteLiteral(column);
+    return query;
+  }
+
 }
diff --git a/public/app/plugins/datasource/postgres/query_ctrl.ts b/public/app/plugins/datasource/postgres/query_ctrl.ts
index afde342474b..b84b0ce0750 100644
--- a/public/app/plugins/datasource/postgres/query_ctrl.ts
+++ b/public/app/plugins/datasource/postgres/query_ctrl.ts
@@ -266,14 +266,28 @@ export class PostgresQueryCtrl extends QueryCtrl {
   }
 
   getWhereSegments(segment, index) {
+    var query, addTemplateVars;
+
     if (segment.type === 'condition') {
       return this.$q.when([this.uiSegmentSrv.newSegment('AND'), this.uiSegmentSrv.newSegment('OR')]);
     }
     if (segment.type === 'operator') {
-      return this.$q.when(this.uiSegmentSrv.newOperators(['=', '!=', '<>', '<', '>']));
+      var columnName = this.whereSegments[index - 1].value;
+      query = this.queryBuilder.buildDatatypeQuery(columnName);
+      return this.datasource.metricFindQuery(query)
+      .then(results => {
+        var datatype = results[0].text;
+        switch (datatype) {
+          case "text":
+          case "character varying":
+            return this.$q.when(this.uiSegmentSrv.newOperators(['=', '!=', '~', '~*','!~','!~*','IN']));
+          default:
+            return this.$q.when(this.uiSegmentSrv.newOperators(['=', '!=', '<', '<=', '>', '>=']));
+        }
+      })
+      .catch(this.handleQueryError.bind(this));
     }
 
-    var query, addTemplateVars;
     if (segment.type === 'key' || segment.type === 'plus-button') {
       query = this.queryBuilder.buildColumnQuery();