@@ -72,7 +72,7 @@ enum RangeBlockDataRequest<E: EthSpec> {
7272 // Indicates if we have made column requests for each of the `expected_custody_columns` or not
7373 all_requests_made : bool ,
7474 /// The column indices corresponding to the request
75- column_peers : HashMap < DataColumnsByRootRequestId , Vec < ColumnIndex > > ,
75+ request_to_column_indices : HashMap < DataColumnsByRootRequestId , Vec < ColumnIndex > > ,
7676 expected_custody_columns : HashSet < ColumnIndex > ,
7777 attempt : usize ,
7878 } ,
@@ -130,7 +130,7 @@ impl<E: EthSpec> RangeBlockComponentsRequest<E> {
130130 requests : HashMap :: new ( ) ,
131131 all_requests_made : false ,
132132 attempt : 0 ,
133- column_peers : HashMap :: new ( ) ,
133+ request_to_column_indices : HashMap :: new ( ) ,
134134 expected_custody_columns,
135135 }
136136 } else {
@@ -155,7 +155,10 @@ impl<E: EthSpec> RangeBlockComponentsRequest<E> {
155155 . iter ( )
156156 . map ( |( k, v) | ( k. peer , v. clone ( ) ) )
157157 . collect ( ) ,
158- RangeBlockDataRequest :: DataColumnsFromRoot { column_peers, .. } => column_peers
158+ RangeBlockDataRequest :: DataColumnsFromRoot {
159+ request_to_column_indices : column_peers,
160+ ..
161+ } => column_peers
159162 . iter ( )
160163 . map ( |( k, v) | ( k. peer , v. clone ( ) ) )
161164 . collect ( ) ,
@@ -200,17 +203,17 @@ impl<E: EthSpec> RangeBlockComponentsRequest<E> {
200203 requests,
201204 attempt : _,
202205 all_requests_made,
203- column_peers ,
206+ request_to_column_indices ,
204207 expected_custody_columns,
205208 } => {
206- for ( request, peers ) in column_requests {
209+ for ( request, indices ) in column_requests {
207210 requests. insert ( request, ByRangeRequest :: Active ( request) ) ;
208- column_peers . insert ( request, peers ) ;
211+ request_to_column_indices . insert ( request, indices ) ;
209212 }
210213
211214 if !* all_requests_made {
212215 let mut all_columns_requested = HashSet :: new ( ) ;
213- for columns in column_peers . values ( ) {
216+ for columns in request_to_column_indices . values ( ) {
214217 all_columns_requested. extend ( columns. iter ( ) ) ;
215218 }
216219 * all_requests_made = all_columns_requested == * expected_custody_columns;
@@ -414,7 +417,7 @@ impl<E: EthSpec> RangeBlockComponentsRequest<E> {
414417 RangeBlockDataRequest :: DataColumnsFromRoot {
415418 all_requests_made,
416419 attempt,
417- column_peers ,
420+ request_to_column_indices ,
418421 expected_custody_columns,
419422 requests,
420423 } => {
@@ -439,7 +442,7 @@ impl<E: EthSpec> RangeBlockComponentsRequest<E> {
439442
440443 // Note: this assumes that only 1 peer is responsible for a column
441444 // with a batch.
442- for ( id, columns) in column_peers {
445+ for ( id, columns) in request_to_column_indices . iter ( ) {
443446 for column in columns {
444447 column_to_peer_id. insert ( * column, id. peer ) ;
445448 }
@@ -467,6 +470,7 @@ impl<E: EthSpec> RangeBlockComponentsRequest<E> {
467470 // delete it from the entries as we are going to make
468471 // a separate attempt for those components.
469472 requests. retain ( |& k, _| k. peer != * peer) ;
473+ request_to_column_indices. retain ( |& k, _| k. peer != * peer) ;
470474 }
471475 }
472476
0 commit comments