From 078ef0627943c6b2989c3abf543da89868221e92 Mon Sep 17 00:00:00 2001
From: Lalit Arvind <lalitarvind.balaji@gmail.com>
Date: Sat, 16 Nov 2024 15:55:19 -0700
Subject: [PATCH] Jinja file and update script to add service in docker compose
 file

---
 .../config_update_scripts/update_compose.py   | 42 +++++++++++++++++++
 clickhouse/jinja-templates/service.yml.jinja  | 30 +++++++++++++
 2 files changed, 72 insertions(+)
 create mode 100644 clickhouse/config_update_scripts/update_compose.py
 create mode 100644 clickhouse/jinja-templates/service.yml.jinja

diff --git a/clickhouse/config_update_scripts/update_compose.py b/clickhouse/config_update_scripts/update_compose.py
new file mode 100644
index 0000000..76f72fd
--- /dev/null
+++ b/clickhouse/config_update_scripts/update_compose.py
@@ -0,0 +1,42 @@
+import yaml
+from jinja2 import Environment, FileSystemLoader
+import subprocess
+import json
+if __name__ == "__main__":
+    
+    # extracting details of each running container in json format    
+    try:
+        all_services = subprocess.check_output(["docker","ps","--format","json"],text=True).split('\n')[:-1]
+    except subprocess.CalledProcessError as e:
+        print(f"Command failed with return code {e.returncode}")
+
+    all_services = [json.loads(s) for s in all_services]
+    # extracting the name, removing the custom id from it and storing it in a list
+    all_service_names = [service['Names'].split('.')[0] for service in all_services]
+    # extracting only 'keeper1', 'server1'...
+    all_service_names = [ name.split('-')[-1] for name in all_service_names]
+
+    # removing all keeepers
+    all_service_names.remove('keeper1')
+    all_service_names.remove('keeper2')
+    all_service_names.remove('keeper3')
+    curr_num_servers = sorted(all_service_names)[-1][-1]
+
+    replication_factor = 2
+    curr_num_shards = curr_num_servers/replication_factor
+
+    env = Environment(loader=FileSystemLoader('.'))
+    template = env.get_template('service.yml.jinja')
+
+    with open('../docker-compose.yaml','r') as f:
+        compose_f = yaml.safe_load(f)
+    
+    new_service1 = template.render(server_num=curr_num_servers+1)
+    new_service2 = template.render(server_num=curr_num_servers+1)
+
+    compose_f['services'].update(new_service1)
+    compose_f['services'].update(new_service2)
+
+    if compose_f:
+        with open('../docker-compose.yaml','w') as yamlfile:
+            yaml.safe_dump(compose_f, yamlfile)
\ No newline at end of file
diff --git a/clickhouse/jinja-templates/service.yml.jinja b/clickhouse/jinja-templates/service.yml.jinja
new file mode 100644
index 0000000..1ae0414
--- /dev/null
+++ b/clickhouse/jinja-templates/service.yml.jinja
@@ -0,0 +1,30 @@
+clickhouse-server{{server_num}}:
+    image: clickhouse/clickhouse-server:latest
+    container_name: clickhouse-server{{server_num}}
+    volumes:
+      - ./node{{server_num}}-config/:/etc/clickhouse-server/config.d/
+      - clickhouse_data{{server_num}}:/var/lib/clickhouse
+    networks:
+      clickhouse-server-network:
+        aliases:
+          - clickhouse-server{{server_num}}
+      clickhouse-keeper-network:
+        aliases:
+          - clickhouse-server{{server_num}}
+    deploy:
+      replicas: 1
+      # placement:
+      #   constraints: [node.labels.role == server]
+      update_config:
+        delay: 10s
+      # resources:
+      #   limits:
+      #     cpus: "0.50"
+      #     memory: 100M
+    depends_on:
+      - clickhouse-keeper1
+      - clickhouse-keeper2
+      - clickhouse-keeper3
+    ports:
+      - "900{{server_num}}:9000"  # Native client port
+      - "8123:8123"  # HTTP interface