Merge branch 'feature' of https://git.valuefrontier.cn/vf/vf_react into feature
This commit is contained in:
zdl
546
WEBSOCKET_INTEGRATION_GUIDE.md
Normal file
546
WEBSOCKET_INTEGRATION_GUIDE.md
Normal file
@@ -0,0 +1,546 @@
|
|||||||
|
# WebSocket 事件实时推送 - 前端集成指南
|
||||||
|
|
||||||
|
## 📦 已创建的文件
|
||||||
|
|
||||||
|
1. **`src/services/socketService.js`** - WebSocket 服务（已扩展）
|
||||||
|
2. **`src/hooks/useEventNotifications.js`** - React Hook
|
||||||
|
3. **`test_websocket.html`** - 测试页面
|
||||||
|
4. **`test_create_event.py`** - 测试脚本
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🚀 快速开始
|
||||||
|
|
||||||
|
### 方案 1：使用 React Hook（推荐）
|
||||||
|
|
||||||
|
在任何 React 组件中使用：
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { useEventNotifications } from 'hooks/useEventNotifications';
|
||||||
|
import { useToast } from '@chakra-ui/react';
|
||||||
|
|
||||||
|
function EventsPage() {
|
||||||
|
const toast = useToast();
|
||||||
|
|
||||||
|
// 订阅事件推送
|
||||||
|
const { newEvent, isConnected } = useEventNotifications({
|
||||||
|
eventType: 'all', // 'all' | 'policy' | 'market' | 'tech' | ...
|
||||||
|
importance: 'all', // 'all' | 'S' | 'A' | 'B' | 'C'
|
||||||
|
enabled: true, // 是否启用订阅
|
||||||
|
onNewEvent: (event) => {
|
||||||
|
// 收到新事件时的处理
|
||||||
|
console.log('🔔 收到新事件:', event);
|
||||||
|
|
||||||
|
// 显示 Toast 通知
|
||||||
|
toast({
|
||||||
|
title: '新事件提醒',
|
||||||
|
description: event.title,
|
||||||
|
status: 'info',
|
||||||
|
duration: 5000,
|
||||||
|
isClosable: true,
|
||||||
|
position: 'top-right',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Box>
|
||||||
|
<Text>连接状态: {isConnected ? '已连接 ✅' : '未连接 ❌'}</Text>
|
||||||
|
{/* 你的事件列表 */}
|
||||||
|
</Box>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 方案 2：在事件列表页面集成（完整示例）
|
||||||
|
|
||||||
|
**在 `src/views/Community/components/EventList.js` 中集成：**
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import React, { useState, useEffect } from 'react';
|
||||||
|
import { Box, Text, Badge, useToast } from '@chakra-ui/react';
|
||||||
|
import { useEventNotifications } from 'hooks/useEventNotifications';
|
||||||
|
|
||||||
|
function EventList() {
|
||||||
|
const [events, setEvents] = useState([]);
|
||||||
|
const [loading, setLoading] = useState(true);
|
||||||
|
const toast = useToast();
|
||||||
|
|
||||||
|
// 1️⃣ 初始加载事件列表（REST API）
|
||||||
|
useEffect(() => {
|
||||||
|
fetchEvents();
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
const fetchEvents = async () => {
|
||||||
|
try {
|
||||||
|
const response = await fetch('/api/events?per_page=20');
|
||||||
|
const data = await response.json();
|
||||||
|
|
||||||
|
if (data.success) {
|
||||||
|
setEvents(data.data.events);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
console.error('加载事件失败:', error);
|
||||||
|
} finally {
|
||||||
|
setLoading(false);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 2️⃣ 订阅 WebSocket 实时推送
|
||||||
|
const { newEvent, isConnected } = useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
importance: 'all',
|
||||||
|
enabled: true, // 可以根据用户设置控制是否启用
|
||||||
|
onNewEvent: (event) => {
|
||||||
|
console.log('🔔 收到新事件:', event);
|
||||||
|
|
||||||
|
// 显示通知
|
||||||
|
toast({
|
||||||
|
title: '📰 新事件发布',
|
||||||
|
description: `${event.title}`,
|
||||||
|
status: 'info',
|
||||||
|
duration: 6000,
|
||||||
|
isClosable: true,
|
||||||
|
position: 'top-right',
|
||||||
|
});
|
||||||
|
|
||||||
|
// 将新事件添加到列表顶部
|
||||||
|
setEvents((prevEvents) => {
|
||||||
|
// 检查是否已存在（防止重复）
|
||||||
|
const exists = prevEvents.some(e => e.id === event.id);
|
||||||
|
if (exists) {
|
||||||
|
return prevEvents;
|
||||||
|
}
|
||||||
|
// 添加到顶部，最多保留 100 个
|
||||||
|
return [event, ...prevEvents].slice(0, 100);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Box>
|
||||||
|
{/* 连接状态指示器 */}
|
||||||
|
<Box mb={4} display="flex" alignItems="center" gap={2}>
|
||||||
|
<Badge colorScheme={isConnected ? 'green' : 'red'}>
|
||||||
|
{isConnected ? '实时推送已开启' : '实时推送未连接'}
|
||||||
|
</Badge>
|
||||||
|
</Box>
|
||||||
|
|
||||||
|
{/* 事件列表 */}
|
||||||
|
{loading ? (
|
||||||
|
<Text>加载中...</Text>
|
||||||
|
) : (
|
||||||
|
<Box>
|
||||||
|
{events.map((event) => (
|
||||||
|
<EventCard key={event.id} event={event} />
|
||||||
|
))}
|
||||||
|
</Box>
|
||||||
|
)}
|
||||||
|
</Box>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
export default EventList;
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 方案 3：只订阅重要事件（S 和 A 级）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { useImportantEventNotifications } from 'hooks/useEventNotifications';
|
||||||
|
|
||||||
|
function Dashboard() {
|
||||||
|
const { importantEvents, isConnected } = useImportantEventNotifications((event) => {
|
||||||
|
// 只会收到 S 和 A 级别的重要事件
|
||||||
|
console.log('⚠️ 重要事件:', event);
|
||||||
|
|
||||||
|
// 播放提示音
|
||||||
|
new Audio('/notification.mp3').play();
|
||||||
|
});
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Box>
|
||||||
|
<Heading>重要事件通知</Heading>
|
||||||
|
{importantEvents.map(event => (
|
||||||
|
<Alert key={event.id} status="warning">
|
||||||
|
<AlertIcon />
|
||||||
|
{event.title}
|
||||||
|
</Alert>
|
||||||
|
))}
|
||||||
|
</Box>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 方案 4：直接使用 Service（不用 Hook）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { useEffect } from 'react';
|
||||||
|
import socketService from 'services/socketService';
|
||||||
|
|
||||||
|
function MyComponent() {
|
||||||
|
useEffect(() => {
|
||||||
|
// 连接
|
||||||
|
socketService.connect();
|
||||||
|
|
||||||
|
// 订阅
|
||||||
|
const unsubscribe = socketService.subscribeToAllEvents((event) => {
|
||||||
|
console.log('新事件:', event);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 清理
|
||||||
|
return () => {
|
||||||
|
unsubscribe();
|
||||||
|
socketService.disconnect();
|
||||||
|
};
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
return <div>...</div>;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🎨 UI 集成示例
|
||||||
|
|
||||||
|
### 1. Toast 通知（Chakra UI）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { useToast } from '@chakra-ui/react';
|
||||||
|
|
||||||
|
const toast = useToast();
|
||||||
|
|
||||||
|
// 在 onNewEvent 回调中
|
||||||
|
onNewEvent: (event) => {
|
||||||
|
toast({
|
||||||
|
title: '新事件',
|
||||||
|
description: event.title,
|
||||||
|
status: 'info',
|
||||||
|
duration: 5000,
|
||||||
|
isClosable: true,
|
||||||
|
position: 'top-right',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 2. 顶部通知栏
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { Alert, AlertIcon, CloseButton } from '@chakra-ui/react';
|
||||||
|
|
||||||
|
function EventNotificationBanner() {
|
||||||
|
const [showNotification, setShowNotification] = useState(false);
|
||||||
|
const [latestEvent, setLatestEvent] = useState(null);
|
||||||
|
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
onNewEvent: (event) => {
|
||||||
|
setLatestEvent(event);
|
||||||
|
setShowNotification(true);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!showNotification || !latestEvent) return null;
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Alert status="info" variant="solid">
|
||||||
|
<AlertIcon />
|
||||||
|
新事件：{latestEvent.title}
|
||||||
|
<CloseButton
|
||||||
|
position="absolute"
|
||||||
|
right="8px"
|
||||||
|
top="8px"
|
||||||
|
onClick={() => setShowNotification(false)}
|
||||||
|
/>
|
||||||
|
</Alert>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 3. 角标提示（红点）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { Badge } from '@chakra-ui/react';
|
||||||
|
|
||||||
|
function EventsMenuItem() {
|
||||||
|
const [unreadCount, setUnreadCount] = useState(0);
|
||||||
|
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
onNewEvent: () => {
|
||||||
|
setUnreadCount(prev => prev + 1);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return (
|
||||||
|
<MenuItem position="relative">
|
||||||
|
事件中心
|
||||||
|
{unreadCount > 0 && (
|
||||||
|
<Badge
|
||||||
|
colorScheme="red"
|
||||||
|
position="absolute"
|
||||||
|
top="-5px"
|
||||||
|
right="-5px"
|
||||||
|
borderRadius="full"
|
||||||
|
>
|
||||||
|
{unreadCount > 99 ? '99+' : unreadCount}
|
||||||
|
</Badge>
|
||||||
|
)}
|
||||||
|
</MenuItem>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 4. 浮动通知卡片
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { Box, Slide, useDisclosure } from '@chakra-ui/react';
|
||||||
|
|
||||||
|
function FloatingEventNotification() {
|
||||||
|
const { isOpen, onClose, onOpen } = useDisclosure();
|
||||||
|
const [event, setEvent] = useState(null);
|
||||||
|
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
onNewEvent: (newEvent) => {
|
||||||
|
setEvent(newEvent);
|
||||||
|
onOpen();
|
||||||
|
|
||||||
|
// 5秒后自动关闭
|
||||||
|
setTimeout(onClose, 5000);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Slide direction="bottom" in={isOpen} style={{ zIndex: 10 }}>
|
||||||
|
<Box
|
||||||
|
p="40px"
|
||||||
|
color="white"
|
||||||
|
bg="blue.500"
|
||||||
|
rounded="md"
|
||||||
|
shadow="md"
|
||||||
|
m={4}
|
||||||
|
>
|
||||||
|
<Text fontWeight="bold">{event?.title}</Text>
|
||||||
|
<Text fontSize="sm">{event?.description}</Text>
|
||||||
|
<Button size="sm" mt={2} onClick={onClose}>
|
||||||
|
关闭
|
||||||
|
</Button>
|
||||||
|
</Box>
|
||||||
|
</Slide>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📋 API 参考
|
||||||
|
|
||||||
|
### `useEventNotifications(options)`
|
||||||
|
|
||||||
|
**参数：**
|
||||||
|
| 参数 | 类型 | 默认值 | 说明 |
|
||||||
|
|------|------|--------|------|
|
||||||
|
| `eventType` | string | `'all'` | 事件类型：`'all'` / `'policy'` / `'market'` / `'tech'` 等 |
|
||||||
|
| `importance` | string | `'all'` | 重要性：`'all'` / `'S'` / `'A'` / `'B'` / `'C'` |
|
||||||
|
| `enabled` | boolean | `true` | 是否启用订阅 |
|
||||||
|
| `onNewEvent` | function | - | 收到新事件时的回调函数 |
|
||||||
|
|
||||||
|
**返回值：**
|
||||||
|
| 属性 | 类型 | 说明 |
|
||||||
|
|------|------|------|
|
||||||
|
| `newEvent` | object | 最新收到的事件对象 |
|
||||||
|
| `isConnected` | boolean | WebSocket 连接状态 |
|
||||||
|
| `error` | object | 错误信息 |
|
||||||
|
| `clearNewEvent` | function | 清除新事件状态 |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### `socketService` API
|
||||||
|
|
||||||
|
```javascript
|
||||||
|
// 连接
|
||||||
|
socketService.connect(options)
|
||||||
|
|
||||||
|
// 断开
|
||||||
|
socketService.disconnect()
|
||||||
|
|
||||||
|
// 订阅所有事件
|
||||||
|
socketService.subscribeToAllEvents(callback)
|
||||||
|
|
||||||
|
// 订阅特定类型
|
||||||
|
socketService.subscribeToEventType('tech', callback)
|
||||||
|
|
||||||
|
// 订阅特定重要性
|
||||||
|
socketService.subscribeToImportantEvents('S', callback)
|
||||||
|
|
||||||
|
// 取消订阅
|
||||||
|
socketService.unsubscribeFromEvents({ eventType: 'all' })
|
||||||
|
|
||||||
|
// 检查连接状态
|
||||||
|
socketService.isConnected()
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🔧 事件数据结构
|
||||||
|
|
||||||
|
收到的 `event` 对象包含：
|
||||||
|
|
||||||
|
```javascript
|
||||||
|
{
|
||||||
|
id: 123,
|
||||||
|
title: "事件标题",
|
||||||
|
description: "事件描述",
|
||||||
|
event_type: "tech", // 类型
|
||||||
|
importance: "S", // 重要性
|
||||||
|
status: "active",
|
||||||
|
created_at: "2025-01-21T14:30:00",
|
||||||
|
hot_score: 85.5,
|
||||||
|
view_count: 1234,
|
||||||
|
related_avg_chg: 5.2, // 平均涨幅
|
||||||
|
related_max_chg: 15.8, // 最大涨幅
|
||||||
|
keywords: ["AI", "芯片"], // 关键词
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ⚙️ 高级配置
|
||||||
|
|
||||||
|
### 1. 条件订阅（用户设置）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
function EventsPage() {
|
||||||
|
const [enableNotifications, setEnableNotifications] = useState(
|
||||||
|
localStorage.getItem('enableEventNotifications') === 'true'
|
||||||
|
);
|
||||||
|
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
enabled: enableNotifications, // 根据用户设置控制
|
||||||
|
onNewEvent: handleNewEvent
|
||||||
|
});
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Switch
|
||||||
|
isChecked={enableNotifications}
|
||||||
|
onChange={(e) => {
|
||||||
|
const enabled = e.target.checked;
|
||||||
|
setEnableNotifications(enabled);
|
||||||
|
localStorage.setItem('enableEventNotifications', enabled);
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
启用事件实时通知
|
||||||
|
</Switch>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 2. 多个订阅（不同类型）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
function MultiSubscriptionExample() {
|
||||||
|
// 订阅科技类事件
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'tech',
|
||||||
|
onNewEvent: (event) => console.log('科技事件:', event)
|
||||||
|
});
|
||||||
|
|
||||||
|
// 订阅政策类事件
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'policy',
|
||||||
|
onNewEvent: (event) => console.log('政策事件:', event)
|
||||||
|
});
|
||||||
|
|
||||||
|
return <div>...</div>;
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
### 3. 防抖处理（避免通知过多）
|
||||||
|
|
||||||
|
```jsx
|
||||||
|
import { debounce } from 'lodash';
|
||||||
|
|
||||||
|
const debouncedNotify = debounce((event) => {
|
||||||
|
toast({
|
||||||
|
title: '新事件',
|
||||||
|
description: event.title,
|
||||||
|
});
|
||||||
|
}, 1000);
|
||||||
|
|
||||||
|
useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
onNewEvent: debouncedNotify
|
||||||
|
});
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🧪 测试步骤
|
||||||
|
|
||||||
|
1. **启动 Flask 服务**
|
||||||
|
```bash
|
||||||
|
python app.py
|
||||||
|
```
|
||||||
|
|
||||||
|
2. **启动 React 应用**
|
||||||
|
```bash
|
||||||
|
npm start
|
||||||
|
```
|
||||||
|
|
||||||
|
3. **创建测试事件**
|
||||||
|
```bash
|
||||||
|
python test_create_event.py
|
||||||
|
```
|
||||||
|
|
||||||
|
4. **观察结果**
|
||||||
|
- 最多等待 30 秒
|
||||||
|
- 前端页面应该显示通知
|
||||||
|
- 控制台输出日志
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🐛 常见问题
|
||||||
|
|
||||||
|
### Q: 没有收到推送？
|
||||||
|
**A:** 检查：
|
||||||
|
1. Flask 服务是否启动
|
||||||
|
2. 浏览器控制台是否有连接错误
|
||||||
|
3. 后端日志是否显示 `[轮询] 发现 X 个新事件`
|
||||||
|
|
||||||
|
### Q: 连接一直失败？
|
||||||
|
**A:** 检查：
|
||||||
|
1. API_BASE_URL 配置是否正确
|
||||||
|
2. CORS 配置是否包含前端域名
|
||||||
|
3. 防火墙/代理设置
|
||||||
|
|
||||||
|
### Q: 收到重复通知？
|
||||||
|
**A:** 检查是否多次调用了 Hook，确保只在需要的地方订阅一次。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📚 更多资源
|
||||||
|
|
||||||
|
- Socket.IO 文档: https://socket.io/docs/v4/
|
||||||
|
- Chakra UI Toast: https://chakra-ui.com/docs/components/toast
|
||||||
|
- React Hooks: https://react.dev/reference/react
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
**完成！🎉** 现在你的前端可以实时接收事件推送了！
|
||||||
257
app.py
257
app.py
@@ -8,6 +8,7 @@ import uuid
|
|||||||
from functools import wraps
|
from functools import wraps
|
||||||
import qrcode
|
import qrcode
|
||||||
from flask_mail import Mail, Message
|
from flask_mail import Mail, Message
|
||||||
|
from flask_socketio import SocketIO, emit, join_room, leave_room
|
||||||
import pytz
|
import pytz
|
||||||
import requests
|
import requests
|
||||||
from celery import Celery
|
from celery import Celery
|
||||||
@@ -40,6 +41,7 @@ from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentClo
|
|||||||
from sqlalchemy import text, desc, and_
|
from sqlalchemy import text, desc, and_
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
from decimal import Decimal
|
from decimal import Decimal
|
||||||
|
from apscheduler.schedulers.background import BackgroundScheduler
|
||||||
|
|
||||||
# 交易日数据缓存
|
# 交易日数据缓存
|
||||||
trading_days = []
|
trading_days = []
|
||||||
@@ -242,6 +244,18 @@ db = SQLAlchemy(app)
|
|||||||
# 初始化邮件服务
|
# 初始化邮件服务
|
||||||
mail = Mail(app)
|
mail = Mail(app)
|
||||||
|
|
||||||
|
# 初始化 Flask-SocketIO（用于实时事件推送）
|
||||||
|
socketio = SocketIO(
|
||||||
|
app,
|
||||||
|
cors_allowed_origins=["http://localhost:3000", "http://127.0.0.1:3000", "http://localhost:5173",
|
||||||
|
"https://valuefrontier.cn", "http://valuefrontier.cn"],
|
||||||
|
async_mode='gevent',
|
||||||
|
logger=True,
|
||||||
|
engineio_logger=False,
|
||||||
|
ping_timeout=120, # 心跳超时时间（秒），客户端120秒内无响应才断开
|
||||||
|
ping_interval=25 # 心跳检测间隔（秒），每25秒发送一次ping
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@login_manager.user_loader
|
@login_manager.user_loader
|
||||||
def load_user(user_id):
|
def load_user(user_id):
|
||||||
@@ -7443,6 +7457,243 @@ def add_event_comment(event_id):
|
|||||||
}), 500
|
}), 500
|
||||||
|
|
||||||
|
|
||||||
|
# ==================== WebSocket 事件处理器（实时事件推送） ====================
|
||||||
|
|
||||||
|
@socketio.on('connect')
|
||||||
|
def handle_connect():
|
||||||
|
"""客户端连接事件"""
|
||||||
|
print(f'[WebSocket] 客户端已连接: {request.sid}')
|
||||||
|
emit('connection_response', {
|
||||||
|
'status': 'connected',
|
||||||
|
'sid': request.sid,
|
||||||
|
'message': '已连接到事件推送服务'
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@socketio.on('subscribe_events')
|
||||||
|
def handle_subscribe(data):
|
||||||
|
"""
|
||||||
|
客户端订阅事件推送
|
||||||
|
data: {
|
||||||
|
'event_type': 'all' | 'policy' | 'market' | 'tech' | ...,
|
||||||
|
'importance': 'all' | 'S' | 'A' | 'B' | 'C',
|
||||||
|
'filters': {...} # 可选的其他筛选条件
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
event_type = data.get('event_type', 'all')
|
||||||
|
importance = data.get('importance', 'all')
|
||||||
|
|
||||||
|
# 加入对应的房间
|
||||||
|
room_name = f"events_{event_type}"
|
||||||
|
join_room(room_name)
|
||||||
|
|
||||||
|
print(f'[WebSocket] 客户端 {request.sid} 订阅了房间: {room_name}')
|
||||||
|
|
||||||
|
emit('subscription_confirmed', {
|
||||||
|
'success': True,
|
||||||
|
'room': room_name,
|
||||||
|
'event_type': event_type,
|
||||||
|
'importance': importance,
|
||||||
|
'message': f'已订阅 {event_type} 类型的事件推送'
|
||||||
|
})
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f'[WebSocket] 订阅失败: {e}')
|
||||||
|
emit('subscription_error', {
|
||||||
|
'success': False,
|
||||||
|
'error': str(e)
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@socketio.on('unsubscribe_events')
|
||||||
|
def handle_unsubscribe(data):
|
||||||
|
"""取消订阅事件推送"""
|
||||||
|
try:
|
||||||
|
event_type = data.get('event_type', 'all')
|
||||||
|
room_name = f"events_{event_type}"
|
||||||
|
leave_room(room_name)
|
||||||
|
|
||||||
|
print(f'[WebSocket] 客户端 {request.sid} 取消订阅房间: {room_name}')
|
||||||
|
|
||||||
|
emit('unsubscription_confirmed', {
|
||||||
|
'success': True,
|
||||||
|
'room': room_name,
|
||||||
|
'message': f'已取消订阅 {event_type} 类型的事件推送'
|
||||||
|
})
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f'[WebSocket] 取消订阅失败: {e}')
|
||||||
|
emit('unsubscription_error', {
|
||||||
|
'success': False,
|
||||||
|
'error': str(e)
|
||||||
|
})
|
||||||
|
|
||||||
|
|
||||||
|
@socketio.on('disconnect')
|
||||||
|
def handle_disconnect():
|
||||||
|
"""客户端断开连接事件"""
|
||||||
|
print(f'[WebSocket] 客户端已断开: {request.sid}')
|
||||||
|
|
||||||
|
|
||||||
|
# ==================== WebSocket 辅助函数 ====================
|
||||||
|
|
||||||
|
def broadcast_new_event(event):
|
||||||
|
"""
|
||||||
|
广播新事件到所有订阅的客户端
|
||||||
|
在创建新事件时调用此函数
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event: Event 模型实例
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
event_data = {
|
||||||
|
'id': event.id,
|
||||||
|
'title': event.title,
|
||||||
|
'description': event.description,
|
||||||
|
'event_type': event.event_type,
|
||||||
|
'importance': event.importance,
|
||||||
|
'status': event.status,
|
||||||
|
'created_at': event.created_at.isoformat() if event.created_at else None,
|
||||||
|
'hot_score': event.hot_score,
|
||||||
|
'view_count': event.view_count,
|
||||||
|
'related_avg_chg': event.related_avg_chg,
|
||||||
|
'related_max_chg': event.related_max_chg,
|
||||||
|
'keywords': event.keywords_list if hasattr(event, 'keywords_list') else event.keywords,
|
||||||
|
}
|
||||||
|
|
||||||
|
# 发送到所有订阅者（all 房间）
|
||||||
|
socketio.emit('new_event', event_data, room='events_all', namespace='/')
|
||||||
|
|
||||||
|
# 发送到特定类型订阅者
|
||||||
|
if event.event_type:
|
||||||
|
room_name = f"events_{event.event_type}"
|
||||||
|
socketio.emit('new_event', event_data, room=room_name, namespace='/')
|
||||||
|
print(f'[WebSocket] 已推送新事件到房间: events_all, {room_name}')
|
||||||
|
else:
|
||||||
|
print(f'[WebSocket] 已推送新事件到房间: events_all')
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f'[WebSocket] 推送新事件失败: {e}')
|
||||||
|
|
||||||
|
|
||||||
|
# ==================== WebSocket 轮询机制（检测新事件） ====================
|
||||||
|
|
||||||
|
# 内存变量：记录上次检查的时间戳和已推送的事件 ID 集合
|
||||||
|
last_checked_time = None
|
||||||
|
pushed_event_ids = set() # 已推送的事件 ID 集合，防止重复推送
|
||||||
|
MAX_PUSHED_IDS_SIZE = 1000 # 已推送 ID 集合的最大容量
|
||||||
|
|
||||||
|
def poll_new_events():
|
||||||
|
"""
|
||||||
|
定期轮询数据库，检查是否有新事件
|
||||||
|
每 30 秒执行一次
|
||||||
|
|
||||||
|
设计思路：
|
||||||
|
1. 使用时间戳查询（created_at），而不是 ID
|
||||||
|
2. 维护已推送事件 ID 集合，避免重复推送
|
||||||
|
3. 使用重叠时间窗口（向前多查60秒），捕获延迟写入的事件
|
||||||
|
4. 定期清理已推送集合，防止内存泄漏
|
||||||
|
"""
|
||||||
|
global last_checked_time, pushed_event_ids
|
||||||
|
|
||||||
|
try:
|
||||||
|
with app.app_context():
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
current_time = datetime.now()
|
||||||
|
|
||||||
|
# 如果是第一次运行，只查询最近 30 秒的事件
|
||||||
|
if last_checked_time is None:
|
||||||
|
query_start_time = current_time - timedelta(seconds=30)
|
||||||
|
else:
|
||||||
|
# 向前多查 60 秒（重叠窗口），防止漏掉延迟写入的事件
|
||||||
|
query_start_time = last_checked_time - timedelta(seconds=60)
|
||||||
|
|
||||||
|
# 查询时间范围内的新事件
|
||||||
|
new_events = Event.query.filter(
|
||||||
|
Event.created_at >= query_start_time,
|
||||||
|
Event.created_at <= current_time,
|
||||||
|
Event.status == 'active'
|
||||||
|
).order_by(Event.created_at.asc()).all()
|
||||||
|
|
||||||
|
# 过滤掉已经推送过的事件
|
||||||
|
unpushed_events = [
|
||||||
|
event for event in new_events
|
||||||
|
if event.id not in pushed_event_ids
|
||||||
|
]
|
||||||
|
|
||||||
|
if unpushed_events:
|
||||||
|
print(f'[轮询] 发现 {len(unpushed_events)} 个新事件（查询到 {len(new_events)} 个，已过滤 {len(new_events) - len(unpushed_events)} 个重复）')
|
||||||
|
|
||||||
|
for event in unpushed_events:
|
||||||
|
# 推送新事件
|
||||||
|
broadcast_new_event(event)
|
||||||
|
# 记录已推送
|
||||||
|
pushed_event_ids.add(event.id)
|
||||||
|
print(f'[轮询] 推送事件 ID={event.id}, 标题={event.title}')
|
||||||
|
|
||||||
|
# 更新检查时间
|
||||||
|
last_checked_time = current_time
|
||||||
|
|
||||||
|
# 清理已推送集合（防止无限增长）
|
||||||
|
if len(pushed_event_ids) > MAX_PUSHED_IDS_SIZE:
|
||||||
|
# 只保留最新的一半
|
||||||
|
sorted_ids = sorted(pushed_event_ids)
|
||||||
|
pushed_event_ids = set(sorted_ids[-MAX_PUSHED_IDS_SIZE//2:])
|
||||||
|
print(f'[轮询] 已清理推送记录，当前保留 {len(pushed_event_ids)} 个')
|
||||||
|
|
||||||
|
else:
|
||||||
|
# 没有新事件，也要更新检查时间
|
||||||
|
last_checked_time = current_time
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f'[轮询] 检查新事件时出错: {e}')
|
||||||
|
|
||||||
|
|
||||||
|
def initialize_event_polling():
|
||||||
|
"""
|
||||||
|
初始化事件轮询机制
|
||||||
|
在应用启动时调用
|
||||||
|
"""
|
||||||
|
global last_checked_time, pushed_event_ids
|
||||||
|
|
||||||
|
try:
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
with app.app_context():
|
||||||
|
# 设置初始检查时间为当前时间
|
||||||
|
# 这样启动后只会推送新创建的事件，不会推送历史事件
|
||||||
|
last_checked_time = datetime.now()
|
||||||
|
pushed_event_ids.clear()
|
||||||
|
|
||||||
|
# 统计数据库中的事件总数
|
||||||
|
total_events = Event.query.filter_by(status='active').count()
|
||||||
|
print(f'[轮询] 初始化完成，数据库中共有 {total_events} 个活跃事件')
|
||||||
|
print(f'[轮询] 起始时间: {last_checked_time.strftime("%Y-%m-%d %H:%M:%S")}')
|
||||||
|
print(f'[轮询] 只会推送此时间之后创建的新事件')
|
||||||
|
|
||||||
|
# 创建后台调度器
|
||||||
|
scheduler = BackgroundScheduler()
|
||||||
|
# 每 30 秒执行一次轮询
|
||||||
|
scheduler.add_job(
|
||||||
|
func=poll_new_events,
|
||||||
|
trigger='interval',
|
||||||
|
seconds=30,
|
||||||
|
id='poll_new_events',
|
||||||
|
name='检查新事件并推送',
|
||||||
|
replace_existing=True
|
||||||
|
)
|
||||||
|
scheduler.start()
|
||||||
|
print('[轮询] 调度器已启动，每 30 秒检查一次新事件')
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f'[轮询] 初始化失败: {e}')
|
||||||
|
|
||||||
|
|
||||||
|
# ==================== 结束 WebSocket 部分 ====================
|
||||||
|
|
||||||
|
|
||||||
@app.route('/api/posts/<int:post_id>/like', methods=['POST'])
|
@app.route('/api/posts/<int:post_id>/like', methods=['POST'])
|
||||||
@login_required
|
@login_required
|
||||||
def like_post(post_id):
|
def like_post(post_id):
|
||||||
@@ -11581,4 +11832,8 @@ if __name__ == '__main__':
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
app.logger.error(f"数据库初始化失败: {e}")
|
app.logger.error(f"数据库初始化失败: {e}")
|
||||||
|
|
||||||
app.run(host='0.0.0.0', port=5001, debug=False)
|
# 初始化事件轮询机制（WebSocket 推送）
|
||||||
|
initialize_event_polling()
|
||||||
|
|
||||||
|
# 使用 socketio.run 替代 app.run 以支持 WebSocket
|
||||||
|
socketio.run(app, host='0.0.0.0', port=5001, debug=False, allow_unsafe_werkzeug=True)
|
||||||
161
src/hooks/useEventNotifications.js
Normal file
161
src/hooks/useEventNotifications.js
Normal file
@@ -0,0 +1,161 @@
|
|||||||
|
// src/hooks/useEventNotifications.js
|
||||||
|
/**
|
||||||
|
* React Hook：用于在组件中订阅事件推送通知
|
||||||
|
*
|
||||||
|
* 使用示例：
|
||||||
|
* ```jsx
|
||||||
|
* import { useEventNotifications } from 'hooks/useEventNotifications';
|
||||||
|
*
|
||||||
|
* function MyComponent() {
|
||||||
|
* const { newEvent, isConnected } = useEventNotifications({
|
||||||
|
* eventType: 'all',
|
||||||
|
* importance: 'all',
|
||||||
|
* onNewEvent: (event) => {
|
||||||
|
* console.log('收到新事件:', event);
|
||||||
|
* // 显示通知...
|
||||||
|
* }
|
||||||
|
* });
|
||||||
|
*
|
||||||
|
* return <div>...</div>;
|
||||||
|
* }
|
||||||
|
* ```
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { useEffect, useState, useRef } from 'react';
|
||||||
|
import { socketService } from '../services/socketService';
|
||||||
|
|
||||||
|
export const useEventNotifications = (options = {}) => {
|
||||||
|
const {
|
||||||
|
eventType = 'all',
|
||||||
|
importance = 'all',
|
||||||
|
enabled = true,
|
||||||
|
onNewEvent,
|
||||||
|
} = options;
|
||||||
|
|
||||||
|
const [isConnected, setIsConnected] = useState(false);
|
||||||
|
const [newEvent, setNewEvent] = useState(null);
|
||||||
|
const [error, setError] = useState(null);
|
||||||
|
const unsubscribeRef = useRef(null);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
// 如果禁用，则不订阅
|
||||||
|
if (!enabled) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 连接状态监听
|
||||||
|
const handleConnect = () => {
|
||||||
|
setIsConnected(true);
|
||||||
|
setError(null);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleDisconnect = () => {
|
||||||
|
setIsConnected(false);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleConnectError = (err) => {
|
||||||
|
setError(err);
|
||||||
|
setIsConnected(false);
|
||||||
|
};
|
||||||
|
|
||||||
|
// 连接 WebSocket
|
||||||
|
socketService.connect();
|
||||||
|
|
||||||
|
// 监听连接事件
|
||||||
|
socketService.on('connect', handleConnect);
|
||||||
|
socketService.on('disconnect', handleDisconnect);
|
||||||
|
socketService.on('connect_error', handleConnectError);
|
||||||
|
|
||||||
|
// 新事件处理函数
|
||||||
|
const handleNewEvent = (eventData) => {
|
||||||
|
setNewEvent(eventData);
|
||||||
|
|
||||||
|
// 调用外部回调
|
||||||
|
if (onNewEvent) {
|
||||||
|
onNewEvent(eventData);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// 订阅事件推送
|
||||||
|
socketService.subscribeToEvents({
|
||||||
|
eventType,
|
||||||
|
importance,
|
||||||
|
onNewEvent: handleNewEvent,
|
||||||
|
onSubscribed: (data) => {
|
||||||
|
console.log('订阅成功:', data);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
// 保存取消订阅函数
|
||||||
|
unsubscribeRef.current = () => {
|
||||||
|
socketService.unsubscribeFromEvents({ eventType });
|
||||||
|
};
|
||||||
|
|
||||||
|
// 组件卸载时清理
|
||||||
|
return () => {
|
||||||
|
console.log('清理 WebSocket 订阅');
|
||||||
|
|
||||||
|
// 取消订阅
|
||||||
|
if (unsubscribeRef.current) {
|
||||||
|
unsubscribeRef.current();
|
||||||
|
}
|
||||||
|
|
||||||
|
// 移除监听器
|
||||||
|
socketService.off('connect', handleConnect);
|
||||||
|
socketService.off('disconnect', handleDisconnect);
|
||||||
|
socketService.off('connect_error', handleConnectError);
|
||||||
|
|
||||||
|
// 断开连接
|
||||||
|
socketService.disconnect();
|
||||||
|
};
|
||||||
|
}, [eventType, importance, enabled, onNewEvent]);
|
||||||
|
|
||||||
|
return {
|
||||||
|
newEvent, // 最新收到的事件
|
||||||
|
isConnected, // WebSocket 连接状态
|
||||||
|
error, // 错误信息
|
||||||
|
clearNewEvent: () => setNewEvent(null), // 清除新事件状态
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 简化版 Hook：只订阅所有事件
|
||||||
|
*/
|
||||||
|
export const useAllEventNotifications = (onNewEvent) => {
|
||||||
|
return useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
importance: 'all',
|
||||||
|
onNewEvent,
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Hook：订阅重要事件（S 和 A 级）
|
||||||
|
*/
|
||||||
|
export const useImportantEventNotifications = (onNewEvent) => {
|
||||||
|
const [importantEvents, setImportantEvents] = useState([]);
|
||||||
|
|
||||||
|
const handleEvent = (event) => {
|
||||||
|
// 只处理 S 和 A 级事件
|
||||||
|
if (event.importance === 'S' || event.importance === 'A') {
|
||||||
|
setImportantEvents(prev => [event, ...prev].slice(0, 10)); // 最多保留 10 个
|
||||||
|
if (onNewEvent) {
|
||||||
|
onNewEvent(event);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const result = useEventNotifications({
|
||||||
|
eventType: 'all',
|
||||||
|
importance: 'all',
|
||||||
|
onNewEvent: handleEvent,
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
...result,
|
||||||
|
importantEvents,
|
||||||
|
clearImportantEvents: () => setImportantEvents([]),
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
export default useEventNotifications;
|
||||||
@@ -186,6 +186,169 @@ class SocketService {
|
|||||||
getSocketId() {
|
getSocketId() {
|
||||||
return this.socket?.id || null;
|
return this.socket?.id || null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ==================== 事件推送专用方法 ====================
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 订阅事件推送
|
||||||
|
* @param {object} options - 订阅选项
|
||||||
|
* @param {string} options.eventType - 事件类型 ('all' | 'policy' | 'market' | 'tech' | ...)
|
||||||
|
* @param {string} options.importance - 重要性 ('all' | 'S' | 'A' | 'B' | 'C')
|
||||||
|
* @param {Function} options.onNewEvent - 收到新事件时的回调函数
|
||||||
|
* @param {Function} options.onSubscribed - 订阅成功的回调函数（可选）
|
||||||
|
*/
|
||||||
|
subscribeToEvents(options = {}) {
|
||||||
|
const {
|
||||||
|
eventType = 'all',
|
||||||
|
importance = 'all',
|
||||||
|
onNewEvent,
|
||||||
|
onSubscribed,
|
||||||
|
} = options;
|
||||||
|
|
||||||
|
if (!this.socket || !this.connected) {
|
||||||
|
logger.warn('socketService', 'Cannot subscribe: socket not connected');
|
||||||
|
// 自动连接
|
||||||
|
this.connect();
|
||||||
|
// 等待连接成功后再订阅
|
||||||
|
this.socket.once('connect', () => {
|
||||||
|
this._doSubscribe(eventType, importance, onNewEvent, onSubscribed);
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
this._doSubscribe(eventType, importance, onNewEvent, onSubscribed);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行订阅操作（内部方法）
|
||||||
|
*/
|
||||||
|
_doSubscribe(eventType, importance, onNewEvent, onSubscribed) {
|
||||||
|
// 发送订阅请求
|
||||||
|
this.emit('subscribe_events', {
|
||||||
|
event_type: eventType,
|
||||||
|
importance: importance,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 监听订阅确认
|
||||||
|
this.socket.once('subscription_confirmed', (data) => {
|
||||||
|
logger.info('socketService', 'Subscription confirmed', data);
|
||||||
|
if (onSubscribed) {
|
||||||
|
onSubscribed(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 监听订阅错误
|
||||||
|
this.socket.once('subscription_error', (error) => {
|
||||||
|
logger.error('socketService', 'Subscription error', error);
|
||||||
|
});
|
||||||
|
|
||||||
|
// 监听新事件推送
|
||||||
|
if (onNewEvent) {
|
||||||
|
// 先移除之前的监听器（避免重复）
|
||||||
|
this.socket.off('new_event');
|
||||||
|
// 添加新的监听器
|
||||||
|
this.socket.on('new_event', (eventData) => {
|
||||||
|
logger.info('socketService', 'New event received', eventData);
|
||||||
|
onNewEvent(eventData);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 取消订阅事件推送
|
||||||
|
* @param {object} options - 取消订阅选项
|
||||||
|
* @param {string} options.eventType - 事件类型
|
||||||
|
* @param {Function} options.onUnsubscribed - 取消订阅成功的回调函数（可选）
|
||||||
|
*/
|
||||||
|
unsubscribeFromEvents(options = {}) {
|
||||||
|
const {
|
||||||
|
eventType = 'all',
|
||||||
|
onUnsubscribed,
|
||||||
|
} = options;
|
||||||
|
|
||||||
|
if (!this.socket || !this.connected) {
|
||||||
|
logger.warn('socketService', 'Cannot unsubscribe: socket not connected');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 发送取消订阅请求
|
||||||
|
this.emit('unsubscribe_events', {
|
||||||
|
event_type: eventType,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 监听取消订阅确认
|
||||||
|
this.socket.once('unsubscription_confirmed', (data) => {
|
||||||
|
logger.info('socketService', 'Unsubscription confirmed', data);
|
||||||
|
|
||||||
|
// 移除新事件监听器
|
||||||
|
this.socket.off('new_event');
|
||||||
|
|
||||||
|
if (onUnsubscribed) {
|
||||||
|
onUnsubscribed(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// 监听取消订阅错误
|
||||||
|
this.socket.once('unsubscription_error', (error) => {
|
||||||
|
logger.error('socketService', 'Unsubscription error', error);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 快捷方法：订阅所有类型的事件
|
||||||
|
* @param {Function} onNewEvent - 收到新事件时的回调函数
|
||||||
|
* @returns {Function} 取消订阅的函数
|
||||||
|
*/
|
||||||
|
subscribeToAllEvents(onNewEvent) {
|
||||||
|
this.subscribeToEvents({
|
||||||
|
eventType: 'all',
|
||||||
|
importance: 'all',
|
||||||
|
onNewEvent,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 返回取消订阅的清理函数
|
||||||
|
return () => {
|
||||||
|
this.unsubscribeFromEvents({ eventType: 'all' });
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 快捷方法：订阅特定重要性的事件
|
||||||
|
* @param {string} importance - 重要性级别 ('S' | 'A' | 'B' | 'C')
|
||||||
|
* @param {Function} onNewEvent - 收到新事件时的回调函数
|
||||||
|
* @returns {Function} 取消订阅的函数
|
||||||
|
*/
|
||||||
|
subscribeToImportantEvents(importance, onNewEvent) {
|
||||||
|
this.subscribeToEvents({
|
||||||
|
eventType: 'all',
|
||||||
|
importance,
|
||||||
|
onNewEvent,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 返回取消订阅的清理函数
|
||||||
|
return () => {
|
||||||
|
this.unsubscribeFromEvents({ eventType: 'all' });
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 快捷方法：订阅特定类型的事件
|
||||||
|
* @param {string} eventType - 事件类型
|
||||||
|
* @param {Function} onNewEvent - 收到新事件时的回调函数
|
||||||
|
* @returns {Function} 取消订阅的函数
|
||||||
|
*/
|
||||||
|
subscribeToEventType(eventType, onNewEvent) {
|
||||||
|
this.subscribeToEvents({
|
||||||
|
eventType,
|
||||||
|
importance: 'all',
|
||||||
|
onNewEvent,
|
||||||
|
});
|
||||||
|
|
||||||
|
// 返回取消订阅的清理函数
|
||||||
|
return () => {
|
||||||
|
this.unsubscribeFromEvents({ eventType });
|
||||||
|
};
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 导出单例
|
// 导出单例
|
||||||
|
|||||||
109
test_create_event.py
Normal file
109
test_create_event.py
Normal file
@@ -0,0 +1,109 @@
|
|||||||
|
"""
|
||||||
|
测试脚本：手动创建事件到数据库
|
||||||
|
用于测试 WebSocket 实时推送功能
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
from datetime import datetime
|
||||||
|
from sqlalchemy import create_engine, Column, Integer, String, Text, Float, DateTime
|
||||||
|
from sqlalchemy.ext.declarative import declarative_base
|
||||||
|
from sqlalchemy.orm import sessionmaker
|
||||||
|
|
||||||
|
# 数据库连接（从 app.py 复制）
|
||||||
|
DATABASE_URI = 'mysql+pymysql://root:Zzl5588161!@111.198.58.126:33060/stock?charset=utf8mb4'
|
||||||
|
|
||||||
|
engine = create_engine(DATABASE_URI, echo=False)
|
||||||
|
Session = sessionmaker(bind=engine)
|
||||||
|
session = Session()
|
||||||
|
|
||||||
|
Base = declarative_base()
|
||||||
|
|
||||||
|
# Event 模型（简化版，只包含必要字段）
|
||||||
|
class Event(Base):
|
||||||
|
__tablename__ = 'events'
|
||||||
|
|
||||||
|
id = Column(Integer, primary_key=True)
|
||||||
|
title = Column(String(500), nullable=False)
|
||||||
|
description = Column(Text)
|
||||||
|
event_type = Column(String(100))
|
||||||
|
importance = Column(String(10))
|
||||||
|
status = Column(String(50), default='active')
|
||||||
|
hot_score = Column(Float, default=0)
|
||||||
|
view_count = Column(Integer, default=0)
|
||||||
|
created_at = Column(DateTime, default=datetime.now)
|
||||||
|
updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now)
|
||||||
|
|
||||||
|
|
||||||
|
def create_test_event():
|
||||||
|
"""创建一个测试事件"""
|
||||||
|
|
||||||
|
import random
|
||||||
|
|
||||||
|
event_types = ['policy', 'market', 'tech', 'industry', 'finance']
|
||||||
|
importances = ['S', 'A', 'B', 'C']
|
||||||
|
|
||||||
|
test_event = Event(
|
||||||
|
title=f'测试事件 - {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}',
|
||||||
|
description=f'这是一个用于测试 WebSocket 实时推送的事件，创建于 {datetime.now()}',
|
||||||
|
event_type=random.choice(event_types),
|
||||||
|
importance=random.choice(importances),
|
||||||
|
status='active',
|
||||||
|
hot_score=round(random.uniform(50, 100), 2),
|
||||||
|
view_count=random.randint(100, 1000)
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
session.add(test_event)
|
||||||
|
session.commit()
|
||||||
|
|
||||||
|
print("✅ 测试事件创建成功!")
|
||||||
|
print(f" ID: {test_event.id}")
|
||||||
|
print(f" 标题: {test_event.title}")
|
||||||
|
print(f" 类型: {test_event.event_type}")
|
||||||
|
print(f" 重要性: {test_event.importance}")
|
||||||
|
print(f" 热度: {test_event.hot_score}")
|
||||||
|
print(f"\n💡 提示: 轮询将在 2 分钟内检测到此事件并推送到前端")
|
||||||
|
print(f" （如果需要立即推送，请将轮询间隔改为更短）")
|
||||||
|
|
||||||
|
return test_event.id
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
session.rollback()
|
||||||
|
print(f"❌ 创建事件失败: {e}")
|
||||||
|
return None
|
||||||
|
finally:
|
||||||
|
session.close()
|
||||||
|
|
||||||
|
|
||||||
|
def create_multiple_events(count=3):
|
||||||
|
"""创建多个测试事件"""
|
||||||
|
print(f"正在创建 {count} 个测试事件...\n")
|
||||||
|
|
||||||
|
for i in range(count):
|
||||||
|
event_id = create_test_event()
|
||||||
|
if event_id:
|
||||||
|
print(f"[{i+1}/{count}] 事件 #{event_id} 创建成功\n")
|
||||||
|
else:
|
||||||
|
print(f"[{i+1}/{count}] 创建失败\n")
|
||||||
|
|
||||||
|
print(f"\n✅ 完成！共创建 {count} 个事件")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
print("=" * 60)
|
||||||
|
print("WebSocket 事件推送测试 - 手动创建事件")
|
||||||
|
print("=" * 60)
|
||||||
|
print()
|
||||||
|
|
||||||
|
if len(sys.argv) > 1:
|
||||||
|
try:
|
||||||
|
count = int(sys.argv[1])
|
||||||
|
create_multiple_events(count)
|
||||||
|
except ValueError:
|
||||||
|
print("❌ 参数必须是数字")
|
||||||
|
print("用法: python test_create_event.py [数量]")
|
||||||
|
else:
|
||||||
|
# 默认创建 1 个事件
|
||||||
|
create_test_event()
|
||||||
|
|
||||||
|
print("\n" + "=" * 60)
|
||||||
289
test_websocket.html
Normal file
289
test_websocket.html
Normal file
@@ -0,0 +1,289 @@
|
|||||||
|
<!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>
|
||||||
|
<script src="https://cdn.socket.io/4.7.4/socket.io.min.js"></script>
|
||||||
|
<style>
|
||||||
|
body {
|
||||||
|
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
||||||
|
max-width: 1000px;
|
||||||
|
margin: 50px auto;
|
||||||
|
padding: 20px;
|
||||||
|
background: #f5f5f5;
|
||||||
|
}
|
||||||
|
.container {
|
||||||
|
background: white;
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 30px;
|
||||||
|
box-shadow: 0 2px 10px rgba(0,0,0,0.1);
|
||||||
|
}
|
||||||
|
h1 {
|
||||||
|
color: #333;
|
||||||
|
border-bottom: 3px solid #4CAF50;
|
||||||
|
padding-bottom: 10px;
|
||||||
|
}
|
||||||
|
.status {
|
||||||
|
padding: 15px;
|
||||||
|
margin: 20px 0;
|
||||||
|
border-radius: 5px;
|
||||||
|
font-weight: bold;
|
||||||
|
}
|
||||||
|
.status.connected {
|
||||||
|
background: #d4edda;
|
||||||
|
color: #155724;
|
||||||
|
border: 1px solid #c3e6cb;
|
||||||
|
}
|
||||||
|
.status.disconnected {
|
||||||
|
background: #f8d7da;
|
||||||
|
color: #721c24;
|
||||||
|
border: 1px solid #f5c6cb;
|
||||||
|
}
|
||||||
|
button {
|
||||||
|
background: #4CAF50;
|
||||||
|
color: white;
|
||||||
|
border: none;
|
||||||
|
padding: 12px 24px;
|
||||||
|
font-size: 16px;
|
||||||
|
border-radius: 5px;
|
||||||
|
cursor: pointer;
|
||||||
|
margin: 5px;
|
||||||
|
}
|
||||||
|
button:hover {
|
||||||
|
background: #45a049;
|
||||||
|
}
|
||||||
|
button:disabled {
|
||||||
|
background: #ccc;
|
||||||
|
cursor: not-allowed;
|
||||||
|
}
|
||||||
|
.log {
|
||||||
|
background: #f9f9f9;
|
||||||
|
border: 1px solid #ddd;
|
||||||
|
border-radius: 5px;
|
||||||
|
padding: 15px;
|
||||||
|
margin-top: 20px;
|
||||||
|
max-height: 400px;
|
||||||
|
overflow-y: auto;
|
||||||
|
font-family: 'Courier New', monospace;
|
||||||
|
font-size: 13px;
|
||||||
|
}
|
||||||
|
.log-item {
|
||||||
|
padding: 8px;
|
||||||
|
margin: 5px 0;
|
||||||
|
border-left: 3px solid #4CAF50;
|
||||||
|
background: white;
|
||||||
|
}
|
||||||
|
.log-item.error {
|
||||||
|
border-left-color: #f44336;
|
||||||
|
background: #ffebee;
|
||||||
|
}
|
||||||
|
.log-item.event {
|
||||||
|
border-left-color: #2196F3;
|
||||||
|
background: #e3f2fd;
|
||||||
|
}
|
||||||
|
.event-card {
|
||||||
|
background: #fff3cd;
|
||||||
|
border: 2px solid #ffc107;
|
||||||
|
border-radius: 8px;
|
||||||
|
padding: 15px;
|
||||||
|
margin: 10px 0;
|
||||||
|
}
|
||||||
|
.event-card h3 {
|
||||||
|
margin-top: 0;
|
||||||
|
color: #856404;
|
||||||
|
}
|
||||||
|
.timestamp {
|
||||||
|
color: #666;
|
||||||
|
font-size: 12px;
|
||||||
|
}
|
||||||
|
.controls {
|
||||||
|
margin: 20px 0;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="container">
|
||||||
|
<h1>🔌 WebSocket 事件推送测试页面</h1>
|
||||||
|
|
||||||
|
<div id="status" class="status disconnected">
|
||||||
|
状态: 未连接
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="controls">
|
||||||
|
<button id="connectBtn" onclick="connect()">连接 WebSocket</button>
|
||||||
|
<button id="subscribeBtn" onclick="subscribe()" disabled>订阅所有事件</button>
|
||||||
|
<button id="unsubscribeBtn" onclick="unsubscribe()" disabled>取消订阅</button>
|
||||||
|
<button id="disconnectBtn" onclick="disconnect()" disabled>断开连接</button>
|
||||||
|
<button onclick="clearLog()">清空日志</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div id="events"></div>
|
||||||
|
|
||||||
|
<h2>📋 日志</h2>
|
||||||
|
<div id="log" class="log"></div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
let socket = null;
|
||||||
|
let isSubscribed = false;
|
||||||
|
|
||||||
|
function updateStatus(connected) {
|
||||||
|
const statusDiv = document.getElementById('status');
|
||||||
|
const connectBtn = document.getElementById('connectBtn');
|
||||||
|
const subscribeBtn = document.getElementById('subscribeBtn');
|
||||||
|
const unsubscribeBtn = document.getElementById('unsubscribeBtn');
|
||||||
|
const disconnectBtn = document.getElementById('disconnectBtn');
|
||||||
|
|
||||||
|
if (connected) {
|
||||||
|
statusDiv.className = 'status connected';
|
||||||
|
statusDiv.textContent = '状态: 已连接 ✓';
|
||||||
|
connectBtn.disabled = true;
|
||||||
|
subscribeBtn.disabled = false;
|
||||||
|
disconnectBtn.disabled = false;
|
||||||
|
} else {
|
||||||
|
statusDiv.className = 'status disconnected';
|
||||||
|
statusDiv.textContent = '状态: 未连接 ✗';
|
||||||
|
connectBtn.disabled = false;
|
||||||
|
subscribeBtn.disabled = true;
|
||||||
|
unsubscribeBtn.disabled = true;
|
||||||
|
disconnectBtn.disabled = true;
|
||||||
|
isSubscribed = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function log(message, type = 'info') {
|
||||||
|
const logDiv = document.getElementById('log');
|
||||||
|
const timestamp = new Date().toLocaleTimeString();
|
||||||
|
const item = document.createElement('div');
|
||||||
|
item.className = `log-item ${type}`;
|
||||||
|
item.innerHTML = `<span class="timestamp">[${timestamp}]</span> ${message}`;
|
||||||
|
logDiv.insertBefore(item, logDiv.firstChild);
|
||||||
|
console.log(`[${timestamp}] ${message}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
function showEvent(event) {
|
||||||
|
const eventsDiv = document.getElementById('events');
|
||||||
|
const card = document.createElement('div');
|
||||||
|
card.className = 'event-card';
|
||||||
|
card.innerHTML = `
|
||||||
|
<h3>🔔 新事件通知</h3>
|
||||||
|
<p><strong>ID:</strong> ${event.id}</p>
|
||||||
|
<p><strong>标题:</strong> ${event.title}</p>
|
||||||
|
<p><strong>类型:</strong> ${event.event_type || 'N/A'}</p>
|
||||||
|
<p><strong>重要性:</strong> ${event.importance || 'N/A'}</p>
|
||||||
|
<p><strong>描述:</strong> ${event.description || 'N/A'}</p>
|
||||||
|
<p><strong>热度:</strong> ${event.hot_score || 0}</p>
|
||||||
|
<p class="timestamp">收到时间: ${new Date().toLocaleString()}</p>
|
||||||
|
`;
|
||||||
|
eventsDiv.insertBefore(card, eventsDiv.firstChild);
|
||||||
|
|
||||||
|
// 只保留最近 5 个事件
|
||||||
|
while (eventsDiv.children.length > 5) {
|
||||||
|
eventsDiv.removeChild(eventsDiv.lastChild);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function connect() {
|
||||||
|
log('正在连接到 WebSocket 服务器...');
|
||||||
|
|
||||||
|
socket = io('http://localhost:5001', {
|
||||||
|
transports: ['websocket', 'polling'],
|
||||||
|
reconnection: true,
|
||||||
|
reconnectionAttempts: 5,
|
||||||
|
reconnectionDelay: 1000
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('connect', () => {
|
||||||
|
log('✅ WebSocket 连接成功！', 'info');
|
||||||
|
updateStatus(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('connection_response', (data) => {
|
||||||
|
log(`📡 服务器响应: ${JSON.stringify(data)}`, 'info');
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('new_event', (eventData) => {
|
||||||
|
log(`🔔 收到新事件: ${eventData.title}`, 'event');
|
||||||
|
showEvent(eventData);
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('subscription_confirmed', (data) => {
|
||||||
|
log(`✅ 订阅成功: ${JSON.stringify(data)}`, 'info');
|
||||||
|
isSubscribed = true;
|
||||||
|
document.getElementById('subscribeBtn').disabled = true;
|
||||||
|
document.getElementById('unsubscribeBtn').disabled = false;
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('unsubscription_confirmed', (data) => {
|
||||||
|
log(`✅ 取消订阅成功: ${JSON.stringify(data)}`, 'info');
|
||||||
|
isSubscribed = false;
|
||||||
|
document.getElementById('subscribeBtn').disabled = false;
|
||||||
|
document.getElementById('unsubscribeBtn').disabled = true;
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('disconnect', () => {
|
||||||
|
log('❌ WebSocket 连接已断开', 'error');
|
||||||
|
updateStatus(false);
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('connect_error', (error) => {
|
||||||
|
log(`❌ 连接错误: ${error.message}`, 'error');
|
||||||
|
});
|
||||||
|
|
||||||
|
socket.on('error', (error) => {
|
||||||
|
log(`❌ 错误: ${error}`, 'error');
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function subscribe() {
|
||||||
|
if (!socket || !socket.connected) {
|
||||||
|
log('❌ 请先连接 WebSocket', 'error');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
log('发送订阅请求...');
|
||||||
|
socket.emit('subscribe_events', {
|
||||||
|
event_type: 'all',
|
||||||
|
importance: 'all'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function unsubscribe() {
|
||||||
|
if (!socket || !socket.connected) {
|
||||||
|
log('❌ 请先连接 WebSocket', 'error');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
log('发送取消订阅请求...');
|
||||||
|
socket.emit('unsubscribe_events', {
|
||||||
|
event_type: 'all'
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function disconnect() {
|
||||||
|
if (socket) {
|
||||||
|
socket.disconnect();
|
||||||
|
log('已手动断开连接');
|
||||||
|
updateStatus(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function clearLog() {
|
||||||
|
document.getElementById('log').innerHTML = '';
|
||||||
|
document.getElementById('events').innerHTML = '';
|
||||||
|
log('日志已清空');
|
||||||
|
}
|
||||||
|
|
||||||
|
// 页面加载时显示提示
|
||||||
|
window.onload = function() {
|
||||||
|
log('📌 测试步骤:');
|
||||||
|
log('1. 确保 Flask 服务已启动 (python app.py)');
|
||||||
|
log('2. 点击 "连接 WebSocket" 按钮');
|
||||||
|
log('3. 点击 "订阅所有事件" 按钮');
|
||||||
|
log('4. 等待后台脚本创建新事件，或等待轮询检测到新事件');
|
||||||
|
log('5. 新事件会自动推送到这个页面');
|
||||||
|
};
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
Reference in New Issue
Block a user