From 898b82426dfeefa7ddc5034a99b8f5859d74777d Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Wed, 11 Aug 2021 21:45:50 +0800
Subject: [PATCH] add wait example

---
 examples/grpc_saga.go | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/examples/grpc_saga.go b/examples/grpc_saga.go
index 3e42485..8d90d6e 100644
--- a/examples/grpc_saga.go
+++ b/examples/grpc_saga.go
@@ -9,11 +9,22 @@ func init() {
 	addSample("grpc_saga", func() string {
 		req := dtmcli.MustMarshal(&TransReq{Amount: 30})
 		gid := dtmgrpc.MustGenGid(DtmGrpcServer)
-		msg := dtmgrpc.NewSaga(DtmGrpcServer, gid).
+		saga := dtmgrpc.NewSaga(DtmGrpcServer, gid).
 			Add(BusiGrpc+"/examples.Busi/TransOut", BusiGrpc+"/examples.Busi/TransOutRevert", req).
 			Add(BusiGrpc+"/examples.Busi/TransIn", BusiGrpc+"/examples.Busi/TransOutRevert", req)
-		err := msg.Submit()
+		err := saga.Submit()
 		dtmcli.FatalIfError(err)
-		return msg.Gid
+		return saga.Gid
+	})
+	addSample("grpc_saga_wait", func() string {
+		req := dtmcli.MustMarshal(&TransReq{Amount: 30})
+		gid := dtmgrpc.MustGenGid(DtmGrpcServer)
+		saga := dtmgrpc.NewSaga(DtmGrpcServer, gid).
+			Add(BusiGrpc+"/examples.Busi/TransOut", BusiGrpc+"/examples.Busi/TransOutRevert", req).
+			Add(BusiGrpc+"/examples.Busi/TransIn", BusiGrpc+"/examples.Busi/TransOutRevert", req)
+		saga.WaitResult = true
+		err := saga.Submit()
+		dtmcli.FatalIfError(err)
+		return saga.Gid
 	})
 }