Added an inner context to cut getOrdersFromPeer short

0xProject · Jun 30, 2020 · e2d5ba7 · e2d5ba7
1 parent 94dd394
commit e2d5ba7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/core/ordersync/ordersync.go b/core/ordersync/ordersync.go
@@ -294,6 +294,8 @@ func (s *Service) GetOrders(ctx context.Context, minPeers int) error {
 		i := 0
 		wg := &sync.WaitGroup{}
 		waitChan := make(chan struct{}, 1)
+		innerCtx, cancel := context.WithCancel(ctx)
+		defer cancel()
 		for _, peerID := range currentNeighbors {
 			if len(successfullySyncedPeers) >= minPeers {
 				return nil
@@ -329,7 +331,7 @@ func (s *Service) GetOrders(ctx context.Context, minPeers int) error {
 			wg.Add(1)
 			go func(id peer.ID) {
 				defer wg.Done()
-				if err := s.getOrdersFromPeer(ctx, id); err != nil {
+				if err := s.getOrdersFromPeer(innerCtx, id); err != nil {
 					log.WithFields(log.Fields{
 						"error":    err.Error(),
 						"provider": id.Pretty(),