Move socket out of global scope into STOMP object.

Author: Tony Young

lib/stomp.js

@@ -26,7 +26,6 @@ var net = require('net'),
     stomp_utils = require('./stomp-utils'),
     exceptions = require('./stomp-exceptions'),
     utils = new stomp_utils.StompUtils(),
-    socket = net.Stream(),
     log = null;
 
 function parse_command(data) {
@@ -92,6 +91,7 @@ function parse_frame(chunk) {
 };
 
 function _connect(stomp) {
+    var socket = stomp.socket = net.Stream();
     log = stomp.log;
     log.debug('Connecting to ' + stomp.host + ':' + stomp.port);
     socket.connect(stomp.port, stomp.host);
@@ -100,10 +100,10 @@ function _connect(stomp) {
         log.debug('Connected to socket');
         if (utils.really_defined(stomp.login)
                 && utils.really_defined(stomp.passcode)) {
-            stomp_connect({login: stomp.login, passcode: stomp.passcode});
+            stomp_connect(stomp, {login: stomp.login, passcode: stomp.passcode});
         }
         else {
-            stomp_connect();
+            stomp_connect(stomp);
         }
     });
 
@@ -141,7 +141,7 @@ function _connect(stomp) {
     });
 };
 
-function stomp_connect(headers) {
+function stomp_connect(stomp, headers) {
     var _frame = new frame.Frame(),
         args = {},
         headers = headers || {};
@@ -150,18 +150,19 @@ function stomp_connect(headers) {
     args['headers'] = headers;
 
     var frame_to_send = _frame.build_frame(args);
-    send_frame(frame_to_send);
+    send_frame(stomp, frame_to_send);
     log.debug('Connected to STOMP');
 };
 
-function _disconnect() {
+function _disconnect(stomp) {
+    var socket = stomp.socket;
     socket.end();
     if (socket.readyState == 'readOnly')
         socket.destroy();
     log.debug('disconnect called');
 };
 
-function send_command(command, headers, body, want_receipt) {
+function send_command(stomp, command, headers, body, want_receipt) {
     var want_receipt = want_receipt || false;
     if (!utils.really_defined(headers))
         headers = {};
@@ -174,12 +175,13 @@ function send_command(command, headers, body, want_receipt) {
 
     var _frame = new frame.Frame();
     var this_frame = _frame.build_frame(args, want_receipt);
-    send_frame(this_frame);
+    send_frame(stomp, this_frame);
 
     return this_frame;
 };
 
-function send_frame(_frame) {
+function send_frame(stomp, _frame) {
+    var socket = stomp.socket;
     var frame_str = _frame.as_string();
 
     if (socket.write(frame_str) === false) {
@@ -246,7 +248,7 @@ Stomp.prototype.handle_new_frame = function(this_frame) {
  * Disconnect from STOMP broker
  */
 Stomp.prototype.disconnect = function() {
-    _disconnect();
+    _disconnect(this);
 }
 
 /**
@@ -257,7 +259,7 @@ Stomp.prototype.subscribe = function(headers) {
     var self = this;
     destination = headers['destination'];
     headers['session'] = self.session;
-    send_command('SUBSCRIBE', headers);
+    send_command(this, 'SUBSCRIBE', headers);
     self._subscribed_to[destination] = true;
     self.log.debug('subscribed to: ' + destination + ' with headers ' + sys.inspect(headers));
 };
@@ -270,7 +272,7 @@ Stomp.prototype.unsubscribe = function(headers) {
     var self = this;
     destination = headers['destination'];
     headers['session'] = self.session;
-    send_command('UNSUBSCRIBE', headers);
+    send_command(this, 'UNSUBSCRIBE', headers);
     self._subscribed_to[destination] = false;
     self.log.debug('no longer subscribed to: ' + destination);
 };
@@ -281,7 +283,7 @@ Stomp.prototype.unsubscribe = function(headers) {
  */
 Stomp.prototype.ack = function(message_id) {
     var self = this;
-    send_command('ACK', {'message-id': message_id});
+    send_command(this, 'ACK', {'message-id': message_id});
     self.log.debug('acknowledged message: ' + message_id);
 };
 
@@ -292,7 +294,7 @@ Stomp.prototype.ack = function(message_id) {
 Stomp.prototype.begin = function() {
     var self = this;
     transaction_id = Math.floor(Math.random()*99999999999).toString();
-    send_command('BEGIN', {'transaction': transaction_id});
+    send_command(this, 'BEGIN', {'transaction': transaction_id});
     self.log.debug('begin transaction: ' + transaction_id);
     return transaction_id;
 };
@@ -303,7 +305,7 @@ Stomp.prototype.begin = function() {
  */
 Stomp.prototype.commit = function(transaction_id) {
     var self = this;
-    send_command('COMMIT', {'transaction': transaction_id});
+    send_command(this, 'COMMIT', {'transaction': transaction_id});
     self.log.debug('commit transaction: ' + transaction_id);
 };
 
@@ -313,7 +315,7 @@ Stomp.prototype.commit = function(transaction_id) {
  */
 Stomp.prototype.abort = function(transaction_id) {
     var self = this;
-    send_command('ABORT', {'transaction': transaction_id});
+    send_command(this, 'ABORT', {'transaction': transaction_id});
     self.log.debug('abort transaction: ' + transaction_id);
 };
 
@@ -328,7 +330,7 @@ Stomp.prototype.send = function(headers, want_receipt) {
     destination = headers['destination'];
     body = headers['body'] || null;
     headers['session'] = self.session;
-    return send_command('SEND', headers, body, want_receipt)
+    return send_command(this, 'SEND', headers, body, want_receipt)
 };
 
 module.exports.Stomp = Stomp;