[ISSUE apache#251] Add shutdownHook to WindowAvg example

Jargon9 · Feb 5, 2023 · 040a9df · 040a9df
1 parent 61b081c
commit 040a9df
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/example/src/main/java/org/apache/rocketmq/streams/examples/window/WindowAvg.java b/example/src/main/java/org/apache/rocketmq/streams/examples/window/WindowAvg.java
@@ -29,6 +29,7 @@
 import java.nio.charset.StandardCharsets;
 import java.time.LocalTime;
 import java.util.Properties;
+import java.util.concurrent.CountDownLatch;
 
 /**
  * 1、启动RocketMQ
@@ -61,6 +62,22 @@ public static void main(String[] args) {
 
  RocketMQStream rocketMQStream = new RocketMQStream(topologyBuilder, properties);
 
- rocketMQStream.start();
+ final CountDownLatch latch = new CountDownLatch(1);
+
+ Runtime.getRuntime().addShutdownHook(new Thread("WindowAvg-shutdown-hook") {
+ @Override
+ public void run() {
+ rocketMQStream.stop();
+ latch.countDown();
+ }
+ });
+
+ try {
+ rocketMQStream.start();
+ latch.await();
+ } catch (final Throwable e) {
+ System.exit(1);
+ }
+ System.exit(0);
  }
 }