Add isConnecting method

NaikSoftware · Feb 28, 2017 · eb74d88 · eb74d88
1 parent 1746741
commit eb74d88
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/lib/src/main/java/ua/naiksoftware/stomp/client/StompClient.java b/lib/src/main/java/ua/naiksoftware/stomp/client/StompClient.java
@@ -35,6 +35,7 @@ public class StompClient {
     private final ConnectionProvider mConnectionProvider;
     private HashMap<String, String> mTopics;
     private boolean mConnected;
+    private boolean isConnecting;
 
     public StompClient(ConnectionProvider connectionProvider) {
         mConnectionProvider = connectionProvider;
@@ -82,15 +83,23 @@ public void connect(List<StompHeader> _headers, boolean reconnect) {
 
                         case CLOSED:
                             mConnected = false;
+                            isConnecting = false;
+                            break;
+
+                        case ERROR:
+                            mConnected = false;
+                            isConnecting = false;
                             break;
                     }
                 });
 
+        isConnecting = true;
         mMessagesSubscription = mConnectionProvider.messages()
                 .map(StompMessage::from)
                 .subscribe(stompMessage -> {
                     if (stompMessage.getStompCommand().equals(StompCommand.CONNECTED)) {
                         mConnected = true;
+                        isConnecting = false;
                         for (ConnectableObservable<Void> observable : mWaitConnectionObservables) {
                             observable.connect();
                         }
@@ -203,4 +212,8 @@ private Observable<Void> unsubscribePath(String dest) {
     public boolean isConnected() {
         return mConnected;
     }
+
+    public boolean isConnecting() {
+        return isConnecting;
+    }
 }