From b99ac93f6d4ede013024323ce8a8649ed7bc0c01 Mon Sep 17 00:00:00 2001
From: "junzhou.guo" <junzhou.guo@yunzhanghu.com>
Date: Fri, 11 Nov 2022 18:59:11 +0800
Subject: [PATCH] add setup param for NewSagaGrpc

---
 client/dtmgrpc/saga.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/client/dtmgrpc/saga.go b/client/dtmgrpc/saga.go
index 9d801aa..cc13df8 100644
--- a/client/dtmgrpc/saga.go
+++ b/client/dtmgrpc/saga.go
@@ -18,8 +18,14 @@ type SagaGrpc struct {
 }
 
 // NewSagaGrpc create a saga
-func NewSagaGrpc(server string, gid string) *SagaGrpc {
-	return &SagaGrpc{Saga: *dtmcli.NewSaga(server, gid)}
+func NewSagaGrpc(server string, gid string, opts ...TransBaseOption) *SagaGrpc {
+	sg := &SagaGrpc{Saga: *dtmcli.NewSaga(server, gid)}
+
+	for _, opt := range opts {
+		opt(&sg.TransBase)
+	}
+
+	return sg
 }
 
 // Add add a saga step