315 lines
8.1 KiB
TypeScript
315 lines
8.1 KiB
TypeScript
/**
|
||
* 分销模块WebSocket实时推送服务
|
||
* 用于推送绑定过期提醒、提现状态更新等实时消息
|
||
*/
|
||
|
||
// 消息类型定义
|
||
export type WebSocketMessageType =
|
||
| 'binding_expiring' // 绑定即将过期
|
||
| 'binding_expired' // 绑定已过期
|
||
| 'binding_converted' // 绑定已转化（用户付款）
|
||
| 'withdrawal_approved' // 提现已通过
|
||
| 'withdrawal_completed' // 提现已完成
|
||
| 'withdrawal_rejected' // 提现已拒绝
|
||
| 'earnings_added' // 收益增加
|
||
| 'system_notice'; // 系统通知
|
||
|
||
// 消息结构
|
||
export interface WebSocketMessage {
|
||
type: WebSocketMessageType;
|
||
userId: string; // 目标用户ID
|
||
data: Record<string, unknown>;
|
||
timestamp: string;
|
||
messageId: string;
|
||
}
|
||
|
||
// 绑定过期提醒数据
|
||
export interface BindingExpiringData {
|
||
bindingId: string;
|
||
visitorNickname?: string;
|
||
visitorPhone?: string;
|
||
daysRemaining: number;
|
||
expireTime: string;
|
||
}
|
||
|
||
// 提现状态更新数据
|
||
export interface WithdrawalUpdateData {
|
||
withdrawalId: string;
|
||
amount: number;
|
||
status: string;
|
||
paymentNo?: string;
|
||
rejectReason?: string;
|
||
}
|
||
|
||
// 收益增加数据
|
||
export interface EarningsAddedData {
|
||
orderId: string;
|
||
orderAmount: number;
|
||
commission: number;
|
||
visitorNickname?: string;
|
||
}
|
||
|
||
/**
|
||
* WebSocket消息队列（服务端存储待发送的消息）
|
||
* 实际项目中应该使用Redis或其他消息队列
|
||
*/
|
||
const messageQueue: Map<string, WebSocketMessage[]> = new Map();
|
||
|
||
/**
|
||
* 生成消息ID
|
||
*/
|
||
function generateMessageId(): string {
|
||
return `msg_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
|
||
}
|
||
|
||
/**
|
||
* 添加消息到队列
|
||
*/
|
||
export function pushMessage(message: Omit<WebSocketMessage, 'messageId' | 'timestamp'>): void {
|
||
const fullMessage: WebSocketMessage = {
|
||
...message,
|
||
messageId: generateMessageId(),
|
||
timestamp: new Date().toISOString(),
|
||
};
|
||
|
||
const userMessages = messageQueue.get(message.userId) || [];
|
||
userMessages.push(fullMessage);
|
||
|
||
// 保留最近100条消息
|
||
if (userMessages.length > 100) {
|
||
userMessages.shift();
|
||
}
|
||
|
||
messageQueue.set(message.userId, userMessages);
|
||
|
||
console.log('[WebSocket] 消息已入队:', {
|
||
type: fullMessage.type,
|
||
userId: fullMessage.userId,
|
||
messageId: fullMessage.messageId,
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 获取用户待处理的消息
|
||
*/
|
||
export function getMessages(userId: string, since?: string): WebSocketMessage[] {
|
||
const userMessages = messageQueue.get(userId) || [];
|
||
|
||
if (since) {
|
||
return userMessages.filter(m => m.timestamp > since);
|
||
}
|
||
|
||
return userMessages;
|
||
}
|
||
|
||
/**
|
||
* 清除用户已读消息
|
||
*/
|
||
export function clearMessages(userId: string, messageIds: string[]): void {
|
||
const userMessages = messageQueue.get(userId) || [];
|
||
const filtered = userMessages.filter(m => !messageIds.includes(m.messageId));
|
||
messageQueue.set(userId, filtered);
|
||
}
|
||
|
||
/**
|
||
* 推送绑定即将过期提醒
|
||
*/
|
||
export function pushBindingExpiringReminder(params: {
|
||
userId: string;
|
||
bindingId: string;
|
||
visitorNickname?: string;
|
||
visitorPhone?: string;
|
||
daysRemaining: number;
|
||
expireTime: string;
|
||
}): void {
|
||
pushMessage({
|
||
type: 'binding_expiring',
|
||
userId: params.userId,
|
||
data: {
|
||
bindingId: params.bindingId,
|
||
visitorNickname: params.visitorNickname,
|
||
visitorPhone: params.visitorPhone,
|
||
daysRemaining: params.daysRemaining,
|
||
expireTime: params.expireTime,
|
||
message: `用户 ${params.visitorNickname || params.visitorPhone || '未知'} 的绑定将在 ${params.daysRemaining} 天后过期`,
|
||
},
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 推送绑定已过期通知
|
||
*/
|
||
export function pushBindingExpiredNotice(params: {
|
||
userId: string;
|
||
bindingId: string;
|
||
visitorNickname?: string;
|
||
visitorPhone?: string;
|
||
}): void {
|
||
pushMessage({
|
||
type: 'binding_expired',
|
||
userId: params.userId,
|
||
data: {
|
||
bindingId: params.bindingId,
|
||
visitorNickname: params.visitorNickname,
|
||
visitorPhone: params.visitorPhone,
|
||
message: `用户 ${params.visitorNickname || params.visitorPhone || '未知'} 的绑定已过期`,
|
||
},
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 推送绑定转化通知（用户付款）
|
||
*/
|
||
export function pushBindingConvertedNotice(params: {
|
||
userId: string;
|
||
bindingId: string;
|
||
orderId: string;
|
||
orderAmount: number;
|
||
commission: number;
|
||
visitorNickname?: string;
|
||
}): void {
|
||
pushMessage({
|
||
type: 'binding_converted',
|
||
userId: params.userId,
|
||
data: {
|
||
bindingId: params.bindingId,
|
||
orderId: params.orderId,
|
||
orderAmount: params.orderAmount,
|
||
commission: params.commission,
|
||
visitorNickname: params.visitorNickname,
|
||
message: `恭喜！用户 ${params.visitorNickname || '未知'} 已付款 ¥${params.orderAmount}，您获得佣金 ¥${params.commission.toFixed(2)}`,
|
||
},
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 推送提现状态更新
|
||
*/
|
||
export function pushWithdrawalUpdate(params: {
|
||
userId: string;
|
||
withdrawalId: string;
|
||
amount: number;
|
||
status: 'approved' | 'completed' | 'rejected';
|
||
paymentNo?: string;
|
||
rejectReason?: string;
|
||
}): void {
|
||
const type: WebSocketMessageType =
|
||
params.status === 'approved' ? 'withdrawal_approved' :
|
||
params.status === 'completed' ? 'withdrawal_completed' : 'withdrawal_rejected';
|
||
|
||
const messages: Record<string, string> = {
|
||
approved: `您的提现申请 ¥${params.amount.toFixed(2)} 已通过审核，正在打款中...`,
|
||
completed: `您的提现 ¥${params.amount.toFixed(2)} 已成功到账，流水号: ${params.paymentNo}`,
|
||
rejected: `您的提现申请 ¥${params.amount.toFixed(2)} 已被拒绝，原因: ${params.rejectReason || '未说明'}`,
|
||
};
|
||
|
||
pushMessage({
|
||
type,
|
||
userId: params.userId,
|
||
data: {
|
||
withdrawalId: params.withdrawalId,
|
||
amount: params.amount,
|
||
status: params.status,
|
||
paymentNo: params.paymentNo,
|
||
rejectReason: params.rejectReason,
|
||
message: messages[params.status],
|
||
},
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 推送收益增加通知
|
||
*/
|
||
export function pushEarningsAdded(params: {
|
||
userId: string;
|
||
orderId: string;
|
||
orderAmount: number;
|
||
commission: number;
|
||
visitorNickname?: string;
|
||
}): void {
|
||
pushMessage({
|
||
type: 'earnings_added',
|
||
userId: params.userId,
|
||
data: {
|
||
orderId: params.orderId,
|
||
orderAmount: params.orderAmount,
|
||
commission: params.commission,
|
||
visitorNickname: params.visitorNickname,
|
||
message: `收益 +¥${params.commission.toFixed(2)}`,
|
||
},
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 推送系统通知
|
||
*/
|
||
export function pushSystemNotice(params: {
|
||
userId: string;
|
||
title: string;
|
||
content: string;
|
||
link?: string;
|
||
}): void {
|
||
pushMessage({
|
||
type: 'system_notice',
|
||
userId: params.userId,
|
||
data: {
|
||
title: params.title,
|
||
content: params.content,
|
||
link: params.link,
|
||
},
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 客户端WebSocket Hook（用于React组件）
|
||
* 使用轮询模式获取实时消息
|
||
*/
|
||
export function createWebSocketClient(userId: string, onMessage: (message: WebSocketMessage) => void) {
|
||
let lastTimestamp = new Date().toISOString();
|
||
let isRunning = false;
|
||
let intervalId: NodeJS.Timeout | null = null;
|
||
|
||
const fetchMessages = async () => {
|
||
if (!isRunning) return;
|
||
|
||
try {
|
||
const response = await fetch(`/api/distribution/messages?userId=${userId}&since=${encodeURIComponent(lastTimestamp)}`);
|
||
if (!response.ok) return;
|
||
|
||
const data = await response.json();
|
||
if (data.success && data.messages?.length > 0) {
|
||
for (const message of data.messages) {
|
||
onMessage(message);
|
||
if (message.timestamp > lastTimestamp) {
|
||
lastTimestamp = message.timestamp;
|
||
}
|
||
}
|
||
}
|
||
} catch (error) {
|
||
console.error('[WebSocketClient] 获取消息失败:', error);
|
||
}
|
||
};
|
||
|
||
return {
|
||
connect: () => {
|
||
isRunning = true;
|
||
// 每3秒轮询一次
|
||
intervalId = setInterval(fetchMessages, 3000);
|
||
// 立即获取一次
|
||
fetchMessages();
|
||
console.log('[WebSocketClient] 已连接，用户:', userId);
|
||
},
|
||
|
||
disconnect: () => {
|
||
isRunning = false;
|
||
if (intervalId) {
|
||
clearInterval(intervalId);
|
||
intervalId = null;
|
||
}
|
||
console.log('[WebSocketClient] 已断开连接');
|
||
},
|
||
|
||
isConnected: () => isRunning,
|
||
};
|
||
}
|