From 1344aeb04d829dd32a8e32e901e6015c58dd3376 Mon Sep 17 00:00:00 2001
From: Alex Gherghisan <alexg@aztecprotocol.com>
Date: Tue, 26 Nov 2024 13:13:25 +0000
Subject: [PATCH] fix: provernet docker compose

---
 docker-compose.provernet.yml | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/docker-compose.provernet.yml b/docker-compose.provernet.yml
index 83e6cd1913fe..7b398b0b0a37 100644
--- a/docker-compose.provernet.yml
+++ b/docker-compose.provernet.yml
@@ -65,7 +65,8 @@ services:
       ARCHIVER_POLLING_INTERVAL_MS: 1000
       ARCHIVER_VIEM_POLLING_INTERVAL_MS: 1000
       PROVER_VIEM_POLLING_INTERVAL_MS: 1000
-      PROVER_AGENT_ENABLED: false
+      PROVER_AGENT_COUNT: 0
+      PROVER_BROKER_URL: http://aztec-prover-broker
       PROVER_PUBLISHER_PRIVATE_KEY: "0xdbda1821b80551c9d65939329250298aa3472ba22feea921c0cf5d620ea67b97"
       PROVER_REAL_PROOFS: "${PROVER_REAL_PROOFS:-false}"
       PROVER_MINIMUM_ESCROW_AMOUNT: 1000000000
@@ -76,6 +77,8 @@ services:
     depends_on:
       aztec-node:
         condition: service_healthy
+      aztec-prover-broker:
+        condition: service_healthy
     healthcheck:
       test: [ "CMD", "curl", "-fSs", "http://127.0.0.1:80/status" ]
       interval: 3s
@@ -84,6 +87,21 @@ services:
     command: [ "start", "--prover-node", "--archiver" ]
     restart: on-failure:5
 
+  aztec-prover-broker:
+    image: "aztecprotocol/${IMAGE:-aztec:master}"
+    ports:
+      - "8084:80"
+    environment:
+      LOG_LEVEL: verbose
+      AZTEC_PORT: 80
+    healthcheck:
+      test: [ "CMD", "curl", "-fSs", "http://127.0.0.1:80/status" ]
+      interval: 3s
+      timeout: 30s
+      start_period: 120s
+    command: [ "start", "--prover-broker" ]
+    restart: on-failure:5
+
   # Prover agent that connects to the prover-node for fetching proving jobs and executing them
   # Multiple instances can be run, or PROVER_AGENT_CONCURRENCY can be increased to run multiple workers in a single instance
   aztec-prover-agent:
@@ -93,13 +111,11 @@ services:
     environment:
       LOG_LEVEL: verbose
       ETHEREUM_HOST: http://ethereum:8545
-      AZTEC_NODE_URL: http://aztec-prover # Deprecated, use PROVER_JOB_SOURCE_URL
-      PROVER_JOB_SOURCE_URL: http://aztec-prover
+      PROVER_BROKER_URL: http://aztec-prover-broker
       L1_CHAIN_ID: 31337
       AZTEC_PORT: 80
       PROVER_REAL_PROOFS: "${PROVER_REAL_PROOFS:-false}"
       PROVER_TEST_DELAY_MS: "${PROVER_TEST_DELAY_MS:-0}"
-      PROVER_AGENT_CONCURRENCY: 2
       BB_SKIP_CLEANUP: "${BB_SKIP_CLEANUP:-0}" # Persist tmp dirs for debugging
       PROVER_ID: "${PROVER_ID:-0x01}"
     volumes:
@@ -107,9 +123,12 @@ services:
       - ./cache/bb-crs/:/root/.bb-crs:rw
       - ./workdir/bb-prover/:/usr/src/yarn-project/bb:rw
     depends_on:
-      aztec-prover:
+      aztec-prover-broker:
         condition: service_healthy
-    command: [ "start", "--prover" ]
+    command: [ "start", "--prover-agent" ]
+    deploy:
+      mode: replicated
+      replicas: 2
     restart: on-failure:5
     healthcheck:
       test: [ "CMD", "curl", "-fSs", "http://127.0.0.1:80/status" ]