增加docker内存限制相关配置

Author: hkfires

docker/.env.docker

@@ -6,8 +6,8 @@
 # =============================================================================
 
 # 主机上映射的端口 (外部访问端口)
-HOST_FASTAPI_PORT=8080
-HOST_STREAM_PORT=8081
+HOST_FASTAPI_PORT=2048
+HOST_STREAM_PORT=3120
 
 # =============================================================================
 # 容器内服务端口配置
@@ -118,6 +118,12 @@ USERSCRIPT_PATH=browser_utils/more_modles.js
 # Docker 特定配置
 # =============================================================================
 
+# 容器资源限制 (可选)
+# 默认不限制。如需限制容器资源，请在你的 .env 文件中取消注释并设置以下值。
+# 例如: DOCKER_MEMORY_LIMIT=4g
+# DOCKER_MEMORY_LIMIT=
+# DOCKER_MEMSWAP_LIMIT=
+
 # 容器重启策略相关
 # 这些配置项在 docker-compose.yml 中使用
 

docker/docker-compose.yml

@@ -4,6 +4,8 @@ services:
       context: ..
       dockerfile: docker/Dockerfile
     container_name: ai-studio-proxy-container
+    mem_limit: ${DOCKER_MEMORY_LIMIT}
+    memswap_limit: ${DOCKER_MEMSWAP_LIMIT}
     ports:
       - "${HOST_FASTAPI_PORT:-8080}:${DEFAULT_FASTAPI_PORT:-2048}"
       - "${HOST_STREAM_PORT:-8081}:${STREAM_PORT:-3120}"