SCB-168 connected to alpha on start

Signed-off-by: seanyinx <sean.yin@huawei.com>
apache · Jan 10, 2018 · 367b1cc · 367b1cc
1 parent 1e7ff02
commit 367b1cc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/...s/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java b/...s/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
@@ -67,7 +67,7 @@ public void tearDown() throws Exception {
     repository.deleteAll();
   }
 
-  @Test
+  @Test(timeout = 5000)
   public void updatesTxStateToAlpha() throws Exception {
     ResponseEntity<String> entity = restTemplate.getForEntity("/greet?name={name}",
         String.class,
@@ -129,7 +129,7 @@ public void updatesTxStateToAlpha() throws Exception {
     assertThat(compensatedMessages.isEmpty(), is(true));
   }
 
-  @Test
+  @Test(timeout = 5000)
   public void compensatesFailedGlobalTransaction() throws Exception {
     ResponseEntity<String> entity = restTemplate.getForEntity("/greet?name={name}",
         String.class,

diff --git a/...ing-starter/src/main/java/org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java b/...ing-starter/src/main/java/org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java
@@ -63,7 +63,11 @@ MessageSender grpcMessageSender(
         serviceConfig,
         handler);
 
-    Runtime.getRuntime().addShutdownHook(new Thread(sender::close));
+    sender.onConnected();
+    Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+      sender.onDisconnected();
+      sender.close();
+    }));
 
     return sender;
   }