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.Context = CopyContext(ctx)
r.Context = ctx
if c.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)
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...)

Loading…
Cancel
Save