diff --git a/dtmgrpc/dtmgimp/types.go b/dtmgrpc/dtmgimp/types.go index bb7fcaf..7710665 100644 --- a/dtmgrpc/dtmgimp/types.go +++ b/dtmgrpc/dtmgimp/types.go @@ -58,7 +58,7 @@ func InvokeBranch(t *dtmimp.TransBase, isRaw bool, msg proto.Message, url string } ctx := TransInfo2Ctx(t.Gid, t.TransType, branchID, op, t.Dtm) ctx = metadata.AppendToOutgoingContext(ctx, Map2Kvs(t.BranchHeaders)...) - if t.TransType == "xa" { // xa branch need addtional phase2_url + if t.TransType == "xa" { // xa branch need additional phase2_url ctx = metadata.AppendToOutgoingContext(ctx, Map2Kvs(map[string]string{dtmpre + "phase2_url": url})...) } return MustGetGrpcConn(server, isRaw).Invoke(ctx, method, msg, reply) diff --git a/dtmgrpc/dtmgimp/utils.go b/dtmgrpc/dtmgimp/utils.go index 0e46c70..b1e4b36 100644 --- a/dtmgrpc/dtmgimp/utils.go +++ b/dtmgrpc/dtmgimp/utils.go @@ -102,6 +102,7 @@ func GetMetaFromContext(ctx context.Context, name string) string { return mdGet(md, name) } +// GetDtmMetaFromContext get dtm header from context func GetDtmMetaFromContext(ctx context.Context, name string) string { md, _ := metadata.FromIncomingContext(ctx) return dtmGet(md, name)