Add context to NewDealStatusStream

filecoin-project · Jul 2, 2020 · e5ad802 · e5ad802
1 parent 79f7787
commit e5ad802
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/storagemarket/impl/client.go b/storagemarket/impl/client.go
@@ -239,7 +239,7 @@ func (c *Client) GetProviderDealState(ctx context.Context, proposalCid cid.Cid)
 		return nil, xerrors.Errorf("could not get client deal state: %w", err)
 	}
 
-	s, err := c.net.NewDealStatusStream(deal.Miner)
+	s, err := c.net.NewDealStatusStream(ctx, deal.Miner)
 	if err != nil {
 		return nil, xerrors.Errorf("failed to open stream to miner: %w", err)
 	}

diff --git a/storagemarket/network/libp2p_impl.go b/storagemarket/network/libp2p_impl.go
@@ -48,8 +48,8 @@ func (impl *libp2pStorageMarketNetwork) NewDealStream(ctx context.Context, id pe
 	return &dealStream{p: id, rw: s, buffered: buffered, host: impl.host}, nil
 }
 
-func (impl *libp2pStorageMarketNetwork) NewDealStatusStream(id peer.ID) (DealStatusStream, error) {
-	s, err := impl.host.NewStream(context.Background(), id, storagemarket.DealStatusProtocolID)
+func (impl *libp2pStorageMarketNetwork) NewDealStatusStream(ctx context.Context, id peer.ID) (DealStatusStream, error) {
+	s, err := impl.host.NewStream(ctx, id, storagemarket.DealStatusProtocolID)
 	if err != nil {
 		log.Warn(err)
 		return nil, err

diff --git a/storagemarket/network/libp2p_impl_test.go b/storagemarket/network/libp2p_impl_test.go
@@ -298,7 +298,7 @@ func TestDealStatusStreamSendReceiveMultipleSuccessful(t *testing.T) {
 	ctx, cancel := context.WithTimeout(ctxBg, 10*time.Second)
 	defer cancel()
 
-	qs, err := nw1.NewDealStatusStream(td.Host2.ID())
+	qs, err := nw1.NewDealStatusStream(ctx, td.Host2.ID())
 	require.NoError(t, err)
 
 	var resp network.DealStatusResponse
@@ -439,7 +439,7 @@ func assertDealStatusRequestReceived(inCtx context.Context, t *testing.T, fromNe
 	ctx, cancel := context.WithTimeout(inCtx, 10*time.Second)
 	defer cancel()
 
-	as1, err := fromNetwork.NewDealStatusStream(toHost)
+	as1, err := fromNetwork.NewDealStatusStream(ctx, toHost)
 	require.NoError(t, err)
 
 	// send query to host2
@@ -465,7 +465,7 @@ func assertDealStatusResponseReceived(inCtx context.Context, t *testing.T,
 	defer cancel()
 
 	// setup query stream host1 --> host 2
-	as1, err := fromNetwork.NewDealStatusStream(toHost)
+	as1, err := fromNetwork.NewDealStatusStream(ctx, toHost)
 	require.NoError(t, err)
 
 	// send queryresponse to host2

diff --git a/storagemarket/network/network.go b/storagemarket/network/network.go
@@ -56,7 +56,7 @@ type StorageReceiver interface {
 type StorageMarketNetwork interface {
 	NewAskStream(context.Context, peer.ID) (StorageAskStream, error)
 	NewDealStream(context.Context, peer.ID) (StorageDealStream, error)
-	NewDealStatusStream(peer.ID) (DealStatusStream, error)
+	NewDealStatusStream(context.Context, peer.ID) (DealStatusStream, error)
 	SetDelegate(StorageReceiver) error
 	StopHandlingRequests() error
 	ID() peer.ID