Data Source API: Add support for Amazon Redshift and Clickhouse (#166)

Co-authored-by: Lukas Pfannschmidt <[email protected]>
grafana-toolbox · Mar 29, 2024 · f091384 · f091384
1 parent 1410eb1
commit f091384
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -13,6 +13,8 @@
   endpoints, and by adding missing ``move_folder``. Thanks, @grafuls.
 * Service Accounts: Unlock endpoint to update service account by id.
   Thanks, @einar-lanfranco.
+* Data Source API: Add support for Amazon Redshift and Clickhouse.
+  Thanks, @lpfann and @mysuperai.
 
 [Niquests]: https://niquests.readthedocs.io/
 [Riquests]: https://requests.readthedocs.io/

diff --git a/grafana_client/knowledge.py b/grafana_client/knowledge.py
@@ -288,7 +288,68 @@ def query_factory(datasource, model: Optional[dict] = None, expression: Optional
                 "default": None,
             },
         ]
+    elif datasource_type == "grafana-redshift-datasource":
+        query = {
+            "datasource": {
+                "type": datasource["type"],
+                "uid": datasource.get("uid"),
+            },
+            "datasourceId": datasource.get("id"),
+            "rawSQL": expression,
+        }
 
+        attrs = [
+            {
+                "name": "format",
+                "default": 0,
+                "choices": [0, 1],
+            },
+            {
+                "name": "intervalMs",
+                "default": 30000,
+            },
+            {
+                "name": "maxDataPoints",
+                "default": 1441,
+            },
+            {
+                "name": "refId",
+                "default": None,
+            },
+        ]
+    elif datasource_type == "grafana-clickhouse-datasource":
+        query = {
+            "datasource": {
+                "type": datasource["type"],
+                "uid": datasource.get("uid"),
+            },
+            "datasourceId": datasource.get("id"),
+            "rawSql": expression,
+        }
+
+        attrs = [
+            {
+                "name": "format",
+                "default": "0",
+                "choices": [0, 1],
+            },
+            {
+                "name": "intervalMs",
+                "default": 30000,
+            },
+            {
+                "name": "maxDataPoints",
+                "default": 1441,
+            },
+            {
+                "name": "refId",
+                "default": None,
+            },
+            {
+                "name": "builderOptions",
+                "default": None
+            }
+        ]
     elif datasource_type == "prometheus":
         query = {
             "datasource": {