From 5cd824833cce13bd71a330e7b2767e6390d54bc0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kjetil=20=C3=98rbekk?= <kjetil.orbekk@gmail.com>
Date: Mon, 12 Mar 2012 13:03:33 +0100
Subject: Add ConnectionState to Client.

---
 same/src/main/java/com/orbekk/same/Client.java | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'same/src/main/java/com/orbekk/same/Client.java')

diff --git a/same/src/main/java/com/orbekk/same/Client.java b/same/src/main/java/com/orbekk/same/Client.java
index 8401c4b..f48865d 100644
--- a/same/src/main/java/com/orbekk/same/Client.java
+++ b/same/src/main/java/com/orbekk/same/Client.java
@@ -16,6 +16,7 @@ import com.orbekk.util.WorkQueue;
 
 public class Client implements DiscoveryListener {
     private Logger logger = LoggerFactory.getLogger(getClass());
+    private ConnectionState connectionState = ConnectionState.DISCONNECTED;
     private ConnectionManager connections;
     private State state;
     private String myUrl;
@@ -69,6 +70,11 @@ public class Client implements DiscoveryListener {
         public void removeStateListener(StateChangedListener listener) {
             stateListeners.remove(listener);
         }
+
+        @Override
+        public ConnectionState getConnectionState() {
+            return Client.this.getConnectionState();
+        }
     }
 
     private ClientInterface clientInterface = new ClientInterfaceImpl();
@@ -76,6 +82,7 @@ public class Client implements DiscoveryListener {
     private ClientService serviceImpl = new ClientService() {
         @Override
         public void setState(String component, String data, long revision) throws Exception {
+            connectionState = ConnectionState.STABLE;
             boolean status = state.update(component, data, revision);
             if (status) {
                 for (StateChangedListener listener : stateListeners) {
@@ -123,6 +130,7 @@ public class Client implements DiscoveryListener {
     }
 
     public void interrupt() {
+        connectionState = ConnectionState.DISCONNECTED;
         discoveryThread.interrupt();
     }
 
@@ -130,8 +138,13 @@ public class Client implements DiscoveryListener {
         return myUrl;
     }
 
+    public ConnectionState getConnectionState() {
+        return connectionState;
+    }
+    
     public void joinNetwork(String masterUrl) {
         logger.info("joinNetwork({})", masterUrl);
+        connectionState = ConnectionState.UNSTABLE;
         MasterService master = connections.getMaster(masterUrl);
         state.clear();
         try {
-- 
cgit v1.2.3