Merge pull request #84 from soledue/master

fix delegate
WrathChaos · Mar 8, 2020 · 7039010 · 7039010
2 parents 73411fa + 59a99b6
commit 7039010
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/StompClientLib/Classes/StompClientLib.swift b/StompClientLib/Classes/StompClientLib.swift
@@ -59,7 +59,7 @@ public enum StompAckMode {
 }
 
 // Fundamental Protocols
-public protocol StompClientLibDelegate {
+public protocol StompClientLibDelegate: class {
     func stompClient(client: StompClientLib!, didReceiveMessageWithJSONBody jsonBody: AnyObject?, akaStringBody stringBody: String?, withHeader header:[String:String]?, withDestination destination: String)
 
     func stompClientDidDisconnect(client: StompClientLib!)
@@ -72,7 +72,7 @@ public protocol StompClientLibDelegate {
 public class StompClientLib: NSObject, SRWebSocketDelegate {
     var socket: SRWebSocket?
     var sessionId: String?
-    var delegate: StompClientLibDelegate?
+    weak var delegate: StompClientLibDelegate?
     var connectionHeaders: [String: String]?
     public var connection: Bool = false
     public var certificateCheckEnabled = true