BOOKKEEPER-114: add a shutdown hook to shut down bookie server safely. (Sijie via ivank)

git-svn-id: https://svn.apache.org/repos/asf/zookeeper/bookkeeper/trunk@1203790 13f79535-47bb-0310-9956-ffa450edef68

Author: ivankelly

CHANGES.txt

@@ -78,6 +78,8 @@ BUGFIXES:
 
   BOOKKEEPER-115: LocalBookKeeper fails after BOOKKEEPER-108 (ivank)
 
+  BOOKKEEPER-114: add a shutdown hook to shut down bookie server safely. (Sijie via ivank)
+
  hedwig-server/
 
   BOOKKEEPER-43: NullPointException when releasing topic (Sijie Guo via breed)

bookkeeper-server/src/main/java/org/apache/bookkeeper/proto/BookieServer.java

@@ -243,8 +243,20 @@ public static void main(String[] args) throws IOException, InterruptedException,
                            conf.getBookiePort(), conf.getZkServers(),
                            conf.getJournalDirName(), sb);
         LOG.info(hello);
-        BookieServer bs = new BookieServer(conf);
+        final BookieServer bs = new BookieServer(conf);
         bs.start();
+        Runtime.getRuntime().addShutdownHook(new Thread() {
+            @Override
+            public void run() {
+                try {
+                    bs.shutdown();
+                    LOG.info("Shut down bookie server successfully");
+                } catch (InterruptedException ie) {
+                    LOG.warn("Exception when shutting down bookie server : ", ie);
+                }
+            }
+        });
+        LOG.info("Register shutdown hook successfully");
         bs.join();
     }