typed messages and handlers

1 files changed, 2 insertions, 2 deletions

diff --git a/src/server/session/agents/message_router.ts b/src/server/session/agents/message_router.ts
index 5848e27ab..707f771d9 100644
--- a/src/server/session/agents/message_router.ts
+++ b/src/server/session/agents/message_router.ts
@@ -8,7 +8,7 @@ export default abstract class MessageRouter {
      * Add a listener at this message. When the monitor process
      * receives a message, it will invoke all registered functions.
      */
-    public addMessageListener = (name: string, handler: MessageHandler, exclusive = false) => {
+    public on = (name: string, handler: MessageHandler, exclusive = false) => {
         const handlers = this.onMessage[name];
         if (exclusive || !handlers) {
             this.onMessage[name] = [handler];
@@ -20,7 +20,7 @@ export default abstract class MessageRouter {
     /**
      * Unregister a given listener at this message.
      */
-    public removeMessageListener = (name: string, handler: MessageHandler) => {
+    public off = (name: string, handler: MessageHandler) => {
         const handlers = this.onMessage[name];
         if (handlers) {
             const index = handlers.indexOf(handler);