docker-compose.yaml

---
version: '3'
services:
  zookeeper:
    image: confluentinc/cp-zookeeper:7.3.0
    hostname: zookeeper
    container_name: zookeeper_ntt
    environment:
      ZOOKEEPER_CLIENT_PORT: 2181
      ZOOKEEPER_TICK_TIME: 1000

  broker:
    image: confluentinc/cp-kafka:7.3.0
    container_name: broker_ntt
    # ports:
    #   - "9092:9092"
    hostname: broker
    depends_on:
      - zookeeper
    environment:
      KAFKA_BROKER_ID: 1
      KAFKA_ZOOKEEPER_CONNECT: 'zookeeper:2181'
      KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: PLAINTEXT:PLAINTEXT,PLAINTEXT_INTERNAL:PLAINTEXT
      KAFKA_ADVERTISED_LISTENERS: PLAINTEXT://localhost:9092,PLAINTEXT_INTERNAL://broker:29092
      KAFKA_LOG_RETENTION_MS: 2000
      KAFKA_LOG_RETENTION_CHECK_INTERVAL_MS: 2000
      KAFKA_LOG_CLEANER_DELETE_RETENTION_MS: 2000
      KAFKA_LOG_SEGMENT_DELETE_DELAY_MS: 2000
      KAFKA_MESSAGE_MAX_BYTES: 50000000
      KAFKA_MAX_MESSAGE_BYTES: 50000000
      KAFKA_REPLICA_FETCH_MAX_BYTES: 50000000
      KAFKA_OFFSETS_TOPIC_REPLICATION_FACTOR: 1
      KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: 1
      KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: 1
      KAFKA_AUTO_CREATE_TOPICS_ENABLE: 'true'

  save_vid_process:
    image: dnp_3:1.0.1
    container_name: save_vid_process
    stdin_open: true
    tty: true
    # replace this with .sh file
    command: > 
              bash -c "
              sleep 5 &&
              python3 create_topics.py &&
              python3 Save_video_process.py
              "
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    depends_on:
      - broker
    
  pose_process:
    image: dnp_3:1.0.1
    container_name: pose_process
    stdin_open: true
    tty: true
    # replace this with .sh file
    command: > 
              bash -c "
              python3 Pose_Process.py
              "
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    depends_on:
      - broker

  tracking_process:
    image: dnp_3:1.0.1
    container_name: tracking_process
    stdin_open: true
    tty: true
    # replace this with .sh file
    command: > 
              bash -c "
              python3 Tracking_process.py
              "
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    depends_on:
      - broker

  read_frame_process:
    image: dnp_3:1.0.1
    container_name: read_frame_process
    stdin_open: true
    tty: true
    # replace this with .sh file
    command: > 
              bash -c "
              python3 Read_frame_process.py
              "
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    depends_on:
      - broker