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