use struct's context

Signed-off-by: Angelo De Caro <[email protected]>

Author: adecaro

Diff for: platform/fabric/chaincode.go

@@ -146,6 +146,11 @@ func (i *ChaincodeDiscover) WithFilterByMSPIDs(mspIDs ...string) *ChaincodeDisco
 	return i
 }
 
+func (i *ChaincodeDiscover) WithContext(context context.Context) *ChaincodeDiscover {
+	i.ChaincodeDiscover.WithContext(context)
+	return i
+}
+
 type ChaincodeInvocation struct {
 	driver.ChaincodeInvocation
 }

Diff for: platform/fabric/core/generic/chaincode/discovery.go

@@ -29,13 +29,15 @@ type Discovery struct {
 	QueryForPeers       bool
 
 	DefaultTTL time.Duration
+	Context    context.Context
 }
 
 func NewDiscovery(chaincode *Chaincode) *Discovery {
 	// set key to the concatenation of chaincode name and version
 	return &Discovery{
 		chaincode:  chaincode,
 		DefaultTTL: chaincode.ChannelConfig.DiscoveryDefaultTTLS(),
+		Context:    context.Background(),
 	}
 }
 
@@ -230,7 +232,7 @@ func (d *Discovery) query(req *discovery.Request) (discovery.Response, error) {
 		ClientIdentity:    signerRaw,
 		ClientTlsCertHash: ClientTLSCertHash,
 	}
-	timeout, cancel := context.WithTimeout(context.Background(), d.chaincode.ChannelConfig.DiscoveryTimeout())
+	timeout, cancel := context.WithTimeout(d.Context, d.chaincode.ChannelConfig.DiscoveryTimeout())
 	defer cancel()
 	cl, err := pc.DiscoveryClient()
 	if err != nil {
@@ -318,6 +320,10 @@ func (d *Discovery) ChaincodeVersion() (string, error) {
 	return "", errors.Errorf("chaincode [%s] not found", d.chaincode.name)
 }
 
+func (d *Discovery) WithContext(context context.Context) {
+	d.Context = context
+}
+
 func ccCall(ccNames ...string) []*peer.ChaincodeCall {
 	var call []*peer.ChaincodeCall
 	for _, ccName := range ccNames {

Diff for: platform/fabric/core/generic/chaincode/invoke.go

@@ -63,6 +63,7 @@ func NewInvoke(chaincode *Chaincode, newChaincodeDiscover NewChaincodeDiscoverFu
 		NumRetries:           int(chaincode.NumRetries),
 		RetrySleep:           chaincode.RetrySleep,
 		NewChaincodeDiscover: newChaincodeDiscover,
+		Context:              context.Background(),
 	}
 }
 
@@ -316,6 +317,8 @@ func (i *Invoke) prepare(query bool) (string, *pb.Proposal, []*pb.ProposalRespon
 			i.EndorsersMSPIDs...,
 		).WithImplicitCollections(
 			i.ImplicitCollectionMSPIDs...,
+		).WithContext(
+			i.Context,
 		)
 		if query {
 			discovery.WithForQuery()
@@ -473,7 +476,8 @@ func (i *Invoke) collectResponses(endorserClients []pb.EndorserClient, signedPro
 	for _, endorser := range endorserClients {
 		go func(endorser pb.EndorserClient) {
 			defer wg.Done()
-			proposalResp, err := endorser.ProcessProposal(context.Background(), signedProposal)
+			// TODO: we could evaluate the policy already here after we get a result to see if still need more answers
+			proposalResp, err := endorser.ProcessProposal(i.Context, signedProposal)
 			if err != nil {
 				errorCh <- err
 				return
@@ -553,7 +557,7 @@ func (i *Invoke) broadcast(txID string, env *common.Envelope) error {
 	if err := i.Chaincode.Broadcaster.Broadcast(i.Context, env); err != nil {
 		return err
 	}
-	return i.Chaincode.Finality.IsFinal(context.Background(), txID)
+	return i.Chaincode.Finality.IsFinal(i.Context, txID)
 }
 
 func (i *Invoke) checkQueryPolicy(errs []error, successes int, n int) error {

Diff for: platform/fabric/driver/chaincode.go

@@ -92,6 +92,7 @@ type ChaincodeDiscover interface {
 	WithImplicitCollections(mspIDs ...string) ChaincodeDiscover
 	WithForQuery() ChaincodeDiscover
 	ChaincodeVersion() (string, error)
+	WithContext(ctx context.Context)
 }
 
 // Chaincode exposes chaincode-related functions