From 072e9bc4f944e1620bfd0c70e3878b05ceec47a5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89mile=20R=C3=A9?= <emile@getprobo.com>
Date: Tue, 28 Apr 2026 19:17:40 +0400
Subject: [PATCH] Wait for healthy stack before start probod
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Émile Ré <emile@getprobo.com>
---
 compose.yaml              |  2 +-
 contrib/lima/provision.sh | 11 +++++++----
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/compose.yaml b/compose.yaml
index cf5ea9850..bb18d22db 100644
--- a/compose.yaml
+++ b/compose.yaml
@@ -4,7 +4,7 @@ services:
     pull_policy: missing
     shm_size: "1g"
     command: >
-      postgres -c "shared_buffers=1GB"
+      postgres -c "shared_buffers=256MB"
                -c "max_connections=200"
                -c "log_statement=all"
     ports:
diff --git a/contrib/lima/provision.sh b/contrib/lima/provision.sh
index 3ea34f8b4..164846ab8 100755
--- a/contrib/lima/provision.sh
+++ b/contrib/lima/provision.sh
@@ -30,6 +30,7 @@ apt-get install -y -qq \
     ca-certificates \
     gnupg \
     lsb-release \
+    postgresql-client \
     unzip
 
 if ! command -v docker &>/dev/null; then
@@ -178,10 +179,11 @@ Requires=docker.service
 After=docker.service
 
 [Service]
-Type=simple
+Type=oneshot
+RemainAfterExit=yes
 User=root
 WorkingDirectory=/workspace
-ExecStart=/usr/bin/docker compose -f compose.yaml up
+ExecStart=/usr/bin/docker compose -f compose.yaml up -d --wait
 ExecStop=/usr/bin/docker compose -f compose.yaml down
 Restart=on-failure
 RestartSec=5s
@@ -200,6 +202,7 @@ After=probo-stack.service
 Type=simple
 User=${LIMA_USER}
 WorkingDirectory=/workspace
+ExecStartPre=/bin/bash -c 'for i in \$(seq 1 10); do pg_isready -h localhost -p 5432 -U probod -d probod -q && exit 0; sleep 1; done; echo "PostgreSQL not ready after 10s"; exit 1'
 ExecStart=/usr/local/bin/gow -r=false run ./cmd/probod -cfg-file /etc/probod/config.yml -format pretty
 Restart=on-failure
 RestartSec=3s
@@ -213,7 +216,7 @@ cat > /etc/systemd/system/probo-console.service << EOF
 [Unit]
 Description=Probo Console Dev Server
 Requires=probo-node-modules.service
-After=probo-node-modules.service
+After=probo-node-modules.service probod.service
 
 [Service]
 Type=simple
@@ -231,7 +234,7 @@ cat > /etc/systemd/system/probo-trust.service << EOF
 [Unit]
 Description=Probo Trust Dev Server
 Requires=probo-node-modules.service
-After=probo-node-modules.service
+After=probo-node-modules.service probod.service
 
 [Service]
 Type=simple