banzai/docker-compose-site.yml at main · LCOGT/banzai · GitHub

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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
services:
  redis:
    image: redis:7
    container_name: banzai-redis
    restart: unless-stopped

  rabbitmq:
    image: rabbitmq:3.12-management
    container_name: banzai-rabbitmq
    ports:
      - "5672:5672"
      - "15672:15672"
    restart: unless-stopped

  banzai-worker:
    build: .
    container_name: banzai-worker
    depends_on:
      - redis
      - rabbitmq
    volumes:
      - ${HOST_DATA_DIR}:/data
      - ${HOST_PROCESSED_DIR}:/output
    environment:
      - TASK_HOST=${TASK_HOST}
      - FITS_BROKER=${FITS_BROKER}
      - FITS_EXCHANGE=${FITS_EXCHANGE}
      - CELERY_TASK_QUEUE_NAME=${CELERY_TASK_QUEUE_NAME}
      - BANZAI_WORKER_LOGLEVEL=${BANZAI_WORKER_LOGLEVEL}
      - OMP_NUM_THREADS=${OMP_NUM_THREADS}
      - OPENTSDB_PYTHON_METRICS_TEST_MODE=${OPENTSDB_PYTHON_METRICS_TEST_MODE}
      - API_ROOT=${API_ROOT}
      - AUTH_TOKEN=${AUTH_TOKEN}
      - DATA_ROOT=/output
    command: ["celery", "-A", "banzai.scheduling", "worker", "--hostname", "banzai-worker-local", "-l", "${BANZAI_WORKER_LOGLEVEL}", "-c", "4", "-Q", "${CELERY_TASK_QUEUE_NAME}"]
    restart: unless-stopped

  banzai-large-worker:
    build: .
    container_name: banzai-large-worker
    depends_on:
      - redis
      - rabbitmq
    volumes:
      - ${HOST_DATA_DIR}:/data
      - ${HOST_PROCESSED_DIR}:/output
    environment:
      - TASK_HOST=${TASK_HOST}
      - FITS_BROKER=${FITS_BROKER}
      - FITS_EXCHANGE=${FITS_EXCHANGE}
      - CELERY_LARGE_TASK_QUEUE_NAME=${CELERY_LARGE_TASK_QUEUE_NAME}
      - BANZAI_WORKER_LOGLEVEL=${BANZAI_WORKER_LOGLEVEL}
      - OMP_NUM_THREADS=${OMP_NUM_THREADS}
      - OPENTSDB_PYTHON_METRICS_TEST_MODE=${OPENTSDB_PYTHON_METRICS_TEST_MODE}
      - API_ROOT=${API_ROOT}
      - AUTH_TOKEN=${AUTH_TOKEN}
      - DATA_ROOT=/output
    command: ["celery", "-A", "banzai.scheduling", "worker", "--hostname", "banzai-large-worker-local", "-l", "${BANZAI_WORKER_LOGLEVEL}", "-c", "4", "-Q", "${CELERY_LARGE_TASK_QUEUE_NAME}"]
    restart: unless-stopped

  banzai-listener:
    build: .
    container_name: banzai-listener
    depends_on:
      - redis
      - rabbitmq
      - banzai-worker
    volumes:
      - ${HOST_DATA_DIR}:/data
      - ${HOST_PROCESSED_DIR}:/output
    environment:
      - DB_ADDRESS=${DB_ADDRESS}
      - CAL_DB_ADDRESS=${CAL_DB_ADDRESS:-${DB_ADDRESS}}
      - API_ROOT=${API_ROOT}
      - AUTH_TOKEN=${AUTH_TOKEN}
      - BANZAI_WORKER_LOGLEVEL=${BANZAI_WORKER_LOGLEVEL}
      - CELERY_LARGE_TASK_QUEUE_NAME=${CELERY_LARGE_TASK_QUEUE_NAME}
      - CELERY_TASK_QUEUE_NAME=${CELERY_TASK_QUEUE_NAME}
      - TASK_HOST=${TASK_HOST}
      - FITS_BROKER_URL=${FITS_BROKER_URL}
      - FITS_EXCHANGE=${FITS_EXCHANGE}
      - OPENTSDB_PYTHON_METRICS_TEST_MODE=${OPENTSDB_PYTHON_METRICS_TEST_MODE}
    entrypoint: ["banzai_run_realtime_pipeline", "--fpack", "--no-bpm",
                 "--broker-url=${FITS_BROKER_URL}",
                 "--db-address=${DB_ADDRESS}",
                 "--calibration-db-address=${CAL_DB_ADDRESS}",
                 "--processed-path=/output"]
    restart: unless-stopped