Log at error level, set the last heartbeat whenever a message is received

Author: jkodumal

CHANGELOG.md

@@ -3,6 +3,10 @@
 
 All notable changes to the LaunchDarkly Java SDK will be documented in this file. This project adheres to [Semantic Versioning](http://semver.org).
 
+## [2.0.5] - 2016-11-09
+### Changed
+- The StreamProcessor now listens for heartbeats from the streaming API, and will automatically reconnect if heartbeats are not received.
+
 ## [2.0.4] - 2016-10-12
 ### Changed
 - Updated GSON dependency version to 2.7

build.gradle

@@ -19,7 +19,7 @@ repositories {
 
 allprojects {
     group = 'com.launchdarkly'
-    version = "2.0.4"
+    version = "2.0.5"
     sourceCompatibility = 1.7
     targetCompatibility = 1.7
 }

src/main/java/com/launchdarkly/client/StreamProcessor.java

@@ -61,11 +61,11 @@ public Future<Void> start() {
 
       @Override
       public void onOpen() throws Exception {
-        lastHeartbeat = DateTime.now();
       }
 
       @Override
       public void onMessage(String name, MessageEvent event) throws Exception {
+        lastHeartbeat = DateTime.now();
         Gson gson = new Gson();
         switch (name) {
           case PUT:
@@ -208,14 +208,13 @@ public void run() {
         try {
           logger.info("Stream stopped receiving heartbeats- reconnecting.");
           es.close();
-          start();
         } catch (IOException e) {
-          logger.warn("Encountered exception closing stream connection: " + e.getMessage());
+          logger.error("Encountered exception closing stream connection: " + e.getMessage());
         } finally {
           if (es.getState() == ReadyState.SHUTDOWN) {
             start();
           } else {
-            logger.warn("Expected ES to be in state SHUTDOWN, but it's currently in state " + es.getState().toString());
+            logger.error("Expected ES to be in state SHUTDOWN, but it's currently in state " + es.getState().toString());
           }
         }
       }