Browse Source

feat: copy context when client call

pull/459/head
Makonike 3 years ago
parent
commit
5508739016
  1. 2
      dtmsvr/trans_class.go
  2. 2
      dtmsvr/trans_status.go

2
dtmsvr/trans_class.go

@ -103,7 +103,7 @@ func TransFromDtmRequest(ctx context.Context, c *dtmgpb.DtmRequest) *TransGlobal
}, },
}} }}
r.ReqExtra = c.ReqExtra r.ReqExtra = c.ReqExtra
r.Context = CopyContext(ctx) r.Context = ctx
if c.Steps != "" { if c.Steps != "" {
dtmimp.MustUnmarshalString(c.Steps, &r.Steps) dtmimp.MustUnmarshalString(c.Steps, &r.Steps)
} }

2
dtmsvr/trans_status.go

@ -200,7 +200,7 @@ func (t *TransGlobal) getGrpcResult(uri string, branchID, op string, branchPaylo
} }
conn := dtmgimp.MustGetGrpcConn(server, true) 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 := dtmgimp.Map2Kvs(t.Ext.Headers)
kvs = append(kvs, dtmgimp.Map2Kvs(t.BranchHeaders)...) kvs = append(kvs, dtmgimp.Map2Kvs(t.BranchHeaders)...)
ctx = metadata.AppendToOutgoingContext(ctx, kvs...) ctx = metadata.AppendToOutgoingContext(ctx, kvs...)

Loading…
Cancel
Save