docker-compose.yml

version: "3.5"
services:
  mongo:
    container_name: mongo
    image: mongo:4.4
    command: --nojournal
    networks:
      - fiware
    ports:
      - "27017:27017"
    environment:
      - MONGO_INITDB_ROOT_USERNAME=root
      - MONGO_INITDB_ROOT_PASSWORD=example
    healthcheck:
      test: |
        host=`hostname --ip-address || echo '127.0.0.1'`; 
        mongo --quiet $host/test --eval 'quit(db.runCommand({ ping: 1 }).ok ? 0 : 2)' && echo 0 || echo 1

  orion:
    image: fiware/orion-ld:1.4.0
    container_name: orion
    depends_on:
      - mongo
    command: -dbhost mongo -dbuser root -dbpwd example #-logLevel DEBUG
    ports:
      - "1026:1026"
    environment:
      ORIONLD_MONGO_USER: root
      ORIONLD_MONGO_PASSWORD: example
    networks:
      - fiware
    healthcheck:
      test: curl --fail -s http://orion:1026/version || exit 1

  spark-master:
    build:
      context: ./docker
      target: master
    container_name: spark-master
    ports:
      - "8081:8080"
      - "7077:7077"
      - "9001:9001"
    environment:
      - INIT_DAEMON_STEP=setup_spark
      - "constraint:node==spark-master"
      - "HOST_CB=orion"
      - "SPARK_MASTER=spark://spark-master:7077"
    networks:
      - fiware
    volumes:
      - ./prediction-job:/prediction-job

  spark-worker:
    build:
      context: ./docker
      target: worker
    container_name: spark-worker
    depends_on:
      - spark-master
    ports:
      - "8082:8081"
    environment:
      - "SPARK_MASTER=spark://spark-master:7077"
      - "INIT_DAEMON_STEP=setup_spark"
      - "constraint:node==spark-worker"
      - "HOST_CB=orion"
      - "SERVER=proxy"
    networks:
      - fiware
    volumes:
      - ./prediction-job:/prediction-job

  mlflow-server:
    image: ghcr.io/mlflow/mlflow:v1.30.0
    container_name: mlflow_server
    ports:
      - "5001:5000"
    command: mlflow server --host 0.0.0.0 --backend-store-uri sqlite:///prediction-job/mlflow/mlflow.db --default-artifact-root ./prediction-job/mlflow/artifacts 
    volumes:
      - ./prediction-job:/prediction-job
    networks:
      - fiware

  initiator-entities:
    image: node:18
    container_name: initiator_entities
    depends_on:
      - orion
    networks:
      - fiware
    command: bash -c "sh /entities/awaitForOrion.sh && sh /entities/createPredictionEntities.sh && sh /entities/createSubscriptions.sh"
    volumes:
      - ./entities:/entities


networks:
  fiware:
    external: true
    name: fiware