Added isConnected function

Author: WrathChaos

StompClientLib/Classes/StompClientLib.swift

@@ -72,8 +72,8 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
     var sessionId: String?
     var delegate: StompClientLibDelegate?
     var connectionHeaders: [String: String]?
+    public var connection: Bool = false
     public var certificateCheckEnabled = true
-    
     private var urlRequest: NSURLRequest?
 
     public func sendJSONForDict(dict: AnyObject, toDestination destination: String) {
@@ -98,6 +98,7 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
     public func openSocketWithURLRequest(request: NSURLRequest, delegate: StompClientLibDelegate, connectionHeaders: [String: String]?) {
         self.connectionHeaders = connectionHeaders
         openSocketWithURLRequest(request: request, delegate: delegate)
+        self.connection = true
     }
 
     private func openSocket() {
@@ -339,7 +340,12 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
         sendFrame(command: StompCommands.commandSend, header: headersToSend, body: message as AnyObject)
     }
 
+    public func isConnected() -> Bool{
+        return connection
+    }
+    
     public func subscribe(destination: String) {
+        connection = true
         subscribeToDestination(destination: destination, ackMode: .AutoMode)
     }
 
@@ -366,6 +372,7 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
     }
 
     public func unsubscribe(destination: String) {
+        connection = false
         var headerToSend = [String: String]()
         headerToSend[StompCommands.commandHeaderDestinationId] = destination
         sendFrame(command: StompCommands.commandUnsubscribe, header: headerToSend, body: nil)
@@ -403,6 +410,7 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
     }
 
     public func disconnect() {
+        connection = false
         var headerToSend = [String: String]()
         headerToSend[StompCommands.commandDisconnect] = String(Int(NSDate().timeIntervalSince1970))
         sendFrame(command: StompCommands.commandDisconnect, header: headerToSend, body: nil)