diff --git a/.env.development b/.env.development
index 52553ff..1be2541 100644
--- a/.env.development
+++ b/.env.development
@@ -6,6 +6,7 @@ VITE_APP_ENV = 'development'
 
 # 开发环境
 VITE_APP_BASE_API = '/dev-api'
+VITE_APP_WS_BASE_API = '/dev-api/resource/websocket'
 
 # 应用访问路径 例如使用前缀 /admin/
 VITE_APP_CONTEXT_PATH = '/'
diff --git a/.env.production b/.env.production
index bf9e644..6add191 100644
--- a/.env.production
+++ b/.env.production
@@ -15,6 +15,7 @@ VITE_APP_SNAILJOB_ADMIN = '/snail-job'
 
 # 生产环境
 VITE_APP_BASE_API = '/prod-api'
+VITE_APP_WS_BASE_API = '/prod-api/resource/websocket'
 
 # 是否在打包时开启压缩，支持 gzip 和 brotli
 VITE_BUILD_COMPRESS = gzip
diff --git a/src/layout/index.vue b/src/layout/index.vue
index 29fb5ff..715334b 100644
--- a/src/layout/index.vue
+++ b/src/layout/index.vue
@@ -68,7 +68,9 @@ onMounted(() => {
 
 onMounted(() => {
   let protocol = window.location.protocol === 'https:' ? 'wss://' : 'ws://';
-  initWebSocket(protocol + window.location.host + import.meta.env.VITE_APP_BASE_API + '/resource/websocket');
+  const base = import.meta.env.VITE_APP_WS_BASE_API;
+  const wsUrl = base.includes(`://`) ? base : `${protocol}${window.location.host}${base}`;
+  initWebSocket(wsUrl);
 });
 
 const handleClickOutside = () => {