[accumulo] correct cleanup to only happen once per jvm.

closes brianfrankcooper#340
jaricftw · Jan 25, 2016 · 1a719ef · 1a719ef
1 parent f839ed0
commit 1a719ef
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/accumulo/src/main/java/com/yahoo/ycsb/db/accumulo/AccumuloClient.java b/accumulo/src/main/java/com/yahoo/ycsb/db/accumulo/AccumuloClient.java
@@ -63,6 +63,16 @@ public class AccumuloClient extends DB {
   private Scanner singleScanner = null; // A scanner for reads/deletes.
   private Scanner scanScanner = null; // A scanner for use by scan()
 
+  static {
+
+    Runtime.getRuntime().addShutdownHook(new Thread() {
+      @Override
+      public void run() {
+        CleanUp.shutdownNow();
+      }
+    });
+  }
+
   @Override
   public void init() throws DBException {
     colFam = new Text(getProperties().getProperty("accumulo.columnFamily"));
@@ -96,7 +106,6 @@ public void cleanup() throws DBException {
     } catch (MutationsRejectedException e) {
       throw new DBException(e);
     }
-    CleanUp.shutdownNow();
   }
 
   /**