Default User
3 months ago
2 changed files with 419 additions and 49 deletions
@ -0,0 +1,284 @@ |
|||||
|
const WebSocket = require('ws'); |
||||
|
const readline = require('readline'); |
||||
|
const mysql = require('mysql2/promise'); |
||||
|
|
||||
|
// 数据库连接配置
|
||||
|
const dbConfig = { |
||||
|
host: 'localhost', |
||||
|
user: 'root', |
||||
|
password: '123456', |
||||
|
database: 'wechat_app' |
||||
|
}; |
||||
|
|
||||
|
// 测试配置
|
||||
|
const TEST_CONFIG = { |
||||
|
wsUrl: 'ws://localhost:3003', |
||||
|
testUserId: 'test_user_' + Date.now(), |
||||
|
testManagerId: '22', |
||||
|
testMessage: '这是一条测试消息 ' + Date.now() |
||||
|
}; |
||||
|
|
||||
|
// 创建命令行交互界面
|
||||
|
const rl = readline.createInterface({ |
||||
|
input: process.stdin, |
||||
|
output: process.stdout |
||||
|
}); |
||||
|
|
||||
|
// WebSocket测试客户端
|
||||
|
class WebSocketTestClient { |
||||
|
constructor() { |
||||
|
this.ws = null; |
||||
|
this.authenticated = false; |
||||
|
this.messages = []; |
||||
|
this.connectionAttempts = 0; |
||||
|
this.maxAttempts = 3; |
||||
|
} |
||||
|
|
||||
|
connect(serverUrl = TEST_CONFIG.wsUrl) { |
||||
|
console.log(`正在连接到服务器: ${serverUrl}`); |
||||
|
this.connectionAttempts++; |
||||
|
|
||||
|
this.ws = new WebSocket(serverUrl); |
||||
|
|
||||
|
this.ws.on('open', () => { |
||||
|
console.log('✅ WebSocket连接已建立'); |
||||
|
this.connectionAttempts = 0; // 重置连接尝试次数
|
||||
|
this.authenticate(); |
||||
|
}); |
||||
|
|
||||
|
this.ws.on('message', (data) => { |
||||
|
try { |
||||
|
const message = JSON.parse(data.toString()); |
||||
|
console.log('📨 收到消息:', JSON.stringify(message, null, 2)); |
||||
|
this.handleMessage(message); |
||||
|
} catch (error) { |
||||
|
console.error('❌ 解析消息失败:', error); |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
this.ws.on('error', (error) => { |
||||
|
console.error('❌ WebSocket错误:', error.message); |
||||
|
}); |
||||
|
|
||||
|
this.ws.on('close', (code, reason) => { |
||||
|
console.log(`🔌 WebSocket连接已关闭: ${code} - ${reason}`); |
||||
|
this.authenticated = false; |
||||
|
|
||||
|
// 如果不是主动关闭，尝试重连
|
||||
|
if (this.connectionAttempts < this.maxAttempts) { |
||||
|
console.log(`尝试重连 (${this.connectionAttempts}/${this.maxAttempts})...`); |
||||
|
setTimeout(() => this.connect(serverUrl), 2000); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
authenticate() { |
||||
|
console.log('正在发送认证信息...'); |
||||
|
const authMessage = { |
||||
|
type: 'auth', |
||||
|
userId: TEST_CONFIG.testUserId, |
||||
|
userType: 'user', |
||||
|
timestamp: Date.now() |
||||
|
}; |
||||
|
|
||||
|
console.log('🔑 发送认证请求:', JSON.stringify(authMessage)); |
||||
|
this.sendMessage(authMessage); |
||||
|
} |
||||
|
|
||||
|
handleMessage(message) { |
||||
|
// 存储接收到的消息
|
||||
|
this.messages.push(message); |
||||
|
|
||||
|
if (message.type === 'auth_success') { |
||||
|
console.log('✅ 认证成功！'); |
||||
|
this.authenticated = true; |
||||
|
this.sendTestMessage(); |
||||
|
} else if (message.type === 'auth_error') { |
||||
|
console.error('❌ 认证失败:', message.message); |
||||
|
} else if (message.type === 'message_sent') { |
||||
|
console.log('✅ 消息发送成功确认:', message.payload); |
||||
|
console.log('消息ID:', message.payload.messageId); |
||||
|
console.log('会话ID:', message.payload.conversationId); |
||||
|
} else if (message.type === 'new_message') { |
||||
|
console.log('📥 收到新消息:', message.payload); |
||||
|
this.messages.push(message.payload); |
||||
|
} else if (message.type === 'error') { |
||||
|
console.error('❌ 收到错误消息:', message.message); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 数据库查询函数
|
||||
|
async queryDatabase() { |
||||
|
let connection = null; |
||||
|
try { |
||||
|
console.log('\n🔍 查询数据库验证消息存储...'); |
||||
|
connection = await mysql.createConnection(dbConfig); |
||||
|
|
||||
|
// 查询最新消息
|
||||
|
const [messages] = await connection.execute( |
||||
|
`SELECT * FROM chat_messages
|
||||
|
ORDER BY created_at DESC |
||||
|
LIMIT 5` |
||||
|
); |
||||
|
|
||||
|
if (messages.length > 0) { |
||||
|
console.log(`\n📊 找到 ${messages.length} 条最新消息:`); |
||||
|
let testMessageFound = false; |
||||
|
|
||||
|
messages.forEach((msg, index) => { |
||||
|
console.log(`\n--- 消息 ${index + 1} ---`); |
||||
|
console.log(`消息ID: ${msg.message_id}`); |
||||
|
console.log(`会话ID: ${msg.conversation_id}`); |
||||
|
console.log(`发送者ID: ${msg.sender_id}`); |
||||
|
console.log(`内容: ${msg.content}`); |
||||
|
|
||||
|
if (msg.content === TEST_CONFIG.testMessage) { |
||||
|
console.log('✅ 测试消息已成功存储到数据库!'); |
||||
|
testMessageFound = true; |
||||
|
} |
||||
|
}); |
||||
|
|
||||
|
if (!testMessageFound) { |
||||
|
console.log('\n⚠️ 未找到测试消息'); |
||||
|
} |
||||
|
} else { |
||||
|
console.log('\n❌ 未找到任何消息记录'); |
||||
|
} |
||||
|
|
||||
|
// 查询最新会话
|
||||
|
const [conversations] = await connection.execute( |
||||
|
`SELECT * FROM chat_conversations
|
||||
|
ORDER BY created_at DESC |
||||
|
LIMIT 5` |
||||
|
); |
||||
|
|
||||
|
if (conversations.length > 0) { |
||||
|
console.log(`\n📋 找到 ${conversations.length} 条最新会话:`); |
||||
|
conversations.forEach((conv, index) => { |
||||
|
console.log(`\n--- 会话 ${index + 1} ---`); |
||||
|
console.log(`会话ID: ${conv.conversation_id}`); |
||||
|
console.log(`用户ID: ${conv.userId}`); |
||||
|
console.log(`客服ID: ${conv.managerId}`); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
} catch (error) { |
||||
|
console.error('\n❌ 数据库查询失败:', error.message); |
||||
|
} finally { |
||||
|
if (connection) { |
||||
|
await connection.end(); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
sendTestMessage() { |
||||
|
// 生成临时会话ID（前端格式: temp_[currentUserId]_[targetId]_[timestamp]）
|
||||
|
const timestamp = Date.now(); |
||||
|
const tempConversationId = `temp_${TEST_CONFIG.testUserId}_${TEST_CONFIG.testManagerId}_${timestamp}`; |
||||
|
|
||||
|
console.log(`📝 生成的临时会话ID: ${tempConversationId}`); |
||||
|
|
||||
|
const testMessage = { |
||||
|
type: 'chat_message', |
||||
|
conversationId: tempConversationId, |
||||
|
receiverId: TEST_CONFIG.testManagerId, |
||||
|
senderId: TEST_CONFIG.testUserId, |
||||
|
senderType: 1, |
||||
|
content: TEST_CONFIG.testMessage, |
||||
|
contentType: 1, |
||||
|
messageId: 'test_msg_' + Date.now(), |
||||
|
timestamp: Date.now() |
||||
|
}; |
||||
|
|
||||
|
console.log('📤 发送测试消息:', JSON.stringify(testMessage)); |
||||
|
this.sendMessage(testMessage); |
||||
|
|
||||
|
// 发送后自动查询数据库验证
|
||||
|
setTimeout(() => { |
||||
|
this.queryDatabase().catch(console.error); |
||||
|
}, 2000); |
||||
|
} |
||||
|
|
||||
|
testDatabaseStorage() { |
||||
|
console.log('\n======================================='); |
||||
|
console.log('📊 测试数据库存储功能'); |
||||
|
console.log('将自动查询数据库验证消息存储'); |
||||
|
console.log('=======================================\n'); |
||||
|
|
||||
|
// 自动查询数据库
|
||||
|
this.queryDatabase().catch(console.error); |
||||
|
|
||||
|
// 询问用户是否要发送更多消息
|
||||
|
this.promptForMoreMessages(); |
||||
|
} |
||||
|
|
||||
|
promptForMoreMessages() { |
||||
|
rl.question('是否发送另一条测试消息？(y/n): ', (answer) => { |
||||
|
if (answer.toLowerCase() === 'y') { |
||||
|
this.sendTestMessage(); |
||||
|
} else { |
||||
|
console.log('\n测试完成！按Ctrl+C退出。'); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
sendMessage(message) { |
||||
|
if (this.ws && this.ws.readyState === WebSocket.OPEN) { |
||||
|
this.ws.send(JSON.stringify(message)); |
||||
|
} else { |
||||
|
console.error('❌ WebSocket未连接，无法发送消息'); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
close() { |
||||
|
if (this.ws) { |
||||
|
this.ws.close(); |
||||
|
} |
||||
|
rl.close(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
// 自动测试函数
|
||||
|
async function autoTest() { |
||||
|
console.log('\n========================================'); |
||||
|
console.log('开始自动测试消息存储功能'); |
||||
|
console.log('========================================\n'); |
||||
|
|
||||
|
const client = new WebSocketTestClient(); |
||||
|
|
||||
|
// 重写handleMessage以支持自动测试
|
||||
|
const originalHandleMessage = client.handleMessage.bind(client); |
||||
|
client.handleMessage = function(message) { |
||||
|
originalHandleMessage(message); |
||||
|
|
||||
|
if (message.type === 'message_sent') { |
||||
|
// 消息发送成功后查询数据库
|
||||
|
setTimeout(() => { |
||||
|
client.queryDatabase().then(() => { |
||||
|
console.log('\n========================================'); |
||||
|
console.log('自动测试完成'); |
||||
|
console.log('========================================'); |
||||
|
process.exit(0); |
||||
|
}); |
||||
|
}, 2000); |
||||
|
} |
||||
|
}; |
||||
|
|
||||
|
client.connect(); |
||||
|
} |
||||
|
|
||||
|
// 启动测试客户端
|
||||
|
console.log('======================================='); |
||||
|
console.log('🚀 消息存储测试工具'); |
||||
|
console.log('这个工具将测试WebSocket连接、认证和消息存储功能'); |
||||
|
console.log('=======================================\n'); |
||||
|
|
||||
|
// 自动模式运行
|
||||
|
console.log('🎯 以自动测试模式运行...'); |
||||
|
autoTest(); |
||||
|
|
||||
|
// 处理进程终止
|
||||
|
process.on('SIGINT', () => { |
||||
|
console.log('\n正在关闭连接...'); |
||||
|
process.exit(0); |
||||
|
}); |
||||
Loading…
Reference in new issue