SAG-Benchmark/docker-compose.yml at main · Zleap-AI/SAG-Benchmark · 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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
version: '3.8'

services:
  elasticsearch:
    image: docker.elastic.co/elasticsearch/elasticsearch:8.11.0
    container_name: new_sag_elasticsearch
    environment:
      - discovery.type=single-node
      - xpack.security.enabled=false
      # - "ES_JAVA_OPTS=-Xms512m -Xmx512m"
      - "ES_JAVA_OPTS=-Xms2g -Xmx2g"
    ports:
      - "${ES_PORT:-9200}:9200"
      - "9300:9300"
    volumes:
      - es_data:/usr/share/elasticsearch/data
    healthcheck:
      test:
        ["CMD-SHELL", "curl -f http://localhost:9200/_cluster/health || exit 1"]
      interval: 30s
      timeout: 10s
      retries: 5
    networks:
      - sag2_network

  mysql:
    image: mysql:8.0
    container_name: sag2_mysql
    environment:
      MYSQL_ROOT_PASSWORD: sag2_root
      MYSQL_DATABASE: sag2
      MYSQL_USER: sag2
      MYSQL_PASSWORD: sag2_pass
    ports:
      - "${MYSQL_PORT:-3306}:3306"
    volumes:
      - sag2_mysql_data:/var/lib/mysql
    command: >
      --default-authentication-plugin=mysql_native_password
      --skip-name-resolve=0
      --max-allowed-packet=64M
      --innodb-buffer-pool-size=512M
    healthcheck:
      test: ["CMD", "mysqladmin", "ping", "-h", "localhost"]
      interval: 10s
      timeout: 5s
      retries: 5
    networks:
      - sag2_network

  mlflow:
    image: ghcr.nju.edu.cn/mlflow/mlflow:latest
    container_name: sag2_mlflow
    environment:
      - MLFLOW_BACKEND_STORE_URI=sqlite:///mlflow/mlflow.db
      - MLFLOW_DEFAULT_ARTIFACT_ROOT=/mlflow/artifacts
    command: >
      mlflow server
      --backend-store-uri sqlite:///mlflow/mlflow.db
      --default-artifact-root /mlflow/artifacts
      --host 0.0.0.0
      --port 5000
    ports:
      - "${MLFLOW_PORT:-5000}:5000"
    volumes:
      - mlflow_data:/mlflow
    healthcheck:
      test: ["CMD-SHELL", "python3 -c \"import urllib.request; urllib.request.urlopen('http://localhost:5000/health')\" || exit 1"]
      interval: 30s
      timeout: 10s
      retries: 5
    networks:
      - sag2_network


  oceanbase:
    image: oceanbase/oceanbase-ce:4.4.2-lts
    container_name: oceanbase-ce
    hostname: oceanbase
    restart: unless-stopped
    ports:
      - "2881:2881"
      - "2882:2882"
    environment:
      MODE: normal
      OB_CLUSTER_NAME: obcluster
      OB_TENANT_NAME: sag2
      # OB_MEMORY_LIMIT: 6G
      # OB_DATAFILE_SIZE: 5G
      # FASTBOOT: "true"
    volumes:
      - ob_data:/root/ob
      - ./ob-start.sh:/root/boot/ob-start.sh:ro
      - ./init.sql:/root/boot/init.sql:ro
    entrypoint: ["bash", "/root/boot/ob-start.sh"]

volumes:
  es_data:
    driver: local
  sag2_mysql_data:
    driver: local
  mlflow_data:
    driver: local
  ob_data:
    driver: local

networks:
  sag2_network:
    driver: bridge