improve notification broadcast

10 years ago · 0376768144
1 changed files with 4 additions and 1 deletions
--- a/lib/wsapp.js
+++ b/lib/wsapp.js
@ -29,7 +29,10 @@ WsApp.start = function(server) {
  var io = require('socket.io')(server);

  WalletService.onNotification(function(serviceInstance, args) {
-    io.to(serviceInstance.walletId).emit('notification', args);
+    var room = serviceInstance.walletId || args.walletId;
+    if (room) {
+      io.to(room).emit('notification', args);
+    }
  });

  io.on('connection', function(socket) {