Skip to content

Commit b205f28

Browse files
Rename requests according to responses
1 parent 5439dbb commit b205f28

File tree

7 files changed

+83
-81
lines changed

7 files changed

+83
-81
lines changed

p2p/src/message.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,13 +21,13 @@ use common::{
2121
use serialization::{Decode, Encode};
2222

2323
#[derive(Debug, Encode, Decode, Clone, PartialEq, Eq)]
24-
pub struct HeaderRequest {
24+
pub struct HeaderListRequest {
2525
locator: Locator,
2626
}
2727

28-
impl HeaderRequest {
28+
impl HeaderListRequest {
2929
pub fn new(locator: Locator) -> Self {
30-
HeaderRequest { locator }
30+
HeaderListRequest { locator }
3131
}
3232

3333
pub fn locator(&self) -> &Locator {
@@ -40,11 +40,11 @@ impl HeaderRequest {
4040
}
4141

4242
#[derive(Debug, Encode, Decode, Clone, PartialEq, Eq)]
43-
pub struct BlockRequest {
43+
pub struct BlockListRequest {
4444
block_ids: Vec<Id<Block>>,
4545
}
4646

47-
impl BlockRequest {
47+
impl BlockListRequest {
4848
pub fn new(block_ids: Vec<Id<Block>>) -> Self {
4949
Self { block_ids }
5050
}
@@ -61,9 +61,9 @@ impl BlockRequest {
6161
#[derive(Debug, Encode, Decode, Clone, PartialEq, Eq)]
6262
pub enum Request {
6363
#[codec(index = 0)]
64-
HeaderRequest(HeaderRequest),
64+
HeaderListRequest(HeaderListRequest),
6565
#[codec(index = 1)]
66-
BlockRequest(BlockRequest),
66+
BlockListRequest(BlockListRequest),
6767
}
6868

6969
#[derive(Debug, Encode, Decode, Clone, PartialEq, Eq)]
@@ -107,9 +107,9 @@ impl BlockListResponse {
107107
#[derive(Debug, Encode, Decode, Clone, PartialEq, Eq)]
108108
pub enum Response {
109109
#[codec(index = 0)]
110-
HeaderResponse(HeaderListResponse),
110+
HeaderListResponse(HeaderListResponse),
111111
#[codec(index = 1)]
112-
BlockResponse(BlockListResponse),
112+
BlockListResponse(BlockListResponse),
113113
}
114114

115115
#[derive(Debug, Encode, Decode, Clone, PartialEq, Eq)]

p2p/src/net/mock/peer.rs

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -395,9 +395,11 @@ mod tests {
395395

396396
assert!(socket2.recv().now_or_never().is_none());
397397
socket2
398-
.send(types::Message::Request(message::Request::HeaderRequest(
399-
message::HeaderRequest::new(Locator::new(vec![])),
400-
)))
398+
.send(types::Message::Request(
399+
message::Request::HeaderListRequest(message::HeaderListRequest::new(Locator::new(
400+
vec![],
401+
))),
402+
))
401403
.await
402404
.unwrap();
403405

p2p/src/net/mock/socket.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ mod tests {
136136
let mut server_socket = MockSocket::new(res1.unwrap().0);
137137
let mut peer_socket = MockSocket::new(res2.unwrap());
138138

139-
let msg = Message::Request(Request::BlockRequest(BlockRequest::new(vec![])));
139+
let msg = Message::Request(Request::BlockListRequest(BlockListRequest::new(vec![])));
140140
peer_socket.send(msg.clone()).await.unwrap();
141141

142142
assert_eq!(server_socket.recv().await.unwrap().unwrap(), msg);

p2p/src/sync/mod.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -513,7 +513,7 @@ where
513513
request_id,
514514
request,
515515
} => match request {
516-
message::Request::HeaderRequest(request) => {
516+
message::Request::HeaderListRequest(request) => {
517517
log::debug!(
518518
"process header request (id {:?}) from peer {}",
519519
request_id, peer_id
@@ -527,7 +527,7 @@ where
527527
).await;
528528
self.handle_error(peer_id, result).await?;
529529
}
530-
message::Request::BlockRequest(request) => {
530+
message::Request::BlockListRequest(request) => {
531531
log::debug!(
532532
"process block request (id {:?}) from peer {}",
533533
request_id, peer_id
@@ -547,7 +547,7 @@ where
547547
request_id,
548548
response,
549549
} => match response {
550-
message::Response::HeaderResponse(response) => {
550+
message::Response::HeaderListResponse(response) => {
551551
log::debug!(
552552
"process header response (id {:?}) from peer {}",
553553
request_id, peer_id
@@ -557,7 +557,7 @@ where
557557
let result = self.process_header_response(peer_id, response.into_headers()).await;
558558
self.handle_error(peer_id, result).await?;
559559
}
560-
message::Response::BlockResponse(response) => {
560+
message::Response::BlockListResponse(response) => {
561561
log::debug!(
562562
"process block response (id {:?}) from peer {}",
563563
request_id, peer_id

p2p/src/sync/request.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ where
5252
/// * `block_ids` - IDs of the blocks that are requested
5353
pub fn make_block_request(&self, block_ids: Vec<Id<Block>>) -> (message::Request, RequestType) {
5454
(
55-
message::Request::BlockRequest(message::BlockRequest::new(block_ids.clone())),
55+
message::Request::BlockListRequest(message::BlockListRequest::new(block_ids.clone())),
5656
RequestType::GetBlocks(block_ids),
5757
)
5858
}
@@ -66,7 +66,7 @@ where
6666
/// * `locator` - locator object that shows the state of the local node
6767
pub fn make_header_request(&self, locator: Locator) -> (message::Request, RequestType) {
6868
(
69-
message::Request::HeaderRequest(message::HeaderRequest::new(locator)),
69+
message::Request::HeaderListRequest(message::HeaderListRequest::new(locator)),
7070
RequestType::GetHeaders,
7171
)
7272
}
@@ -76,15 +76,15 @@ where
7676
/// # Arguments
7777
/// * `headers` - the headers that were requested
7878
pub fn make_header_response(&self, headers: Vec<BlockHeader>) -> message::Response {
79-
message::Response::HeaderResponse(message::HeaderListResponse::new(headers))
79+
message::Response::HeaderListResponse(message::HeaderListResponse::new(headers))
8080
}
8181

8282
/// Make block response
8383
///
8484
/// # Arguments
8585
/// * `blocks` - the blocks that were requested
8686
pub fn make_block_response(&self, blocks: Vec<Block>) -> message::Response {
87-
message::Response::BlockResponse(message::BlockListResponse::new(blocks))
87+
message::Response::BlockListResponse(message::BlockListResponse::new(blocks))
8888
}
8989

9090
/// Helper function for sending a request to remote

p2p/src/sync/tests/request_response.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ async fn test_request_response() {
3232
mgr1.peer_sync_handle
3333
.send_request(
3434
*conn2.peer_id(),
35-
Request::HeaderRequest(HeaderRequest::new(Locator::new(vec![]))),
35+
Request::HeaderListRequest(HeaderListRequest::new(Locator::new(vec![]))),
3636
)
3737
.await
3838
.unwrap();
@@ -45,13 +45,13 @@ async fn test_request_response() {
4545
{
4646
assert_eq!(
4747
request,
48-
Request::HeaderRequest(HeaderRequest::new(Locator::new(vec![])))
48+
Request::HeaderListRequest(HeaderListRequest::new(Locator::new(vec![])))
4949
);
5050

5151
mgr2.peer_sync_handle
5252
.send_response(
5353
request_id,
54-
Response::HeaderResponse(HeaderListResponse::new(vec![])),
54+
Response::HeaderListResponse(HeaderListResponse::new(vec![])),
5555
)
5656
.await
5757
.unwrap();
@@ -75,7 +75,7 @@ async fn test_multiple_requests_and_responses() {
7575
.peer_sync_handle
7676
.send_request(
7777
*conn2.peer_id(),
78-
Request::HeaderRequest(HeaderRequest::new(Locator::new(vec![]))),
78+
Request::HeaderListRequest(HeaderListRequest::new(Locator::new(vec![]))),
7979
)
8080
.await
8181
.unwrap();
@@ -85,7 +85,7 @@ async fn test_multiple_requests_and_responses() {
8585
.peer_sync_handle
8686
.send_request(
8787
*conn2.peer_id(),
88-
Request::HeaderRequest(HeaderRequest::new(Locator::new(vec![]))),
88+
Request::HeaderListRequest(HeaderListRequest::new(Locator::new(vec![]))),
8989
)
9090
.await
9191
.unwrap();
@@ -100,7 +100,7 @@ async fn test_multiple_requests_and_responses() {
100100
mgr2.peer_sync_handle
101101
.send_response(
102102
request_id,
103-
Response::HeaderResponse(HeaderListResponse::new(vec![])),
103+
Response::HeaderListResponse(HeaderListResponse::new(vec![])),
104104
)
105105
.await
106106
.unwrap();

0 commit comments

Comments
 (0)