Rename interface to OkHttp3SocketClientFactory.java. Modify ParseLiveQueryClient to accept OkHttpClient instances.

Author: rogerhu

ParseLiveQuery/src/main/java/com/parse/OkHttp3SocketClientFactory.java

@@ -0,0 +1,127 @@
+package com.parse;
+
+import android.util.Log;
+
+import java.net.URI;
+import java.util.Locale;
+
+import okhttp3.OkHttpClient;
+import okhttp3.Request;
+import okhttp3.Response;
+import okhttp3.WebSocket;
+import okhttp3.WebSocketListener;
+import okio.ByteString;
+
+/* package */
+
+/* package */ public class OkHttp3SocketClientFactory implements WebSocketClientFactory {
+
+    OkHttpClient mClient;
+
+    public OkHttp3SocketClientFactory(OkHttpClient client) {
+        mClient = client;
+    }
+
+    public OkHttp3SocketClientFactory() {
+        mClient = new OkHttpClient();
+    }
+
+    @Override
+    public WebSocketClient createInstance(WebSocketClient.WebSocketClientCallback webSocketClientCallback, URI hostUrl) {
+        return new OkHttp3WebSocketClient(mClient, webSocketClientCallback, hostUrl);
+    }
+
+    public WebSocketClient createInstance(OkHttpClient client, WebSocketClient.WebSocketClientCallback webSocketClientCallback, URI hostUrl) {
+        return new OkHttp3WebSocketClient(mClient, webSocketClientCallback, hostUrl);
+    }
+
+    class OkHttp3WebSocketClient implements WebSocketClient {
+
+        private static final String LOG_TAG = "OkHttpWebSocketClient";
+
+        private final WebSocketClientCallback webSocketClientCallback;
+        private WebSocket webSocket;
+        private State state = State.NONE;
+        private final OkHttpClient client;
+        private final String url;
+        private final int STATUS_CODE = 200;
+        private final String CLOSING_MSG = "User invoked close";
+
+        private final WebSocketListener handler = new WebSocketListener() {
+            @Override
+            public void onOpen(WebSocket webSocket, Response response) {
+                setState(State.CONNECTED);
+                webSocketClientCallback.onOpen();
+            }
+
+            @Override
+            public void onMessage(WebSocket webSocket, String text) {
+                webSocketClientCallback.onMessage(text);
+            }
+
+            @Override
+            public void onMessage(WebSocket webSocket, ByteString bytes) {
+                Log.w(LOG_TAG, String.format(Locale.US,
+                        "Socket got into inconsistent state and received %s instead.",
+                        bytes.toString()));
+            }
+
+            @Override
+            public void onClosed(WebSocket webSocket, int code, String reason) {
+                setState(State.DISCONNECTED);
+                webSocketClientCallback.onClose();
+            }
+
+            @Override
+            public void onFailure(okhttp3.WebSocket webSocket, Throwable t, Response response) {
+                webSocketClientCallback.onError(t);
+            }
+        };
+
+        private OkHttp3WebSocketClient(OkHttpClient okHttpClient,
+                WebSocketClientCallback webSocketClientCallback, URI hostUrl) {
+            client = okHttpClient;
+            this.webSocketClientCallback = webSocketClientCallback;
+            url = hostUrl.toString();
+        }
+
+        @Override
+        public synchronized void open() {
+            if (State.NONE == state) {
+                // OkHttp3 connects as soon as the socket is created so do it here.
+                Request request = new Request.Builder()
+                        .url(url)
+                        .build();
+
+                webSocket = client.newWebSocket(request, handler);
+                setState(State.CONNECTING);
+            }
+        }
+
+        @Override
+        public synchronized void close() {
+            setState(State.DISCONNECTING);
+            webSocket.close(STATUS_CODE, CLOSING_MSG);
+        }
+
+        @Override
+        public void send(String message) {
+            if (state == State.CONNECTED) {
+                webSocket.send(message);
+            }
+        }
+
+        @Override
+        public State getState() {
+            return state;
+        }
+
+        private synchronized void setState(State newState) {
+            this.state = newState;
+            this.webSocketClientCallback.stateChanged();
+        }
+
+
+    }
+
+}

ParseLiveQuery/src/main/java/com/parse/OkHttp3WebSocketClient.java

@@ -3,8 +3,9 @@
 import java.net.URI;
 import java.util.concurrent.Executor;
 
-public interface ParseLiveQueryClient {
+import okhttp3.OkHttpClient;
 
+public interface ParseLiveQueryClient {
     <T extends ParseObject> SubscriptionHandling<T> subscribe(ParseQuery<T> query);
 
     <T extends ParseObject> void unsubscribe(final ParseQuery<T> query);
@@ -21,6 +22,10 @@ public static ParseLiveQueryClient getClient(URI uri) {
             return new ParseLiveQueryClientImpl(uri);
         }
 
+        public static <T extends ParseObject> ParseLiveQueryClient getClient(URI uri, OkHttpClient okHttpClient) {
+            return new ParseLiveQueryClientImpl(uri, okHttpClient);
+        }
+
         /* package */
         static ParseLiveQueryClient getClient(URI uri, WebSocketClientFactory webSocketClientFactory, Executor taskExecutor) {
             return new ParseLiveQueryClientImpl(uri, webSocketClientFactory, taskExecutor);

ParseLiveQuery/src/main/java/com/parse/ParseLiveQueryClient.java

@@ -12,6 +12,7 @@
 
 import bolts.Continuation;
 import bolts.Task;
+import okhttp3.OkHttpClient;
 
 import static com.parse.Parse.checkInit;
 
@@ -32,7 +33,11 @@
     private boolean userInitiatedDisconnect = false;
 
     /* package */ ParseLiveQueryClientImpl(URI uri) {
-        this(uri, new TubeSockWebSocketClient.TubeWebSocketClientFactory(), Task.BACKGROUND_EXECUTOR);
+        this(uri, new OkHttp3SocketClientFactory(new OkHttpClient()), Task.BACKGROUND_EXECUTOR);
+    }
+
+    /* package */ ParseLiveQueryClientImpl(URI uri, OkHttpClient okHttpClient) {
+        this(uri, new OkHttp3SocketClientFactory(okHttpClient), Task.BACKGROUND_EXECUTOR);
     }
 
     /* package */ ParseLiveQueryClientImpl(URI uri, WebSocketClientFactory webSocketClientFactory, Executor taskExecutor) {