实现ServerRpc和ClientRpc装饰器
This commit is contained in:
YHH
504
packages/network-shared/src/rpc/RpcCallProxy.ts
Normal file
504
packages/network-shared/src/rpc/RpcCallProxy.ts
Normal file
@@ -0,0 +1,504 @@
|
||||
import { createLogger } from '@esengine/ecs-framework';
|
||||
import { EventEmitter } from '../utils/EventEmitter';
|
||||
import {
|
||||
RpcCallRequest,
|
||||
RpcCallResponse,
|
||||
RpcError,
|
||||
RpcErrorType,
|
||||
RpcCallInfo,
|
||||
RpcCallStatus,
|
||||
RpcStats,
|
||||
RpcOptions,
|
||||
ClientRpcInvoker,
|
||||
ServerRpcInvoker
|
||||
} from '../types/RpcTypes';
|
||||
import { MessageType, RpcTarget } from '../types/NetworkTypes';
|
||||
|
||||
/**
|
||||
* 网络发送器接口
|
||||
*/
|
||||
export interface NetworkSender {
|
||||
sendMessage(message: object): Promise<void>;
|
||||
}
|
||||
|
||||
/**
|
||||
* RPC调用代理事件
|
||||
*/
|
||||
export interface RpcCallProxyEvents {
|
||||
callSent: (request: RpcCallRequest) => void;
|
||||
responseReceived: (response: RpcCallResponse) => void;
|
||||
callTimeout: (callId: string) => void;
|
||||
callFailed: (callId: string, error: RpcError) => void;
|
||||
retryAttempt: (callId: string, attempt: number) => void;
|
||||
}
|
||||
|
||||
/**
|
||||
* RPC调用代理配置
|
||||
*/
|
||||
export interface RpcCallProxyConfig {
|
||||
/** 默认超时时间(毫秒) */
|
||||
defaultTimeout: number;
|
||||
/** 最大重试次数 */
|
||||
maxRetries: number;
|
||||
/** 重试延迟基数(毫秒) */
|
||||
retryDelayBase: number;
|
||||
/** 重试延迟倍数 */
|
||||
retryDelayMultiplier: number;
|
||||
/** 最大重试延迟(毫秒) */
|
||||
maxRetryDelay: number;
|
||||
/** 是否启用离线队列 */
|
||||
enableOfflineQueue: boolean;
|
||||
/** 离线队列最大大小 */
|
||||
maxOfflineQueueSize: number;
|
||||
/** 调用ID生成器 */
|
||||
generateCallId?: () => string;
|
||||
}
|
||||
|
||||
/**
|
||||
* RPC调用代理
|
||||
* 负责发送RPC调用并处理响应
|
||||
*/
|
||||
export class RpcCallProxy extends EventEmitter {
|
||||
private logger = createLogger('RpcCallProxy');
|
||||
private config: RpcCallProxyConfig;
|
||||
private networkSender: NetworkSender;
|
||||
|
||||
/** 待处理的调用 */
|
||||
private pendingCalls = new Map<string, RpcCallInfo>();
|
||||
|
||||
/** 离线队列 */
|
||||
private offlineQueue: RpcCallRequest[] = [];
|
||||
|
||||
/** 是否在线 */
|
||||
private isOnline = true;
|
||||
|
||||
/** 统计信息 */
|
||||
private stats: RpcStats = {
|
||||
totalCalls: 0,
|
||||
successfulCalls: 0,
|
||||
failedCalls: 0,
|
||||
averageResponseTime: 0,
|
||||
pendingCalls: 0,
|
||||
timeoutCalls: 0,
|
||||
retryCount: 0,
|
||||
lastUpdated: Date.now()
|
||||
};
|
||||
|
||||
/** 重试定时器 */
|
||||
private retryTimers = new Map<string, ReturnType<typeof setTimeout>>();
|
||||
|
||||
constructor(
|
||||
networkSender: NetworkSender,
|
||||
config: Partial<RpcCallProxyConfig> = {}
|
||||
) {
|
||||
super();
|
||||
|
||||
this.networkSender = networkSender;
|
||||
this.config = {
|
||||
defaultTimeout: 30000,
|
||||
maxRetries: 3,
|
||||
retryDelayBase: 1000,
|
||||
retryDelayMultiplier: 2,
|
||||
maxRetryDelay: 10000,
|
||||
enableOfflineQueue: true,
|
||||
maxOfflineQueueSize: 100,
|
||||
generateCallId: () => `rpc_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`,
|
||||
...config
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 客户端RPC调用器
|
||||
*/
|
||||
public get clientRpc(): ClientRpcInvoker {
|
||||
return <TArgs extends readonly unknown[], TReturn>(
|
||||
methodName: string,
|
||||
args: TArgs,
|
||||
options?: Partial<RpcOptions>
|
||||
): Promise<TReturn> => {
|
||||
return this.call(methodName, args, options);
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 服务端RPC调用器（用于服务端调用客户端）
|
||||
*/
|
||||
public get serverRpc(): ServerRpcInvoker {
|
||||
return <TArgs extends readonly unknown[], TReturn>(
|
||||
clientId: string,
|
||||
methodName: string,
|
||||
args: TArgs,
|
||||
options?: Partial<RpcOptions>
|
||||
): Promise<TReturn> => {
|
||||
const callOptions = {
|
||||
...options,
|
||||
target: RpcTarget.Client
|
||||
};
|
||||
return this.call(methodName, args, callOptions, clientId);
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 发起RPC调用
|
||||
*/
|
||||
public async call<TArgs extends readonly unknown[], TReturn>(
|
||||
methodName: string,
|
||||
args: TArgs,
|
||||
options: Partial<RpcOptions> = {},
|
||||
targetId?: string
|
||||
): Promise<TReturn> {
|
||||
const callId = this.config.generateCallId!();
|
||||
const timeout = options.timeout || this.config.defaultTimeout;
|
||||
|
||||
const request: RpcCallRequest<TArgs> = {
|
||||
callId,
|
||||
methodName,
|
||||
args,
|
||||
senderId: 'client', // 这应该从认证系统获取
|
||||
targetId,
|
||||
timestamp: Date.now(),
|
||||
options: {
|
||||
reliable: true,
|
||||
priority: 5,
|
||||
timeout,
|
||||
...options
|
||||
}
|
||||
};
|
||||
|
||||
// 创建Promise和调用信息
|
||||
return new Promise<TReturn>((resolve, reject) => {
|
||||
const callInfo: RpcCallInfo<TArgs> = {
|
||||
request,
|
||||
status: RpcCallStatus.PENDING,
|
||||
resolve: resolve as (value: unknown) => void,
|
||||
reject: (reason: RpcError) => reject(reason),
|
||||
retryCount: 0,
|
||||
createdAt: Date.now()
|
||||
};
|
||||
|
||||
this.pendingCalls.set(callId, callInfo);
|
||||
this.stats.pendingCalls++;
|
||||
this.stats.totalCalls++;
|
||||
|
||||
// 设置超时
|
||||
setTimeout(() => {
|
||||
this.handleTimeout(callId);
|
||||
}, timeout);
|
||||
|
||||
// 发送调用
|
||||
this.sendCall(callInfo);
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理RPC响应
|
||||
*/
|
||||
public handleResponse(response: RpcCallResponse): void {
|
||||
const callInfo = this.pendingCalls.get(response.callId);
|
||||
if (!callInfo) {
|
||||
this.logger.warn(`收到未知调用的响应: ${response.callId}`);
|
||||
return;
|
||||
}
|
||||
|
||||
// 清理定时器
|
||||
const timer = this.retryTimers.get(response.callId);
|
||||
if (timer) {
|
||||
clearTimeout(timer);
|
||||
this.retryTimers.delete(response.callId);
|
||||
}
|
||||
|
||||
// 更新状态
|
||||
callInfo.status = response.success ? RpcCallStatus.COMPLETED : RpcCallStatus.FAILED;
|
||||
callInfo.completedAt = Date.now();
|
||||
|
||||
// 更新统计
|
||||
if (response.success) {
|
||||
this.stats.successfulCalls++;
|
||||
this.updateAverageResponseTime(response.duration);
|
||||
} else {
|
||||
this.stats.failedCalls++;
|
||||
}
|
||||
|
||||
this.stats.pendingCalls--;
|
||||
|
||||
// 处理结果
|
||||
if (response.success) {
|
||||
callInfo.resolve!(response.result);
|
||||
} else {
|
||||
callInfo.reject!(response.error!);
|
||||
this.emit('callFailed', response.callId, response.error!);
|
||||
}
|
||||
|
||||
// 清理
|
||||
this.pendingCalls.delete(response.callId);
|
||||
this.emit('responseReceived', response);
|
||||
}
|
||||
|
||||
/**
|
||||
* 设置网络状态
|
||||
*/
|
||||
public setOnlineStatus(online: boolean): void {
|
||||
const wasOnline = this.isOnline;
|
||||
this.isOnline = online;
|
||||
|
||||
if (online && !wasOnline) {
|
||||
// 从离线状态恢复，处理离线队列
|
||||
this.processOfflineQueue();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 取消RPC调用
|
||||
*/
|
||||
public cancelCall(callId: string): boolean {
|
||||
const callInfo = this.pendingCalls.get(callId);
|
||||
if (!callInfo) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// 清理定时器
|
||||
const timer = this.retryTimers.get(callId);
|
||||
if (timer) {
|
||||
clearTimeout(timer);
|
||||
this.retryTimers.delete(callId);
|
||||
}
|
||||
|
||||
// 更新状态
|
||||
callInfo.status = RpcCallStatus.CANCELLED;
|
||||
|
||||
// 拒绝Promise
|
||||
callInfo.reject!({
|
||||
type: RpcErrorType.CLIENT_ERROR,
|
||||
message: '调用被取消'
|
||||
});
|
||||
|
||||
// 清理
|
||||
this.pendingCalls.delete(callId);
|
||||
this.stats.pendingCalls--;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取统计信息
|
||||
*/
|
||||
public getStats(): RpcStats {
|
||||
this.stats.lastUpdated = Date.now();
|
||||
return { ...this.stats };
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取待处理的调用
|
||||
*/
|
||||
public getPendingCalls(): RpcCallInfo[] {
|
||||
return Array.from(this.pendingCalls.values());
|
||||
}
|
||||
|
||||
/**
|
||||
* 重置统计信息
|
||||
*/
|
||||
public resetStats(): void {
|
||||
this.stats = {
|
||||
totalCalls: 0,
|
||||
successfulCalls: 0,
|
||||
failedCalls: 0,
|
||||
averageResponseTime: 0,
|
||||
pendingCalls: this.pendingCalls.size,
|
||||
timeoutCalls: 0,
|
||||
retryCount: 0,
|
||||
lastUpdated: Date.now()
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新配置
|
||||
*/
|
||||
public updateConfig(newConfig: Partial<RpcCallProxyConfig>): void {
|
||||
Object.assign(this.config, newConfig);
|
||||
}
|
||||
|
||||
/**
|
||||
* 销毁代理
|
||||
*/
|
||||
public destroy(): void {
|
||||
// 取消所有待处理的调用
|
||||
const pendingCallIds = Array.from(this.pendingCalls.keys());
|
||||
for (const callId of pendingCallIds) {
|
||||
this.cancelCall(callId);
|
||||
}
|
||||
|
||||
// 清理定时器
|
||||
for (const timer of this.retryTimers.values()) {
|
||||
clearTimeout(timer);
|
||||
}
|
||||
this.retryTimers.clear();
|
||||
|
||||
// 清理队列
|
||||
this.offlineQueue.length = 0;
|
||||
|
||||
this.removeAllListeners();
|
||||
}
|
||||
|
||||
/**
|
||||
* 发送调用
|
||||
*/
|
||||
private async sendCall<T extends readonly unknown[]>(callInfo: RpcCallInfo<T>): Promise<void> {
|
||||
try {
|
||||
// 检查网络状态
|
||||
if (!this.isOnline) {
|
||||
if (this.config.enableOfflineQueue) {
|
||||
this.addToOfflineQueue(callInfo.request);
|
||||
} else {
|
||||
throw new Error('网络不可用');
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
// 构建网络消息
|
||||
const message = {
|
||||
type: MessageType.RPC_CALL,
|
||||
messageId: callInfo.request.callId,
|
||||
timestamp: Date.now(),
|
||||
senderId: callInfo.request.senderId,
|
||||
data: callInfo.request,
|
||||
reliable: callInfo.request.options.reliable,
|
||||
priority: callInfo.request.options.priority
|
||||
};
|
||||
|
||||
// 发送消息
|
||||
await this.networkSender.sendMessage(message);
|
||||
|
||||
callInfo.status = RpcCallStatus.SENT;
|
||||
callInfo.sentAt = Date.now();
|
||||
|
||||
this.emit('callSent', callInfo.request);
|
||||
|
||||
} catch (error) {
|
||||
this.logger.error(`发送RPC调用失败: ${callInfo.request.methodName}`, error);
|
||||
|
||||
// 检查是否可以重试
|
||||
if (callInfo.retryCount < this.config.maxRetries) {
|
||||
this.scheduleRetry(callInfo);
|
||||
} else {
|
||||
this.handleCallFailure(callInfo, {
|
||||
type: RpcErrorType.NETWORK_ERROR,
|
||||
message: String(error)
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理超时
|
||||
*/
|
||||
private handleTimeout(callId: string): void {
|
||||
const callInfo = this.pendingCalls.get(callId);
|
||||
if (!callInfo || callInfo.status === RpcCallStatus.COMPLETED) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 检查是否可以重试
|
||||
if (callInfo.retryCount < this.config.maxRetries) {
|
||||
this.scheduleRetry(callInfo);
|
||||
} else {
|
||||
this.stats.timeoutCalls++;
|
||||
this.handleCallFailure(callInfo, {
|
||||
type: RpcErrorType.TIMEOUT,
|
||||
message: `调用超时: ${callInfo.request.options.timeout}ms`
|
||||
});
|
||||
this.emit('callTimeout', callId);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 安排重试
|
||||
*/
|
||||
private scheduleRetry<T extends readonly unknown[]>(callInfo: RpcCallInfo<T>): void {
|
||||
callInfo.retryCount++;
|
||||
this.stats.retryCount++;
|
||||
|
||||
// 计算延迟时间（指数退避）
|
||||
const baseDelay = this.config.retryDelayBase * Math.pow(this.config.retryDelayMultiplier, callInfo.retryCount - 1);
|
||||
const delay = Math.min(baseDelay, this.config.maxRetryDelay);
|
||||
|
||||
callInfo.nextRetryTime = Date.now() + delay;
|
||||
|
||||
this.emit('retryAttempt', callInfo.request.callId, callInfo.retryCount);
|
||||
|
||||
const timer = setTimeout(() => {
|
||||
this.retryTimers.delete(callInfo.request.callId);
|
||||
this.sendCall(callInfo);
|
||||
}, delay);
|
||||
|
||||
this.retryTimers.set(callInfo.request.callId, timer);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理调用失败
|
||||
*/
|
||||
private handleCallFailure<T extends readonly unknown[]>(callInfo: RpcCallInfo<T>, error: RpcError): void {
|
||||
callInfo.status = RpcCallStatus.FAILED;
|
||||
callInfo.completedAt = Date.now();
|
||||
|
||||
callInfo.reject!(error);
|
||||
|
||||
this.pendingCalls.delete(callInfo.request.callId);
|
||||
this.stats.pendingCalls--;
|
||||
this.stats.failedCalls++;
|
||||
|
||||
this.emit('callFailed', callInfo.request.callId, error);
|
||||
}
|
||||
|
||||
/**
|
||||
* 添加到离线队列
|
||||
*/
|
||||
private addToOfflineQueue<T extends readonly unknown[]>(request: RpcCallRequest<T>): void {
|
||||
if (this.offlineQueue.length >= this.config.maxOfflineQueueSize) {
|
||||
// 移除最旧的请求
|
||||
this.offlineQueue.shift();
|
||||
}
|
||||
|
||||
this.offlineQueue.push(request);
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理离线队列
|
||||
*/
|
||||
private async processOfflineQueue(): Promise<void> {
|
||||
if (!this.isOnline || this.offlineQueue.length === 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
const queue = [...this.offlineQueue];
|
||||
this.offlineQueue.length = 0;
|
||||
|
||||
for (const request of queue) {
|
||||
try {
|
||||
// 重新创建调用信息
|
||||
const callInfo: RpcCallInfo = {
|
||||
request,
|
||||
status: RpcCallStatus.PENDING,
|
||||
retryCount: 0,
|
||||
createdAt: Date.now()
|
||||
};
|
||||
|
||||
this.pendingCalls.set(request.callId, callInfo);
|
||||
await this.sendCall(callInfo);
|
||||
|
||||
} catch (error) {
|
||||
this.logger.error(`处理离线队列失败: ${request.methodName}`, error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 更新平均响应时间
|
||||
*/
|
||||
private updateAverageResponseTime(responseTime: number): void {
|
||||
const totalResponses = this.stats.successfulCalls;
|
||||
const currentAverage = this.stats.averageResponseTime;
|
||||
|
||||
this.stats.averageResponseTime =
|
||||
(currentAverage * (totalResponses - 1) + responseTime) / totalResponses;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user