add rpc system to improve ipc
This commit is contained in:
4
events/EventSystem/CallbackStore.ts
Normal file
4
events/EventSystem/CallbackStore.ts
Normal file
@@ -0,0 +1,4 @@
|
||||
export interface CallbackStore {
|
||||
wrappedCallback: any;
|
||||
callback: any;
|
||||
}
|
||||
12
events/EventSystem/EventBus.ts
Normal file
12
events/EventSystem/EventBus.ts
Normal file
@@ -0,0 +1,12 @@
|
||||
import { ipcMain, ipcRenderer } from 'electron'
|
||||
|
||||
import { IpcMainEventBus } from './IpcMainEventBus'
|
||||
import { IpcRendererEventBus } from './IpcRendererEventBus'
|
||||
import { Rpc } from './Rpc'
|
||||
|
||||
export const rendererEvents = new IpcRendererEventBus(ipcRenderer)
|
||||
export const backendEvents = new IpcMainEventBus(ipcMain)
|
||||
|
||||
// Preferred way to communicate typesafe
|
||||
export const rendererRpc = new Rpc(rendererEvents)
|
||||
export const backendRpc = new Rpc(backendEvents)
|
||||
8
events/EventSystem/EventBusInterface.ts
Normal file
8
events/EventSystem/EventBusInterface.ts
Normal file
@@ -0,0 +1,8 @@
|
||||
import { Event } from '../Events';
|
||||
|
||||
export interface EventBusInterface {
|
||||
subscribe<MessageType>(event: Event<MessageType>, callback: (msg: MessageType) => void): void;
|
||||
unsubscribeAll<MessageType>(event: Event<MessageType>): void;
|
||||
emit<MessageType>(event: Event<MessageType>, msg: MessageType): void;
|
||||
unsubscribe<MessageType>(event: Event<MessageType>, callback: any): void;
|
||||
}
|
||||
42
events/EventSystem/EventDispatcher.ts
Normal file
42
events/EventSystem/EventDispatcher.ts
Normal file
@@ -0,0 +1,42 @@
|
||||
import { EventEmitter } from 'events'
|
||||
|
||||
interface CallbackStore {
|
||||
wrappedCallback: any
|
||||
callback: any
|
||||
}
|
||||
|
||||
export class EventDispatcher<Message> {
|
||||
private emitter = new EventEmitter()
|
||||
private callbacks: Array<CallbackStore> = []
|
||||
|
||||
public dispatch(msg: Message) {
|
||||
this.emitter.emit('event', msg)
|
||||
}
|
||||
|
||||
public subscribe(callback: (msg: Message) => void) {
|
||||
const wrappedCallback = (msg: Message) => {
|
||||
callback(msg)
|
||||
}
|
||||
this.emitter.on('event', wrappedCallback)
|
||||
|
||||
this.callbacks.push({
|
||||
callback,
|
||||
wrappedCallback,
|
||||
})
|
||||
}
|
||||
|
||||
public unsubscribe(callback: (msg: Message) => void) {
|
||||
const item = this.callbacks.find(store => store.callback === callback)
|
||||
if (!item) {
|
||||
return
|
||||
}
|
||||
|
||||
this.emitter.removeListener('event', item.wrappedCallback)
|
||||
this.callbacks = this.callbacks.filter(a => a !== item)
|
||||
}
|
||||
|
||||
public removeAllListeners() {
|
||||
this.emitter.removeAllListeners()
|
||||
this.callbacks = []
|
||||
}
|
||||
}
|
||||
34
events/EventSystem/IpcMainEventBus.ts
Normal file
34
events/EventSystem/IpcMainEventBus.ts
Normal file
@@ -0,0 +1,34 @@
|
||||
import { IpcMain } from 'electron';
|
||||
import { Event } from '../Events';
|
||||
import { EventBusInterface } from "./EventBusInterface";
|
||||
|
||||
export class IpcMainEventBus implements EventBusInterface {
|
||||
private ipc: IpcMain;
|
||||
private client: any;
|
||||
constructor(ipc: IpcMain) {
|
||||
this.ipc = ipc;
|
||||
}
|
||||
|
||||
public subscribe<MessageType>(subscribeEvent: Event<MessageType>, callback: (msg: MessageType) => void) {
|
||||
console.log('subscribing', subscribeEvent.topic);
|
||||
this.ipc.on(subscribeEvent.topic, (event: any, arg: any) => {
|
||||
this.client = event.sender;
|
||||
callback(arg);
|
||||
});
|
||||
}
|
||||
|
||||
public unsubscribeAll<MessageType>(event: Event<MessageType>) {
|
||||
console.log('unsubscribeAll', event.topic);
|
||||
this.ipc.removeAllListeners(event.topic);
|
||||
}
|
||||
|
||||
public unsubscribe<MessageType>(event: Event<MessageType>, callback: any) {
|
||||
throw new Error('Not implemented'); // Todo: implement
|
||||
}
|
||||
|
||||
public emit<MessageType>(event: Event<MessageType>, msg: MessageType) {
|
||||
if (!this.client.isDestroyed()) {
|
||||
this.client.send(event.topic, msg);
|
||||
}
|
||||
}
|
||||
}
|
||||
42
events/EventSystem/IpcRendererEventBus.ts
Normal file
42
events/EventSystem/IpcRendererEventBus.ts
Normal file
@@ -0,0 +1,42 @@
|
||||
import { CallbackStore } from "./CallbackStore"
|
||||
import { EventBusInterface } from "./EventBusInterface"
|
||||
import { Event } from '../Events'
|
||||
import { IpcRenderer } from 'electron'
|
||||
|
||||
export class IpcRendererEventBus implements EventBusInterface {
|
||||
private ipc: IpcRenderer
|
||||
private callbacks: Array<CallbackStore> = []
|
||||
|
||||
constructor(ipc: IpcRenderer) {
|
||||
this.ipc = ipc
|
||||
}
|
||||
|
||||
public subscribe<MessageType>(event: Event<MessageType>, callback: (msg: MessageType) => void) {
|
||||
const wrappedCallback = (_: any, arg: any) => {
|
||||
callback(arg)
|
||||
}
|
||||
console.log("subscribing", event.topic)
|
||||
this.ipc.on(event.topic, wrappedCallback)
|
||||
this.callbacks.push({
|
||||
callback,
|
||||
wrappedCallback,
|
||||
})
|
||||
}
|
||||
|
||||
public unsubscribeAll<MessageType>(event: Event<MessageType>) {
|
||||
this.ipc.removeAllListeners(event.topic)
|
||||
}
|
||||
|
||||
public unsubscribe<MessageType>(event: Event<MessageType>, callback: any) {
|
||||
const item = this.callbacks.find(store => store.callback === callback)
|
||||
if (!item) {
|
||||
return
|
||||
}
|
||||
this.ipc.removeListener(event.topic, item.wrappedCallback)
|
||||
this.callbacks = this.callbacks.filter(a => a !== item)
|
||||
}
|
||||
|
||||
public emit<MessageType>(event: Event<MessageType>, msg: MessageType) {
|
||||
this.ipc.send(event.topic, msg)
|
||||
}
|
||||
}
|
||||
54
events/EventSystem/Rpc.ts
Normal file
54
events/EventSystem/Rpc.ts
Normal file
@@ -0,0 +1,54 @@
|
||||
import { Event } from '../Events';
|
||||
import { EventBusInterface } from './EventBusInterface'
|
||||
import { v4 } from 'uuid';
|
||||
|
||||
export type RpcEvent<RequstType, ResponseType> = {
|
||||
topic: string;
|
||||
};
|
||||
|
||||
export class Rpc {
|
||||
constructor(private participant: EventBusInterface) { }
|
||||
|
||||
async call<RpcRequest, RpcResponse>(event: RpcEvent<RpcRequest, RpcResponse>, request: RpcRequest, timeout: number = 0): Promise<RpcResponse> {
|
||||
return new Promise((resolve, reject) => {
|
||||
let id = v4();
|
||||
|
||||
let responseEvent: Event<any> = { topic: `${event.topic}/response` };
|
||||
let requestEvent: Event<any> = { topic: `${event.topic}/request` };
|
||||
let callback = (result: { id: string; payload: RpcResponse; error: unknown }) => {
|
||||
this.participant.unsubscribe(responseEvent as any, callback);
|
||||
if (result.id === id) {
|
||||
if (result.error) {
|
||||
reject(result.error)
|
||||
} else {
|
||||
resolve(result.payload);
|
||||
}
|
||||
}
|
||||
console.log("received", result)
|
||||
};
|
||||
this.participant.subscribe(responseEvent, callback);
|
||||
this.participant.emit(requestEvent, { id, payload: request });
|
||||
|
||||
if (timeout > 0) {
|
||||
setTimeout(() => {
|
||||
reject(new Error(`Did not respond to ${event.topic} within ${timeout}ms`))
|
||||
this.participant.unsubscribe(responseEvent as any, callback);
|
||||
}, 10000)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
async on<RpcRequest, RpcResponse>(event: RpcEvent<RpcRequest, RpcResponse>, handler: (request: RpcRequest) => Promise<RpcResponse>) {
|
||||
this.participant.subscribe<RpcRequest>({ topic: `${event.topic}/request` } as RpcEvent<any, any>, async (request) => {
|
||||
let payload;
|
||||
let error;
|
||||
try {
|
||||
payload = await handler((request as any).payload);
|
||||
} catch (e) {
|
||||
error = e
|
||||
}
|
||||
console.log("Responding with", payload, error)
|
||||
this.participant.emit({ topic: `${event.topic}/response` }, { id: (request as any).id, payload, error });
|
||||
});
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user