error message add data, push message content string->bytes

mpusher · Aug 19, 2016 · b2e612a · b2e612a
1 parent a1723ea
commit b2e612a
Show file tree

Hide file tree

Showing 6 changed files with 12 additions and 11 deletions.
diff --git a/mpush_client/ClientListener.swift b/mpush_client/ClientListener.swift
@@ -16,7 +16,7 @@ protocol ClientListener {
 
     func onHandshakeOk(client: Client, heartbeat: Int);
 
-    func onReceivePush(client: Client, content: String);
+    func onReceivePush(client: Client, content: NSData);
 
     func onKickUser(deviceId: String, userId: String);
 }
diff --git a/mpush_client/DefaultClientListener.swift b/mpush_client/DefaultClientListener.swift
@@ -39,7 +39,7 @@ final class DefaultClientListener: ClientListener {
         }
     }
 
-    func onReceivePush(client: Client, content: String) {
+    func onReceivePush(client: Client, content: NSData) {
         if let listener = self.listener {
             dispatch_async(dispatch_queue, {listener.onReceivePush(client, content: content)})
         }

diff --git a/mpush_client/ErrorMessage.swift b/mpush_client/ErrorMessage.swift
@@ -9,14 +9,16 @@
 import Foundation
 
 final class ErrorMessage: ByteBufMessage, CustomDebugStringConvertible {
-    var cmd:Int8 = 0;
-    var code:Int8 = 0;
-    var reason:String?;
+    var cmd: Int8 = 0;
+    var code: Int8 = 0;
+    var reason: String?;
+    var data: String?;
 
     override func decode(body: RFIReader) {
         cmd = body.readByte();
         code = body.readByte();
         reason = body.readString();
+        data = body.readString();
     }
 
 

diff --git a/mpush_client/PacketDecoder.swift b/mpush_client/PacketDecoder.swift
@@ -25,7 +25,6 @@ final class PacketDecoder {
                 return Packet.HB_PACKET;
             }
             buffer.reset();
-
         }
         return nil;
     }
@@ -54,7 +53,7 @@ final class PacketDecoder {
             let data = buffer.readBytes(bodyLength)
             body = NSData(bytes: data, length: data.count);
         }
-        return Packet(cmd: command,cc: cc,flags: flags,sessionId: sessionId,lrc:lrc,body:body);
+        return Packet(cmd: command, cc: cc, flags: flags, sessionId: sessionId, lrc: lrc, body: body);
     }
 
 }
diff --git a/mpush_client/PushMessage.swift b/mpush_client/PushMessage.swift
@@ -9,13 +9,13 @@
 import Foundation
 
 final class PushMessage: BaseMessage, CustomDebugStringConvertible {
-    var content:String!;
+    var content:NSData!;
 
     override func decode(body:NSData) {
-        content = String(data: body, encoding:NSUTF8StringEncoding)
+        content = body;
     }
 
     var debugDescription: String {
-        return "PushMessage={content:\(content)}"
+        return "PushMessage={content:\(content.length)}"
     }
 }
diff --git a/mpush_clientTests/mpush_clientTests.swift b/mpush_clientTests/mpush_clientTests.swift
@@ -60,7 +60,7 @@ class mpush_clientTests: XCTestCase {
             }
         }
 
-        func onReceivePush(client: Client, content: String){}
+        func onReceivePush(client: Client, content: NSData){}
 
         func onKickUser(deviceId: String, userId: String){}
     }