# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
version: '3.3'
services:
  ballista-scheduler:
    image: ballista-scheduler
    build:
      dockerfile: dev/docker/ballista-scheduler.Dockerfile
      context: .
    command: "--bind-host 0.0.0.0"
    ports:
      - "50050:50050"
    environment:
      - RUST_LOG=ballista=info
    volumes:
      - ./benchmarks/data:/data
    healthcheck:
      test: ["CMD", "nc", "-z", "ballista-scheduler", "50050"]
      interval: 5s
      timeout: 5s
      retries: 5
  ballista-executor:
    image: ballista-executor
    build:
      dockerfile: dev/docker/ballista-executor.Dockerfile
      context: .
    command: "--bind-host 0.0.0.0 --scheduler-host ballista-scheduler --scheduler-connect-timeout-seconds 15"
    deploy:
      replicas: 2
    restart: always
    environment:
      - RUST_LOG=ballista=info
    volumes:
      - ./benchmarks/data:/data
    depends_on:
      - ballista-scheduler
    healthcheck:
      test: ["CMD", "nc", "-z", "ballista-executor", "50051"]
      interval: 5s
      timeout: 5s
      retries: 5
  ballista-client:
    image: ballista-benchmarks
    build:
      dockerfile: dev/docker/ballista-benchmarks.Dockerfile
      context: .
    command: ["/bin/bash", "-c", "sleep infinity"]
    ports:
      - "50051:50051"
    environment:
      - RUST_LOG=info
    volumes:
      - ./benchmarks/data:/data
    depends_on:
      - ballista-scheduler
      - ballista-executor