From 55087390163d9b0c8eb94d47bf3f476d9844733c Mon Sep 17 00:00:00 2001 From: Makonike Date: Mon, 28 Aug 2023 00:49:36 +0800 Subject: [PATCH] feat: copy context when client call --- dtmsvr/trans_class.go | 2 +- dtmsvr/trans_status.go | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dtmsvr/trans_class.go b/dtmsvr/trans_class.go index 2038bdf..aad32e4 100644 --- a/dtmsvr/trans_class.go +++ b/dtmsvr/trans_class.go @@ -103,7 +103,7 @@ func TransFromDtmRequest(ctx context.Context, c *dtmgpb.DtmRequest) *TransGlobal }, }} r.ReqExtra = c.ReqExtra - r.Context = CopyContext(ctx) + r.Context = ctx if c.Steps != "" { dtmimp.MustUnmarshalString(c.Steps, &r.Steps) } diff --git a/dtmsvr/trans_status.go b/dtmsvr/trans_status.go index baa51dd..4cb2a96 100644 --- a/dtmsvr/trans_status.go +++ b/dtmsvr/trans_status.go @@ -200,7 +200,7 @@ func (t *TransGlobal) getGrpcResult(uri string, branchID, op string, branchPaylo } conn := dtmgimp.MustGetGrpcConn(server, true) - ctx := dtmgimp.TransInfo2Ctx(t.Context, t.Gid, t.TransType, branchID, op, "") + ctx := dtmgimp.TransInfo2Ctx(CopyContext(t.Context), t.Gid, t.TransType, branchID, op, "") kvs := dtmgimp.Map2Kvs(t.Ext.Headers) kvs = append(kvs, dtmgimp.Map2Kvs(t.BranchHeaders)...) ctx = metadata.AppendToOutgoingContext(ctx, kvs...)