From 33af740985089b4eea384129cd949f8e302eb459 Mon Sep 17 00:00:00 2001 From: Francis Lam Date: Tue, 4 Mar 2014 21:36:48 -0500 Subject: [PATCH] Added missing notificationMsg types to notificationHandler --- rpcwebsocket.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/rpcwebsocket.go b/rpcwebsocket.go index b4e4d9ee..e9c805bc 100644 --- a/rpcwebsocket.go +++ b/rpcwebsocket.go @@ -291,6 +291,10 @@ out: wsc := (*wsClient)(n) blockNotifications[wsc.quit] = wsc + case *notificationUnregisterBlocks: + wsc := (*wsClient)(n) + delete(blockNotifications, wsc.quit) + case *notificationRegisterClient: wsc := (*wsClient)(n) clients[wsc.quit] = wsc @@ -322,6 +326,14 @@ out: case *notificationUnregisterAddr: m.removeAddrRequest(watchedAddrs, n.wsc, n.addr) + case *notificationRegisterNewMempoolTxs: + wsc := (*wsClient)(n) + txNotifications[wsc.quit] = wsc + + case *notificationUnregisterNewMempoolTxs: + wsc := (*wsClient)(n) + delete(txNotifications, wsc.quit) + default: rpcsLog.Warn("Unhandled notification type") }