docker-compose.yml

version: '3'

services:

  server:
    build:
      context: server
      args:
        - MODEL_NAME=identity
        - SERVING_PORT=8500
    environment:
      - MODEL_NAME=identity
      - SERVING_PORT=8500
    ports:
      - "8500:8500"

  wsgi-client:
    build:
      context: client
      args:
        - CLIENT_PORT=8000
    command:
      gunicorn --workers=4 --bind=0.0.0.0:8000 wsgi_client:api
    environment:
      - MODEL_NAME=identity
      - SERVING_HOST=server
      - SERVING_PORT=8500
    ports:
      - "8000:8000"
    links:
      - server

  tornado-client:
    build:
      context: client
      args:
        - CLIENT_PORT=8001
    command:
      python3 tornado_client.py --client_port=8001
    environment:
      - MODEL_NAME=identity
      - SERVING_HOST=server
      - SERVING_PORT=8500
    ports:
      - "8001:8001"
    links:
      - server

  grpc-benchmark:
    build:
      context: client
    command:
      python3 grpc_client.py --num_requests=10000 --max_concurrent=10
    environment:
      - MODEL_NAME=identity
      - SERVING_HOST=server
      - SERVING_PORT=8500
    links:
      - server

  wsgi-benchmark:
    build:
      context: client
    command:
      ab -n 10000 -c 10 http://wsgi-client:8000/prediction
    links:
      - server
      - wsgi-client

  tornado-benchmark:
    build:
      context: client
    command:
      ab -n 10000 -c 10 http://tornado-client:8001/prediction
    links:
      - server
      - tornado-client