changed checkout record code to make sure timeouts are recorded

drucci · Jun 19, 2012 · be22441 · be22441
1 parent f527daa
commit be22441
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/src/java/voldemort/store/socket/clientrequest/ClientRequestExecutorPool.java b/src/java/voldemort/store/socket/clientrequest/ClientRequestExecutorPool.java
@@ -133,20 +133,21 @@ public SocketStore create(String storeName,
      */
 
     public ClientRequestExecutor checkout(SocketDestination destination) {
+        // time checkout
+        long start = System.nanoTime();
+        ClientRequestExecutor clientRequestExecutor;
         try {
-            // time checkout
-            long start = System.nanoTime();
-            ClientRequestExecutor clientRequestExecutor = pool.checkout(destination);
+            clientRequestExecutor = pool.checkout(destination);
+        } catch(Exception e) {
+            throw new UnreachableStoreException("Failure while checking out socket for "
+                                                + destination + ": ", e);
+        } finally {
             long end = System.nanoTime();
             if(stats != null) {
                 stats.recordCheckoutTimeUs(destination, (end - start) / Time.NS_PER_US);
             }
-
-            return clientRequestExecutor;
-        } catch(Exception e) {
-            throw new UnreachableStoreException("Failure while checking out socket for "
-                                                + destination + ": ", e);
         }
+        return clientRequestExecutor;
     }
 
     /**