# 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.

services:
  data-hot1:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: data
    hostname: data-hot1
    command: data --node-discovery-mode=file --node-discovery-file-path=/etc/banyandb/nodes.yaml --node-labels type=hot
    volumes:
      - ./data-generator/tmp/metadata:/tmp/measure/data/sw_metadata
      - ./nodes.yaml:/etc/banyandb/nodes.yaml
    networks:
      - e2e

  data-warm1:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: data
    hostname: data-warm1
    command: data --node-discovery-mode=file --node-discovery-file-path=/etc/banyandb/nodes.yaml --node-labels type=warm
    volumes:
      - ./nodes.yaml:/etc/banyandb/nodes.yaml
    networks:
      - e2e

  data-cold1:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: data
    hostname: data-cold1
    command: data --node-discovery-mode=file --node-discovery-file-path=/etc/banyandb/nodes.yaml --node-labels type=cold
    volumes:
      - ./data-generator/tmp/measure:/tmp/measure
      - ./data-generator/tmp/stream:/tmp/stream
      - ./data-generator/tmp/property:/tmp/property
      - ./data-generator/tmp/trace:/tmp/trace
      - ./nodes.yaml:/etc/banyandb/nodes.yaml
    networks:
      - e2e

  liaison:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: liaison
    command: liaison --node-discovery-mode=file --node-discovery-file-path=/etc/banyandb/nodes.yaml --data-node-selector type=hot
    volumes:
      - ./nodes.yaml:/etc/banyandb/nodes.yaml
    networks:
      - e2e

  oap:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: oap
    environment:
      SW_STORAGE: banyandb
      SW_STORAGE_BANYANDB_TARGETS: "liaison:17912"
      SW_STORAGE_BANYANDB_METRICS_MINUTE_ENABLE_WARM_STAGE: "true"
      SW_STORAGE_BANYANDB_METRICS_MINUTE_ENABLE_COLD_STAGE: "true"
      SW_STORAGE_BANYANDB_METRICS_HOUR_ENABLE_WARM_STAGE: "true"
      SW_STORAGE_BANYANDB_METRICS_HOUR_ENABLE_COLD_STAGE: "true"
      SW_STORAGE_BANYANDB_METRICS_DAY_ENABLE_WARM_STAGE: "true"
      SW_STORAGE_BANYANDB_METRICS_DAY_ENABLE_COLD_STAGE: "true"
      SW_STORAGE_BANYANDB_RECORDS_ENABLE_WARM_STAGE: "true"
      SW_STORAGE_BANYANDB_RECORDS_ENABLE_COLD_STAGE: "true"
      SW_STORAGE_BANYANDB_TRACE_ENABLE_WARM_STAGE: "true"
      SW_STORAGE_BANYANDB_TRACE_ENABLE_COLD_STAGE: "true"
    ports:
      - 12800:12800
      - 11800:11800
    depends_on:
      liaison:
        condition: service_healthy

  agent:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: agent
    networks:
      - e2e

  provider:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: provider
    ports:
      - 9090
    networks:
      - e2e
    depends_on:
      oap:
        condition: service_healthy
      agent:
        condition: service_completed_successfully

  consumer:
    extends:
      file: ../../script/docker-compose/base-compose.yml
      service: consumer
    ports:
      - 9092
    depends_on:
      oap:
        condition: service_healthy
      provider:
        condition: service_healthy
      agent:
        condition: service_completed_successfully

networks:
  e2e:

volumes:
  sw_agent:
