Spaces:
Build error
Build error
File size: 1,303 Bytes
322135d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
#!/usr/bin/env bash
# Generate a docker-compose configuration for a cluster.
#
# Parameters:
# - $1: number of nodes (default: 5)
# - $2: base port (default: 6330)
#
# Example usage:
# ./generate_docker_compose_cluster.sh 5 > docker-compose.yaml
set -euo pipefail
declare NODES="${1:-5}"
declare BASE_PORT="${2:-6330}"
cat <<-EOF
version: "3.7"
services:
EOF
for ((node=0; node<NODES; node++))
do
declare SERVICE_NAME=qdrant_node_$node
declare HTTP_PORT=$((BASE_PORT + node * 10 + 3))
declare GRPC_PORT=$((BASE_PORT + node * 10 + 4))
if ((node == 0))
then
declare COMMAND="./qdrant --uri 'http://$SERVICE_NAME:6335'"
else
declare COMMAND="bash -c \"sleep $((10 + node / 10 + RANDOM % 10)) && ./qdrant --bootstrap 'http://qdrant_node_0:6335' --uri 'http://$SERVICE_NAME:6335'\""
fi
cat <<-EOF
$SERVICE_NAME:
image: qdrant/qdrant:latest
command: $COMMAND
restart: always
environment:
- QDRANT__SERVICE__GRPC_PORT=6334
- QDRANT__CLUSTER__ENABLED=true
- QDRANT__CLUSTER__P2P__PORT=6335
- QDRANT__CLUSTER__CONSENSUS__MAX_MESSAGE_QUEUE_SIZE=5000
- QDRANT__LOG_LEVEL=debug,raft=info
ports:
- "$HTTP_PORT:6333"
- "$GRPC_PORT:6334"
deploy:
resources:
limits:
cpus: "0.3"
EOF
done
|