Function forwardMessage

packages/message/server.ts:291–328

(
  prefix: string,
  path: string,
  receiverFrom: Server,
  senderTo: MessageSend,
  middleware?: ApiFunctionSync
)

// 转发消息
export function forwardMessage(
  prefix: string,
  path: string,
  receiverFrom: Server,
  senderTo: MessageSend,
  middleware?: ApiFunctionSync
) {
  const handler = async (params: any, fromCon: IGetSender): Promise<any> => {
    const fromConnect: MessageConnect | undefined = fromCon.getConnect();
    if (fromConnect) {
      const toCon: MessageConnect = await connect(senderTo, `${prefix}/${path}`, params);
      fromConnect.onMessage(toCon.sendMessage.bind(toCon));
      toCon.onMessage(fromConnect.sendMessage.bind(fromConnect));
      fromConnect.onDisconnect(toCon.disconnect.bind(toCon));
      toCon.onDisconnect(fromConnect.disconnect.bind(fromConnect));
      return undefined;
    } else {
      return sendMessage(senderTo, prefix + "/" + path, params);
    }
  };
  receiverFrom.on(path, (params, sender) => {
    if (middleware) {
      // 此处是为了处理CustomEventMessage的同步消息情况
      const resp = middleware(params, sender) as any;
      if (resp instanceof Promise) {
        return resp.then((data) => {
          if (data !== false) {
            return data;
          }
          return handler(params, sender);
        });
      } else if (resp !== false) {
        return resp;
      }
    }
    return handler(params, sender);
  });
}