AngularJS websocket factory

gistfile1.js

app.factory('socket', function ($rootScope) {
  var socket = new WebSocket("ws://localhost:8000/data");
  return {
    onopen: function (callback) {
      socket.onopen = function () {  
        $rootScope.$apply(function () {
          callback.apply(socket);
        });
      };
    },
    onmessage: function (callback) {
      socket.onmessage = function (e) {
        var data = e.data;
        $rootScope.$apply(function () {
          if (callback) {
            callback.apply(socket, [data]);
          }
        });
      }
    },
    onerror: function (callback) {
      socket.onerror = function (error) {
        $rootScope.$apply(function () {
          if (callback) {
            callback.apply(socket, [error]);
          }
        });
      };
    },
    send: function (message) {
      socket.send(message);
    }
  };
});

/*
Usage:

app.controller('AppController', ['$rootScope', '$scope', '$log', 'socket', function($rootScope, $scope, $location, $log, socket) {
  socket.onopen(function () {
  });
  
  socket.onerror(function (error) {
    $log.error(error);
  });
  
  socket.onmessage(function (message) {
    $log.info(message);  
  });
  
  socket.send('message);
});
*/