Log query stats in case of Closed Channel also

Signed-off-by: Chris Larsen <clarsen@yahoo-inc.com>
OpenTSDB · May 14, 2017 · bf3e374 · bf3e374
1 parent 2316752
commit bf3e374
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/tsd/AbstractHttpQuery.java b/src/tsd/AbstractHttpQuery.java
@@ -386,6 +386,9 @@ public void notFound() {
    */
   public void sendStatusOnly(final HttpResponseStatus status) {
     if (!chan.isConnected()) {
+      if(stats != null) {
+        stats.markSendFailed();
+      }
       done();
       return;
     }
@@ -414,6 +417,9 @@ public void sendBuffer(final HttpResponseStatus status,
                           final ChannelBuffer buf,
                           final String contentType) {
     if (!chan.isConnected()) {
+      if(stats != null) {
+        stats.markSendFailed();
+      }
       done();
       return;
     }
@@ -442,7 +448,10 @@ public void sendBuffer(final HttpResponseStatus status,
   private class SendSuccess implements ChannelFutureListener {
     @Override
     public void operationComplete(final ChannelFuture future) throws Exception {
-      stats.markSent();
+      if(future.isSuccess()) {
+        stats.markSent();}
+      else
+        stats.markSendFailed();
     }
   }