import Taro from '@tarojs/taro';
|
import { getStorage } from './tools';
|
import { _UID } from '../constants/api';
|
|
type MessageCallback = (data: any) => void;
|
|
class WebSocketManager {
|
private static instance: WebSocketManager;
|
private socketTask: Taro.SocketTask | null = null;
|
private url: string = '';
|
private isConnected: boolean = false;
|
private reconnectAttempts: number = 0;
|
private maxReconnectAttempts: number = 5;
|
private reconnectTimer: any = null;
|
private heartbeatTimer: any = null;
|
private heartbeatInterval: number = 30000; // 30秒心跳
|
private listeners: Map<string, MessageCallback[]> = new Map();
|
private globalEventsBound: boolean = false;
|
|
private constructor() { }
|
|
public static getInstance(): WebSocketManager {
|
if (!WebSocketManager.instance) {
|
WebSocketManager.instance = new WebSocketManager();
|
}
|
return WebSocketManager.instance;
|
}
|
|
/**
|
* 初始化连接
|
* @param url WebSocket地址
|
*/
|
public async connect(url: string) {
|
if (this.socketTask || this.isConnected) {
|
console.log('WebSocket 已连接或正在连接中');
|
return;
|
}
|
|
this.url = url;
|
console.log(`正在连接 WebSocket: ${url}`);
|
|
const uid = await getStorage(_UID) || '-1';
|
const task = Taro.connectSocket({
|
header: {
|
'XP-Uid': '20260227200715' || uid.toString(),
|
},
|
url: this.url,
|
success: () => { },
|
fail: (err) => {
|
console.error('WebSocket 连接失败:', err);
|
this.handleReconnect();
|
}
|
});
|
const maybeTask: any = task as any;
|
if (maybeTask && typeof maybeTask.onOpen === 'function') {
|
this.socketTask = maybeTask;
|
this.initEventListeners();
|
return;
|
}
|
if (maybeTask && typeof maybeTask.then === 'function') {
|
maybeTask
|
.then((resolvedTask) => {
|
this.socketTask = resolvedTask;
|
this.initEventListeners();
|
})
|
.catch((err) => {
|
console.error('WebSocket 连接失败:', err);
|
this.handleReconnect();
|
});
|
return;
|
}
|
this.socketTask = null;
|
this.initEventListeners();
|
}
|
|
public isConnectedStatus() {
|
return this.isConnected;
|
}
|
|
/**
|
* 初始化事件监听
|
*/
|
private initEventListeners() {
|
if (this.socketTask && typeof this.socketTask.onOpen === 'function') {
|
this.socketTask.onOpen(async () => {
|
console.log('WebSocket 连接已打开');
|
this.isConnected = true;
|
this.reconnectAttempts = 0;
|
this.startHeartbeat();
|
const uid = await getStorage(_UID) || '-1';
|
this.send({ type: 'auth', appUserId: '20260227200715', timestamp: Date.now() });
|
this.notifyListeners('open', { status: 'connected' });
|
});
|
|
this.socketTask.onMessage((res) => {
|
try {
|
const data = JSON.parse(res.data as string);
|
const content = JSON.parse(data.content || '{}');
|
this.handleIncomingMessage(data, content);
|
if (data.type) {
|
this.notifyListeners(data.type, data);
|
}
|
this.notifyListeners('message', data);
|
} catch (e) {
|
console.warn('解析 WebSocket 消息失败:', res.data);
|
this.notifyListeners('message', res.data);
|
}
|
});
|
|
this.socketTask.onClose((res) => {
|
console.log('WebSocket 连接已关闭', res);
|
this.isConnected = false;
|
this.stopHeartbeat();
|
this.socketTask = null;
|
|
if (res.code !== 1000) {
|
this.handleReconnect();
|
}
|
this.notifyListeners('close', res);
|
});
|
|
this.socketTask.onError((err) => {
|
console.error('WebSocket 错误:', err);
|
this.isConnected = false;
|
this.notifyListeners('error', err);
|
});
|
return;
|
}
|
|
if (this.globalEventsBound) return;
|
this.globalEventsBound = true;
|
|
Taro.onSocketOpen(async () => {
|
console.log('WebSocket 连接已打开');
|
this.isConnected = true;
|
this.reconnectAttempts = 0;
|
this.startHeartbeat();
|
const uid = await getStorage(_UID) || '-1';
|
this.send({ type: 'auth', appUserId: '20260227200715', timestamp: Date.now() });
|
this.notifyListeners('open', { status: 'connected' });
|
});
|
|
Taro.onSocketMessage((res) => {
|
try {
|
const data = JSON.parse(res.data as string);
|
const content = JSON.parse(data.content || '{}');
|
this.handleIncomingMessage(data, content);
|
if (data.type) {
|
this.notifyListeners(data.type, data);
|
}
|
this.notifyListeners('message', data);
|
} catch (e) {
|
console.warn('解析 WebSocket 消息失败:', res.data);
|
this.notifyListeners('message', res.data);
|
}
|
});
|
|
Taro.onSocketClose((res) => {
|
console.log('WebSocket 连接已关闭', res);
|
this.isConnected = false;
|
this.stopHeartbeat();
|
this.socketTask = null;
|
|
if (res.code !== 1000) {
|
this.handleReconnect();
|
}
|
this.notifyListeners('close', res);
|
});
|
|
Taro.onSocketError((err) => {
|
console.error('WebSocket 错误:', err);
|
this.isConnected = false;
|
this.notifyListeners('error', err);
|
});
|
}
|
|
/**
|
* 发送消息
|
* @param data 消息内容
|
*/
|
public send(data: any) {
|
if (!this.isConnected) {
|
console.warn('WebSocket 未连接，消息未发送:', data);
|
return;
|
}
|
|
const message = typeof data === 'string' ? data : JSON.stringify(data);
|
if (this.socketTask) {
|
this.socketTask.send({
|
data: message,
|
fail: (err) => {
|
console.error('发送消息失败:', err);
|
}
|
});
|
} else {
|
Taro.sendSocketMessage({
|
data: message,
|
fail: (err) => {
|
console.error('发送消息失败:', err);
|
}
|
});
|
}
|
}
|
|
/**
|
* 关闭连接
|
*/
|
public close() {
|
this.stopHeartbeat();
|
if (this.reconnectTimer) {
|
clearTimeout(this.reconnectTimer);
|
this.reconnectTimer = null;
|
}
|
|
if (this.socketTask) {
|
this.socketTask.close({
|
code: 1000,
|
reason: '用户主动关闭连接'
|
});
|
this.socketTask = null;
|
}
|
this.isConnected = false;
|
}
|
|
/**
|
* 处理重连
|
*/
|
private handleReconnect() {
|
if (this.reconnectAttempts >= this.maxReconnectAttempts) {
|
console.error('达到最大重连次数，停止重连。');
|
return;
|
}
|
|
if (this.reconnectTimer) return;
|
|
const delay = Math.min(1000 * Math.pow(2, this.reconnectAttempts), 30000); // 指数退避
|
console.log(`正在尝试重连，延迟 ${delay}ms... (第 ${this.reconnectAttempts + 1}/${this.maxReconnectAttempts} 次尝试)`);
|
|
this.reconnectTimer = setTimeout(() => {
|
this.reconnectAttempts++;
|
this.reconnectTimer = null;
|
this.connect(this.url);
|
}, delay);
|
}
|
|
/**
|
* 开启心跳
|
*/
|
private startHeartbeat() {
|
this.stopHeartbeat();
|
this.heartbeatTimer = setInterval(() => {
|
if (this.isConnected) {
|
this.send({ type: 'ping', timestamp: Date.now() });
|
}
|
}, this.heartbeatInterval);
|
}
|
|
/**
|
* 停止心跳
|
*/
|
private stopHeartbeat() {
|
if (this.heartbeatTimer) {
|
clearInterval(this.heartbeatTimer);
|
this.heartbeatTimer = null;
|
}
|
}
|
|
/**
|
* 注册事件监听
|
* @param event 事件名称
|
* @param callback 回调函数
|
*/
|
public on(event: string, callback: MessageCallback) {
|
if (!this.listeners.has(event)) {
|
this.listeners.set(event, []);
|
}
|
this.listeners.get(event)?.push(callback);
|
}
|
|
/**
|
* 移除事件监听
|
* @param event 事件名称
|
* @param callback 回调函数
|
*/
|
public off(event: string, callback: MessageCallback) {
|
const callbacks = this.listeners.get(event);
|
if (callbacks) {
|
const index = callbacks.indexOf(callback);
|
if (index !== -1) {
|
callbacks.splice(index, 1);
|
}
|
}
|
}
|
|
/**
|
* 通知监听器
|
* @param event 事件名称
|
* @param data 数据
|
*/
|
private notifyListeners(event: string, data: any) {
|
const callbacks = this.listeners.get(event);
|
if (callbacks) {
|
callbacks.forEach(callback => {
|
try {
|
callback(data);
|
} catch (e) {
|
console.error(`WebSocket 事件 ${event} 监听器执行出错:`, e);
|
}
|
});
|
}
|
}
|
|
// 处理传入消息
|
private handleIncomingMessage(data: any, content: any) {
|
if (data?.type === 'ORDER_STATUS') {
|
const status = content?.status;
|
const orderId = content?.orderId;
|
if (typeof status === 'number') {
|
const mapped = this.mapOrderStatusToTripStatus(status);
|
if (typeof mapped.tripStatus === 'number') {
|
Taro.setStorageSync('tripStatus', mapped.tripStatus);
|
Taro.setStorageSync('orderId', orderId);
|
}
|
this.notifyListeners('order_status', { status, ...mapped, data, content });
|
}
|
}
|
if (data?.type === 'VEHICLE_LOCATION') {
|
const longitude = content?.lon;
|
const latitude = content?.lat;
|
if (typeof longitude === 'number' && typeof latitude === 'number') {
|
this.notifyListeners('vehicle_location', { longitude, latitude, data, content });
|
}
|
}
|
if (data?.type === 'ORDER_ROUTE') {
|
const routeType = content?.routeType;
|
const direction = Number(content?.direction);
|
// 2-12 新增
|
const duration = Number(JSON.parse(content?.route ?? '{}')?.duration ?? '0');
|
const distance = Number(JSON.parse(content?.route ?? '{}')?.distance ?? '0');
|
// 2-12 新增
|
const points = this.normalizeRoutePoints(content);
|
if (!Number.isNaN(duration)) {
|
Taro.setStorageSync('orderRouteDuration', Math.round(duration));
|
Taro.setStorageSync('orderRouteDurationUnit', 's');
|
}
|
if (!Number.isNaN(distance)) {
|
const distanceKm = Number((distance / 1000).toFixed(1));
|
Taro.setStorageSync('orderRouteDistance', distanceKm);
|
Taro.setStorageSync('orderRouteDistanceUnit', 'km');
|
}
|
if (points.length > 0) {
|
const cached = Taro.getStorageSync('orderRoutePoints') || {};
|
if (typeof routeType === 'number') {
|
cached[String(routeType)] = points;
|
Taro.setStorageSync('orderRouteLatestType', routeType);
|
} else {
|
cached.latest = points;
|
}
|
Taro.setStorageSync('orderRoutePoints', cached);
|
this.notifyListeners('order_route', { routeType, direction, points, data, content });
|
}
|
}
|
}
|
|
// 归一化路线点
|
private normalizeRoutePoints(content: any) {
|
const route = JSON.parse(content?.route ?? '{}');
|
const rawPoints = route?.polyline ?? [];
|
const arrPoint=JSON.parse(rawPoints);
|
if (!Array.isArray(arrPoint)) return [];
|
return arrPoint.map((item: any) => {
|
const latitude = item?.latitude;
|
const longitude = item?.longitude;
|
if (typeof latitude === 'number' && typeof longitude === 'number') {
|
return { latitude, longitude };
|
}
|
return null;
|
}).filter(Boolean);
|
}
|
|
// 映射订单状态到行程状态
|
private mapOrderStatusToTripStatus(status: number) {
|
switch (status) {
|
case 0:
|
return { tripStatus: 1, showCancelPopup: true };
|
case 1:
|
return { tripStatus: 3, showPassengerNotice: true };
|
case 2:
|
return { tripStatus: 3 };
|
case 3:
|
return { tripStatus: 4 };
|
case 4:
|
return { tripStatus: 5, showAnnouncementsPopup: true };
|
case 5:
|
case 6:
|
return { tripStatus: 6 };
|
case 7:
|
return { tripStatus: 7 };
|
case 8:
|
return { tripStatus: 8 };
|
case 9:
|
case 10:
|
return { tripStatus: 9 };
|
default:
|
return { tripStatus: undefined };
|
}
|
}
|
}
|
|
export default WebSocketManager.getInstance();
|
