suntransit/docker-compose.yml at main · rishitoshsingh/suntransit · 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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
version: '3.8'

services:
  zookeeper:
    image: confluentinc/cp-zookeeper:7.5.0
    container_name: zookeeper
    environment:
      ZOOKEEPER_CLIENT_PORT: 2181
      ZOOKEEPER_TICK_TIME: 2000
    healthcheck:
      test: ["CMD", "bash", "-c", "echo ruok | nc localhost 2181 | grep imok"]
      interval: 5s
      timeout: 5s
      retries: 50
    networks:
      - suntransit-net

  kafka:
    image: confluentinc/cp-kafka:7.5.0
    container_name: kafka
    depends_on:
      - zookeeper
    ports:
      - "9092:9092"
    environment:
      KAFKA_BROKER_ID: 1
      KAFKA_ZOOKEEPER_CONNECT: zookeeper:2181
      KAFKA_ADVERTISED_LISTENERS: PLAINTEXT://kafka:9092
      KAFKA_LISTENER_SECURITY_PROTOCOL_MAP: PLAINTEXT:PLAINTEXT
      KAFKA_INTER_BROKER_LISTENER_NAME: PLAINTEXT
      KAFKA_OFFSETS_TOPIC_REPLICATION_FACTOR: 1
      KAFKA_LOG_RETENTION_MS: 7200000
    healthcheck:
      test: ["CMD", "bash", "-c", "echo > /dev/tcp/localhost/9092"]
      interval: 5s
      timeout: 5s
      retries: 50
    networks:
      - suntransit-net

  producer-valley-metro:
    build: ./producer
    container_name: producer-valley-metro
    depends_on:
      kafka:
        condition: service_healthy
    environment:
      KAFKA_BROKER: kafka:9092
      KAFKA_TOPIC: valley_metro_positions
      GTFS_AGENCY: valley-metro
      GTFS_URL: https://mna.mecatran.com/utw/ws/gtfsfeed/vehicles/valleymetro?apiKey=4f22263f69671d7f49726c3011333e527368211f
    networks:
      - suntransit-net

  producer-mbta:
    build: ./producer
    container_name: producer-mbta
    depends_on:
      kafka:
        condition: service_healthy
    environment:
      KAFKA_BROKER: kafka:9092
      KAFKA_TOPIC: mbta_positions
      GTFS_AGENCY: mbta-metro
      GTFS_URL: https://cdn.mbta.com/realtime/VehiclePositions.pb
    networks:
      - suntransit-net

  spark-master:
    build: ./spark-image
    container_name: spark-master
    depends_on:
      - kafka
      - redis
    volumes:
      - ./spark-jobs:/app
    environment:
      - SPARK_MODE=master
      - SPARK_MASTER_HOST=spark-master
      - USER=root
      - HADOOP_USER_NAME=spark
    ports:
      - "8080:8080"
      - "4040:4040"   # Spark Application UI (first app)
      - "4041:4041"   # Spark Application UI (second app)
      - "7078:7078"   # Spark driver port
    deploy:
      resources:
        limits:
          cpus: '1.0'
          memory: 1G
    networks:
      - suntransit-net


  spark-worker:
    build: ./spark-image
    depends_on:
      - spark-master
      - kafka
      - redis
    environment:
      - SPARK_MODE=worker
      - SPARK_MASTER_URL=spark://spark-master:7077
      - SPARK_WORKER_CORES=1
      - SPARK_WORKER_MEMORY=2G
      - USER=spark
      - HADOOP_USER_NAME=spark
    # ports:
    #   - "8081:8081"
    deploy:
      resources:
        limits:
          cpus: '1.0'
          memory: 2G
    networks:
      - suntransit-net

  redis:
    image: redis/redis-stack:latest
    container_name: suntransit-redis
    ports:
      - "6379:6379"   # Redis server port
      - "8001:8001"   # RedisInsight UI port
    environment:
      - REDIS_ARGS=--requirepass myStrongPassword123
      - REDISTIMESERIES_ARGS="RETENTION_POLICY=300000"  # 2 minutes retention in ms
    networks:
      - suntransit-net

  # dashboard:
  #   build: ./streamlit-dashboard
  #   container_name: suntransit-dashboard
  #   depends_on:
  #     - redis
  #   ports:
  #     - "80:8501"
  #   environment:
  #     - REDIS_HOST=redis
  #     - REDIS_PORT=6379
  #     - REDIS_DB=0
  #     - REDIS_PASSWORD=myStrongPassword123
  #   networks:
  #     - suntransit-net
  flask-web:
    build: ./flask_app
    ports:
      - "80:80"
    volumes:
      - ./flask_app:/app
    depends_on:
    - redis
    environment:
      - TIMEZONE=America/Phoenix
    networks:
      - suntransit-net
# kafka-ui:
#   image: provectuslabs/kafka-ui:latest
#   container_name: kafka-ui
#   ports:
#     - "8181:8080"
#   environment:
#     - KAFKA_CLUSTERS_0_NAME=local
#     - KAFKA_CLUSTERS_0_BOOTSTRAPSERVERS=kafka:9092
#     - KAFKA_CLUSTERS_0_ZOOKEEPER=zookeeper:2181
#   depends_on:
#     - kafka
#     - zookeeper
#   networks:
#     - suntransit-net

# spark-dev:
#   build:
#     context: .
#     dockerfile: Dockerfile.spark-dev
#   container_name: spark-dev
#   depends_on:
#     - kafka
#   volumes:
#     - ./notebooks:/app
#   ports:
#     - "8881:8888"  # JupyterLab
#     - "4141:4040"  # Spark UI
#   environment:
#     - SPARK_MODE=client
#   networks:
#     - suntransit-net
# python-dev:
#   build:
#     context: .
#     dockerfile: Dockerfile.python-dev
#   container_name: python-dev
#   volumes:
#     - ./notebooks:/app
#   ports:
#     - "8888:8888"  # JupyterLab
#   networks:
#     - suntransit-net
networks:
  suntransit-net:
    driver: bridge