From 2f78bd7a21538805d5d4cd56031ae5acf8c5ae87 Mon Sep 17 00:00:00 2001
From: Rajesh Rajendran <rjshrjndrn@users.noreply.github.com>
Date: Sun, 4 Jul 2021 11:19:20 +0000
Subject: [PATCH] feat(install): override default db resource allocation (#74)

If you need to override the default resource allocation or dbs, which is
indented to be for small machines, you can override the default
allocation here.

Ref: https://github.com/openreplay/openreplay/issues/64

Signed-off-by: Rajesh Rajendran <rjshrjndrn@gmail.com>
---
 .../roles/openreplay/templates/clickhouse.yaml     |  4 ++++
 .../roles/openreplay/templates/postgresql.yaml     |  7 +++++--
 scripts/helm/roles/openreplay/templates/redis.yaml |  6 +++++-
 scripts/helm/vars.yaml                             | 14 ++++++++++++++
 4 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/scripts/helm/roles/openreplay/templates/clickhouse.yaml b/scripts/helm/roles/openreplay/templates/clickhouse.yaml
index 1b6929bff..79fd26bcb 100644
--- a/scripts/helm/roles/openreplay/templates/clickhouse.yaml
+++ b/scripts/helm/roles/openreplay/templates/clickhouse.yaml
@@ -1,3 +1,6 @@
+{% if db_resource_override.clickhouse %}{
+{{ db_resource_override.clickhouse|to_nice_yaml(indent=2) }}
+{% else %}
 resources:
   limits:
     cpu: {{ ( 100 * scale|float ) | int }}m
@@ -5,3 +8,4 @@ resources:
   requests:
     cpu: 100m
     memory: 128Mi
+{% endif %}
diff --git a/scripts/helm/roles/openreplay/templates/postgresql.yaml b/scripts/helm/roles/openreplay/templates/postgresql.yaml
index f9e208797..b9855d36a 100644
--- a/scripts/helm/roles/openreplay/templates/postgresql.yaml
+++ b/scripts/helm/roles/openreplay/templates/postgresql.yaml
@@ -1,8 +1,11 @@
+{% if db_resource_override.postgresql %}
+{{ db_resource_override.postgresql|to_nice_yaml(indent=2) }}
+{% else %}
 resources:
   limits:
-    cpu: {{ ( 250 * scale|float ) | int }}m
+    cpu: {{ ( 250 * scale|float ) | int }/db_resource_override/db_resource_override//db_resource_override/db_resource_override/gm
     memory: {{ ( 512 * scale|float ) | int }}Mi
   requests:
     cpu: 250m
     memory: 256Mi
-
+{% endif %}
diff --git a/scripts/helm/roles/openreplay/templates/redis.yaml b/scripts/helm/roles/openreplay/templates/redis.yaml
index 589caa57a..e7b0a4cba 100644
--- a/scripts/helm/roles/openreplay/templates/redis.yaml
+++ b/scripts/helm/roles/openreplay/templates/redis.yaml
@@ -1,7 +1,11 @@
 fullnameOverride: redis
+usePassword: false
 cluster:
   enabled: false
 redis:
+{% if db_resource_override.redis %}
+  {{ db_resource_override.redis|to_nice_yaml(indent=2) }}
+{% else %}
   resources:
     limits:
       cpu: {{ ( 100 * scale|float ) | int }}m
@@ -9,4 +13,4 @@ redis:
     requests:
       cpu: 100m
       memory: 128Mi
-usePassword: false
+{% endif %}
diff --git a/scripts/helm/vars.yaml b/scripts/helm/vars.yaml
index 1ea4cb879..631ba5006 100644
--- a/scripts/helm/vars.yaml
+++ b/scripts/helm/vars.yaml
@@ -72,3 +72,17 @@ enable_monitoring: "false"
 #
 # Username: admin
 grafana_password: ""
+
+## Advanced
+# If you need to override the default cpu/memory allocation of databases.
+db_resource_override:
+  postgresql: {}
+    # resources:
+    #   limits:
+    #     cpu: 1000m
+    #     memory: 1024Mi
+    #   requests:
+    #     cpu: 250m
+    #     memory: 256Mi
+  redis: {}
+  clickhouse: {}