Added missing notificationMsg types to notificationHandler

btcsuite · Mar 5, 2014 · 33af740 · 33af740
1 parent b452acd
commit 33af740
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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")
 			}