|
23 | 23 | import com.appunite.websocket.rx.messages.RxEventDisconnected; |
24 | 24 | import com.appunite.websocket.rx.messages.RxEventPong; |
25 | 25 | import com.appunite.websocket.rx.messages.RxEventStringMessage; |
26 | | -import com.squareup.okhttp.OkHttpClient; |
27 | | -import com.squareup.okhttp.Request; |
28 | | -import com.squareup.okhttp.Response; |
29 | | -import com.squareup.okhttp.ws.WebSocket; |
30 | | -import com.squareup.okhttp.ws.WebSocketCall; |
31 | | -import com.squareup.okhttp.ws.WebSocketListener; |
| 26 | +import okhttp3.OkHttpClient; |
| 27 | +import okhttp3.Request; |
| 28 | +import okhttp3.Response; |
| 29 | +import okhttp3.ResponseBody; |
| 30 | +import okhttp3.ws.WebSocket; |
| 31 | +import okhttp3.ws.WebSocketCall; |
| 32 | +import okhttp3.ws.WebSocketListener; |
32 | 33 |
|
33 | 34 | import java.io.IOException; |
34 | 35 |
|
35 | 36 | import javax.annotation.Nonnull; |
36 | 37 | import javax.annotation.Nullable; |
37 | 38 |
|
38 | 39 | import okio.Buffer; |
39 | | -import okio.BufferedSource; |
40 | 40 | import rx.Observable; |
41 | 41 | import rx.Subscriber; |
42 | 42 | import rx.functions.Action0; |
@@ -122,19 +122,19 @@ private void returnException(IOException e) { |
122 | 122 | } |
123 | 123 |
|
124 | 124 | @Override |
125 | | - public void onMessage(BufferedSource payload, WebSocket.PayloadType type) throws IOException { |
| 125 | + public void onMessage(ResponseBody message) throws IOException { |
126 | 126 | try { |
127 | 127 | final WebSocket sender = webSocketOrNull(); |
128 | 128 | if (sender == null) { |
129 | 129 | return; |
130 | 130 | } |
131 | | - if (WebSocket.PayloadType.BINARY.equals(type)) { |
132 | | - subscriber.onNext(new RxEventBinaryMessage(sender, payload.readByteArray())); |
133 | | - } else if (WebSocket.PayloadType.TEXT.equals(type)) { |
134 | | - subscriber.onNext(new RxEventStringMessage(sender, payload.readUtf8())); |
| 131 | + if (WebSocket.BINARY.equals(message.contentType())) { |
| 132 | + subscriber.onNext(new RxEventBinaryMessage(sender, message.bytes())); |
| 133 | + } else if (WebSocket.TEXT.equals(message.contentType())) { |
| 134 | + subscriber.onNext(new RxEventStringMessage(sender, message.string())); |
135 | 135 | } |
136 | 136 | } finally { |
137 | | - payload.close(); |
| 137 | + message.close(); |
138 | 138 | } |
139 | 139 | } |
140 | 140 |
|
|
0 commit comments