Adding functionality to stop the reconnect timer

Author: JKim0119

StompClientLib/Classes/StompClientLib.swift

@@ -80,6 +80,8 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
     public var certificateCheckEnabled = true
     private var urlRequest: NSURLRequest?
 
+    private var reconnectTimer : Timer?
+    
     public func sendJSONForDict(dict: AnyObject, toDestination destination: String) {
         do {
             let theJSONData = try JSONSerialization.data(withJSONObject: dict, options: JSONSerialization.WritingOptions())
@@ -456,7 +458,7 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
     // TODO: MAKE A VARIABLE TO CHECK RECONNECT OPTION IS AVAILABLE OR NOT
     public func reconnect(request: NSURLRequest, delegate: StompClientLibDelegate, connectionHeaders: [String: String] = [String: String](), time: Double = 1.0, exponentialBackoff: Bool = true){
         if #available(iOS 10.0, *) {
-            Timer.scheduledTimer(withTimeInterval: time, repeats: true, block: { _ in
+            reconnectTimer = Timer.scheduledTimer(withTimeInterval: time, repeats: true, block: { _ in
                 self.reconnectLogic(request: request, delegate: delegate
                     , connectionHeaders: connectionHeaders)
             })
@@ -478,6 +480,17 @@ public class StompClientLib: NSObject, SRWebSocketDelegate {
         }
     }
 
+    public func stopReconnect() {
+        reconnectTimer?.invalidate()
+        reconnectTimer = nil
+    }
+    
+    public func stopReconnectAndDisconnect() {
+        reconnectTimer?.invalidate()
+        reconnectTimer = nil
+        self.disconnect()
+    }
+    
     private func checkConnectionHeader(connectionHeaders: [String: String] = [String: String]()) -> Bool{
         if (connectionHeaders.isEmpty){
             // No connection header