This commit is contained in:
孙小云
parent
31142dff80
commit
151163031a
|
|
@ -381,3 +381,41 @@ services:
|
|||
- wvp-pro
|
||||
links:
|
||||
- wvp-pro
|
||||
pgvector-db:
|
||||
container_name: pgvector-db
|
||||
image: registry.t-aaron.com/pgvector/pgvector:pg16
|
||||
environment:
|
||||
POSTGRES_USER: drgraph
|
||||
POSTGRES_PASSWORD: yingping
|
||||
POSTGRES_DB: th_agenter
|
||||
TZ: Asia/Shanghai
|
||||
ports:
|
||||
- "5432:5432"
|
||||
volumes:
|
||||
- ./hyf_backend/pgdata:/var/lib/postgresql/data
|
||||
- ./hyf_backend/initdb:/docker-entrypoint-initdb.d
|
||||
restart: unless-stopped
|
||||
healthcheck:
|
||||
test: ["CMD-SHELL", "pg_isready -U drgraph -d th_agenter"]
|
||||
interval: 5s
|
||||
timeout: 5s
|
||||
retries: 5
|
||||
hyf-backend:
|
||||
container_name: hyf-backend
|
||||
image: hyf-backend-runtime
|
||||
build:
|
||||
context: ./hyf_backend
|
||||
dockerfile: dockerfile
|
||||
environment:
|
||||
- TZ=Asia/Shanghai
|
||||
- DATABASE_URL=postgresql+asyncpg://drgraph:yingping@pgvector-db:5432/th_agenter
|
||||
ports:
|
||||
- "8000:8000"
|
||||
volumes:
|
||||
- ./hyf_backend/data/uploads:/app/data/uploads
|
||||
- ./hyf_backend/data/chroma:/app/data/chroma
|
||||
- ./hyf_backend/logs:/app/webIOs/output/logs
|
||||
depends_on:
|
||||
pgvector-db:
|
||||
condition: service_healthy
|
||||
restart: unless-stopped
|
||||
|
|
|
|||
|
|
@ -0,0 +1,13 @@
|
|||
|
||||
FROM registry.t-aaron.com/hyf-backend-base:latest
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
# 复制项目代码（源码在 src 目录下）
|
||||
COPY src/ .
|
||||
|
||||
# 暴露端口
|
||||
EXPOSE 8000
|
||||
|
||||
# 启动命令：先执行数据库迁移，再启动 uvicorn
|
||||
CMD ["sh", "-c", "alembic upgrade head 2>/dev/null || true && uvicorn main:app --host 0.0.0.0 --port 8000"]
|
||||
|
|
@ -0,0 +1,5 @@
|
|||
-- 自动创建 pgvector 扩展
|
||||
CREATE EXTENSION IF NOT EXISTS vector;
|
||||
|
||||
-- 验证扩展已安装
|
||||
SELECT extname, extversion FROM pg_extension WHERE extname = 'vector';
|
||||
|
|
@ -0,0 +1,242 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="zh-CN">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>WebSocket 测试</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: Arial, sans-serif;
|
||||
max-width: 800px;
|
||||
margin: 50px auto;
|
||||
padding: 20px;
|
||||
}
|
||||
.container {
|
||||
border: 1px solid #ddd;
|
||||
border-radius: 5px;
|
||||
padding: 20px;
|
||||
}
|
||||
h1 {
|
||||
color: #333;
|
||||
}
|
||||
.status {
|
||||
padding: 10px;
|
||||
margin: 10px 0;
|
||||
border-radius: 3px;
|
||||
font-weight: bold;
|
||||
}
|
||||
.status.connected {
|
||||
background-color: #d4edda;
|
||||
color: #155724;
|
||||
}
|
||||
.status.disconnected {
|
||||
background-color: #f8d7da;
|
||||
color: #721c24;
|
||||
}
|
||||
.status.connecting {
|
||||
background-color: #fff3cd;
|
||||
color: #856404;
|
||||
}
|
||||
button {
|
||||
padding: 10px 20px;
|
||||
margin: 5px;
|
||||
border: none;
|
||||
border-radius: 3px;
|
||||
cursor: pointer;
|
||||
font-size: 14px;
|
||||
}
|
||||
.btn-connect {
|
||||
background-color: #28a745;
|
||||
color: white;
|
||||
}
|
||||
.btn-disconnect {
|
||||
background-color: #dc3545;
|
||||
color: white;
|
||||
}
|
||||
.btn-send {
|
||||
background-color: #007bff;
|
||||
color: white;
|
||||
}
|
||||
button:disabled {
|
||||
opacity: 0.5;
|
||||
cursor: not-allowed;
|
||||
}
|
||||
#messages {
|
||||
border: 1px solid #ddd;
|
||||
padding: 10px;
|
||||
height: 300px;
|
||||
overflow-y: auto;
|
||||
background-color: #f9f9f9;
|
||||
margin-top: 10px;
|
||||
}
|
||||
.message {
|
||||
padding: 5px;
|
||||
margin: 5px 0;
|
||||
border-left: 3px solid #007bff;
|
||||
background-color: white;
|
||||
}
|
||||
.message.received {
|
||||
border-left-color: #28a745;
|
||||
}
|
||||
.message.sent {
|
||||
border-left-color: #007bff;
|
||||
}
|
||||
.message.error {
|
||||
border-left-color: #dc3545;
|
||||
background-color: #fff5f5;
|
||||
}
|
||||
.message .time {
|
||||
color: #666;
|
||||
font-size: 12px;
|
||||
}
|
||||
input[type="text"] {
|
||||
width: 100%;
|
||||
padding: 8px;
|
||||
margin: 10px 0;
|
||||
border: 1px solid #ddd;
|
||||
border-radius: 3px;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="container">
|
||||
<h1>WebSocket 连接测试</h1>
|
||||
|
||||
<div>
|
||||
<label for="wsUrl">WebSocket 地址:</label>
|
||||
<input type="text" id="wsUrl" value="ws://220.154.136.137:9988/prod-api/websocket/statistics">
|
||||
</div>
|
||||
|
||||
<div id="status" class="status disconnected">状态: 未连接</div>
|
||||
|
||||
<div>
|
||||
<button class="btn-connect" onclick="connect()">连接</button>
|
||||
<button class="btn-disconnect" onclick="disconnect()" disabled>断开</button>
|
||||
<button class="btn-send" onclick="sendMessage()" disabled>发送测试消息</button>
|
||||
<button onclick="clearMessages()">清空消息</button>
|
||||
</div>
|
||||
|
||||
<div>
|
||||
<label for="messageInput">发送消息:</label>
|
||||
<input type="text" id="messageInput" placeholder="输入要发送的消息" disabled>
|
||||
</div>
|
||||
|
||||
<h3>消息记录:</h3>
|
||||
<div id="messages"></div>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
let ws = null;
|
||||
const statusDiv = document.getElementById('status');
|
||||
const messagesDiv = document.getElementById('messages');
|
||||
const wsUrlInput = document.getElementById('wsUrl');
|
||||
const messageInput = document.getElementById('messageInput');
|
||||
|
||||
function updateStatus(text, className) {
|
||||
statusDiv.textContent = '状态: ' + text;
|
||||
statusDiv.className = 'status ' + className;
|
||||
}
|
||||
|
||||
function addMessage(text, type = 'info') {
|
||||
const messageDiv = document.createElement('div');
|
||||
messageDiv.className = 'message ' + type;
|
||||
|
||||
const time = new Date().toLocaleTimeString();
|
||||
messageDiv.innerHTML = `
|
||||
<span class="time">[${time}]</span> ${text}
|
||||
`;
|
||||
|
||||
messagesDiv.appendChild(messageDiv);
|
||||
messagesDiv.scrollTop = messagesDiv.scrollHeight;
|
||||
}
|
||||
|
||||
function connect() {
|
||||
const url = wsUrlInput.value.trim();
|
||||
if (!url) {
|
||||
alert('请输入 WebSocket 地址');
|
||||
return;
|
||||
}
|
||||
|
||||
updateStatus('连接中...', 'connecting');
|
||||
addMessage('正在连接到: ' + url, 'info');
|
||||
|
||||
try {
|
||||
ws = new WebSocket(url);
|
||||
|
||||
ws.onopen = function(event) {
|
||||
updateStatus('已连接', 'connected');
|
||||
addMessage('✅ WebSocket 连接成功!', 'received');
|
||||
|
||||
document.querySelector('.btn-connect').disabled = true;
|
||||
document.querySelector('.btn-disconnect').disabled = false;
|
||||
document.querySelector('.btn-send').disabled = false;
|
||||
messageInput.disabled = false;
|
||||
};
|
||||
|
||||
ws.onmessage = function(event) {
|
||||
addMessage('📨 收到消息: ' + event.data, 'received');
|
||||
};
|
||||
|
||||
ws.onerror = function(error) {
|
||||
addMessage('❌ WebSocket 错误: ' + error, 'error');
|
||||
console.error('WebSocket error:', error);
|
||||
};
|
||||
|
||||
ws.onclose = function(event) {
|
||||
updateStatus('已断开', 'disconnected');
|
||||
addMessage('🔌 连接已关闭 (code: ' + event.code + ', reason: ' + event.reason + ')', 'error');
|
||||
|
||||
document.querySelector('.btn-connect').disabled = false;
|
||||
document.querySelector('.btn-disconnect').disabled = true;
|
||||
document.querySelector('.btn-send').disabled = true;
|
||||
messageInput.disabled = true;
|
||||
};
|
||||
|
||||
} catch (error) {
|
||||
updateStatus('连接失败', 'disconnected');
|
||||
addMessage('❌ 连接失败: ' + error.message, 'error');
|
||||
}
|
||||
}
|
||||
|
||||
function disconnect() {
|
||||
if (ws) {
|
||||
ws.close();
|
||||
ws = null;
|
||||
}
|
||||
}
|
||||
|
||||
function sendMessage() {
|
||||
const message = messageInput.value.trim();
|
||||
if (!message) {
|
||||
alert('请输入要发送的消息');
|
||||
return;
|
||||
}
|
||||
|
||||
if (ws && ws.readyState === WebSocket.OPEN) {
|
||||
ws.send(message);
|
||||
addMessage('📤 发送消息: ' + message, 'sent');
|
||||
messageInput.value = '';
|
||||
} else {
|
||||
alert('WebSocket 未连接');
|
||||
}
|
||||
}
|
||||
|
||||
function clearMessages() {
|
||||
messagesDiv.innerHTML = '';
|
||||
}
|
||||
|
||||
// 回车发送消息
|
||||
messageInput.addEventListener('keypress', function(event) {
|
||||
if (event.key === 'Enter') {
|
||||
sendMessage();
|
||||
}
|
||||
});
|
||||
|
||||
// 页面加载完成后的提示
|
||||
window.onload = function() {
|
||||
addMessage('👋 欢迎使用 WebSocket 测试工具', 'info');
|
||||
addMessage('📝 点击"连接"按钮开始测试', 'info');
|
||||
};
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
Loading…
Reference in New Issue