[incubator-servicecomb-saga] 07/07: SCB-168 connected to alpha on start

[ningjiang]

This is an automated email from the ASF dual-hosted git repository.

ningjiang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-servicecomb-saga.git

commit 367b1cc60c26b8bb30e42c6013de212f0e8568e9
Author: seanyinx <sean....@huawei.com>
AuthorDate: Mon Jan 8 17:14:35 2018 +0800

    SCB-168 connected to alpha on start
    
    Signed-off-by: seanyinx <sean....@huawei.com>
---
 .../org/apache/servicecomb/saga/integration/pack/tests/PackIT.java  | 4 ++--
 .../org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java | 6 +++++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git 
a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
 
b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
index 32d4254..56092be 100644
--- 
a/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
+++ 
b/integration-tests/pack-tests/src/test/java/org/apache/servicecomb/saga/integration/pack/tests/PackIT.java
@@ -67,7 +67,7 @@ public class PackIT {
     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 class PackIT {
     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/omega/omega-spring-starter/src/main/java/org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java
 
b/omega/omega-spring-starter/src/main/java/org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java
index 5c6a397..b5d118a 100644
--- 
a/omega/omega-spring-starter/src/main/java/org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java
+++ 
b/omega/omega-spring-starter/src/main/java/org/apache/servicecomb/saga/omega/spring/OmegaSpringConfig.java
@@ -63,7 +63,11 @@ class OmegaSpringConfig {
         serviceConfig,
         handler);
 
-    Runtime.getRuntime().addShutdownHook(new Thread(sender::close));
+    sender.onConnected();
+    Runtime.getRuntime().addShutdownHook(new Thread(() -> {
+      sender.onDisconnected();
+      sender.close();
+    }));
 
     return sender;
   }

-- 
To stop receiving notification emails like this one, please contact
"commits@servicecomb.apache.org" <commits@servicecomb.apache.org>.