fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-05-06 16:12:18 +00:00
Code Issues Projects Releases Wiki Activity

swarm-derive/: Allow mixing of ignored fields (#2570)

Browse Source
This commit is contained in:
Addy Bryant 2022-03-18 11:37:00 -04:00 committed by GitHub
parent 7f97d49073
commit b0dc4bc1c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 90 additions and 128 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
swarm-derive/CHANGELOG.md
View File

@ -1,3 +1,9 @@
# 0.27.1
- Allow mixing of ignored fields. See [PR 2570].
[PR 2570]: https://github.com/libp2p/rust-libp2p/pull/2570
# 0.27.0 [2022-02-22] # 0.27.0 [2022-02-22]
- Adjust to latest changes in `libp2p-swarm`. - Adjust to latest changes in `libp2p-swarm`.

2
swarm-derive/Cargo.toml
View File

@ -3,7 +3,7 @@ name = "libp2p-swarm-derive"
edition = "2021" edition = "2021"
rust-version = "1.56.1" rust-version = "1.56.1"
description = "Procedural macros of libp2p-core" description = "Procedural macros of libp2p-core"
version = "0.27.0" version = "0.27.1"
authors = ["Parity Technologies <admin@parity.io>"] authors = ["Parity Technologies <admin@parity.io>"]
license = "MIT" license = "MIT"
repository = "https://github.com/libp2p/rust-libp2p" repository = "https://github.com/libp2p/rust-libp2p"

183
swarm-derive/src/lib.rs
View File

@ -92,6 +92,13 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
event_process event_process
}; };
// The fields of the struct we are interested in (no ignored fields).
let data_struct_fields = data_struct
.fields
.iter()
.filter(|f| !is_ignored(f))
.collect::<Vec<_>>();
// The final out event. // The final out event.
// If we find a `#[behaviour(out_event = "Foo")]` attribute on the struct, we set `Foo` as // If we find a `#[behaviour(out_event = "Foo")]` attribute on the struct, we set `Foo` as
// the out event. Otherwise we use `()`. // the out event. Otherwise we use `()`.
@ -117,10 +124,8 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// Build the `where ...` clause of the trait implementation. // Build the `where ...` clause of the trait implementation.
let where_clause = { let where_clause = {
let additional = data_struct let additional = data_struct_fields
.fields
.iter() .iter()
.filter(|x| !is_ignored(x))
.flat_map(|field| { .flat_map(|field| {
let ty = &field.ty; let ty = &field.ty;
vec![ vec![
@ -147,57 +152,42 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// Build the list of statements to put in the body of `addresses_of_peer()`. // Build the list of statements to put in the body of `addresses_of_peer()`.
let addresses_of_peer_stmts = { let addresses_of_peer_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote! { out.extend(self.#i.addresses_of_peer(peer_id)); },
return None; None => quote! { out.extend(self.#field_n.addresses_of_peer(peer_id)); },
}
Some(match field.ident {
Some(ref i) => quote! { out.extend(self.#i.addresses_of_peer(peer_id)); },
None => quote! { out.extend(self.#field_n.addresses_of_peer(peer_id)); },
})
}) })
}; };
// Build the list of statements to put in the body of `inject_connection_established()`. // Build the list of statements to put in the body of `inject_connection_established()`.
let inject_connection_established_stmts = { let inject_connection_established_stmts = {
data_struct.fields.iter().enumerate().filter_map(move |(field_n, field)| { data_struct_fields.iter().enumerate().map(move |(field_n, field)| {
if is_ignored(field) { match field.ident {
return None;
}
Some(match field.ident {
Some(ref i) => quote!{ self.#i.inject_connection_established(peer_id, connection_id, endpoint, errors, other_established); }, Some(ref i) => quote!{ self.#i.inject_connection_established(peer_id, connection_id, endpoint, errors, other_established); },
None => quote!{ self.#field_n.inject_connection_established(peer_id, connection_id, endpoint, errors, other_established); }, None => quote!{ self.#field_n.inject_connection_established(peer_id, connection_id, endpoint, errors, other_established); },
}) }
}) })
}; };
// Build the list of statements to put in the body of `inject_address_change()`. // Build the list of statements to put in the body of `inject_address_change()`.
let inject_address_change_stmts = { let inject_address_change_stmts = {
data_struct.fields.iter().enumerate().filter_map(move |(field_n, field)| { data_struct_fields.iter().enumerate().map(move |(field_n, field)| {
if is_ignored(field) { match field.ident {
return None;
}
Some(match field.ident {
Some(ref i) => quote!{ self.#i.inject_address_change(peer_id, connection_id, old, new); }, Some(ref i) => quote!{ self.#i.inject_address_change(peer_id, connection_id, old, new); },
None => quote!{ self.#field_n.inject_address_change(peer_id, connection_id, old, new); }, None => quote!{ self.#field_n.inject_address_change(peer_id, connection_id, old, new); },
}) }
}) })
}; };
// Build the list of statements to put in the body of `inject_connection_closed()`. // Build the list of statements to put in the body of `inject_connection_closed()`.
let inject_connection_closed_stmts = { let inject_connection_closed_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
// The outmost handler belongs to the last behaviour. // The outmost handler belongs to the last behaviour.
.rev() .rev()
.filter(|f| !is_ignored(f.1))
.enumerate() .enumerate()
.map(move |(enum_n, (field_n, field))| { .map(move |(enum_n, (field_n, field))| {
let handler = if field_n == 0 { let handler = if field_n == 0 {
@ -222,13 +212,11 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// Build the list of statements to put in the body of `inject_dial_failure()`. // Build the list of statements to put in the body of `inject_dial_failure()`.
let inject_dial_failure_stmts = { let inject_dial_failure_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
// The outmost handler belongs to the last behaviour. // The outmost handler belongs to the last behaviour.
.rev() .rev()
.filter(|f| !is_ignored(f.1))
.enumerate() .enumerate()
.map(move |(enum_n, (field_n, field))| { .map(move |(enum_n, (field_n, field))| {
let handler = if field_n == 0 { let handler = if field_n == 0 {
@ -258,12 +246,10 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// Build the list of statements to put in the body of `inject_listen_failure()`. // Build the list of statements to put in the body of `inject_listen_failure()`.
let inject_listen_failure_stmts = { let inject_listen_failure_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.rev() .rev()
.filter(|f| !is_ignored(f.1))
.enumerate() .enumerate()
.map(move |(enum_n, (field_n, field))| { .map(move |(enum_n, (field_n, field))| {
let handler = if field_n == 0 { let handler = if field_n == 0 {
@ -288,125 +274,78 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// Build the list of statements to put in the body of `inject_new_listener()`. // Build the list of statements to put in the body of `inject_new_listener()`.
let inject_new_listener_stmts = { let inject_new_listener_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote! { self.#i.inject_new_listener(id); },
return None; None => quote! { self.#field_n.inject_new_listener(id); },
}
Some(match field.ident {
Some(ref i) => quote! { self.#i.inject_new_listener(id); },
None => quote! { self.#field_n.inject_new_listener(id); },
})
}) })
}; };
// Build the list of statements to put in the body of `inject_new_listen_addr()`. // Build the list of statements to put in the body of `inject_new_listen_addr()`.
let inject_new_listen_addr_stmts = { let inject_new_listen_addr_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote! { self.#i.inject_new_listen_addr(id, addr); },
return None; None => quote! { self.#field_n.inject_new_listen_addr(id, addr); },
}
Some(match field.ident {
Some(ref i) => quote! { self.#i.inject_new_listen_addr(id, addr); },
None => quote! { self.#field_n.inject_new_listen_addr(id, addr); },
})
}) })
}; };
// Build the list of statements to put in the body of `inject_expired_listen_addr()`. // Build the list of statements to put in the body of `inject_expired_listen_addr()`.
let inject_expired_listen_addr_stmts = { let inject_expired_listen_addr_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote! { self.#i.inject_expired_listen_addr(id, addr); },
return None; None => quote! { self.#field_n.inject_expired_listen_addr(id, addr); },
}
Some(match field.ident {
Some(ref i) => quote! { self.#i.inject_expired_listen_addr(id, addr); },
None => quote! { self.#field_n.inject_expired_listen_addr(id, addr); },
})
}) })
}; };
// Build the list of statements to put in the body of `inject_new_external_addr()`. // Build the list of statements to put in the body of `inject_new_external_addr()`.
let inject_new_external_addr_stmts = { let inject_new_external_addr_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote! { self.#i.inject_new_external_addr(addr); },
return None; None => quote! { self.#field_n.inject_new_external_addr(addr); },
}
Some(match field.ident {
Some(ref i) => quote! { self.#i.inject_new_external_addr(addr); },
None => quote! { self.#field_n.inject_new_external_addr(addr); },
})
}) })
}; };
// Build the list of statements to put in the body of `inject_expired_external_addr()`. // Build the list of statements to put in the body of `inject_expired_external_addr()`.
let inject_expired_external_addr_stmts = { let inject_expired_external_addr_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote! { self.#i.inject_expired_external_addr(addr); },
return None; None => quote! { self.#field_n.inject_expired_external_addr(addr); },
}
Some(match field.ident {
Some(ref i) => quote! { self.#i.inject_expired_external_addr(addr); },
None => quote! { self.#field_n.inject_expired_external_addr(addr); },
})
}) })
}; };
// Build the list of statements to put in the body of `inject_listener_error()`. // Build the list of statements to put in the body of `inject_listener_error()`.
let inject_listener_error_stmts = { let inject_listener_error_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote!(self.#i.inject_listener_error(id, err);),
return None; None => quote!(self.#field_n.inject_listener_error(id, err);),
}
Some(match field.ident {
Some(ref i) => quote!(self.#i.inject_listener_error(id, err);),
None => quote!(self.#field_n.inject_listener_error(id, err);),
})
}) })
}; };
// Build the list of statements to put in the body of `inject_listener_closed()`. // Build the list of statements to put in the body of `inject_listener_closed()`.
let inject_listener_closed_stmts = { let inject_listener_closed_stmts = {
data_struct data_struct_fields
.fields
.iter() .iter()
.enumerate() .enumerate()
.filter_map(move |(field_n, field)| { .map(move |(field_n, field)| match field.ident {
if is_ignored(field) { Some(ref i) => quote!(self.#i.inject_listener_closed(id, reason);),
return None; None => quote!(self.#field_n.inject_listener_closed(id, reason);),
}
Some(match field.ident {
Some(ref i) => quote!(self.#i.inject_listener_closed(id, reason);),
None => quote!(self.#field_n.inject_listener_closed(id, reason);),
})
}) })
}; };
@ -414,14 +353,14 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// //
// The event type is a construction of nested `#either_ident`s of the events of the children. // The event type is a construction of nested `#either_ident`s of the events of the children.
// We call `inject_event` on the corresponding child. // We call `inject_event` on the corresponding child.
let inject_node_event_stmts = data_struct.fields.iter().enumerate().filter(|f| !is_ignored(f.1)).enumerate().map(|(enum_n, (field_n, field))| { let inject_node_event_stmts = data_struct_fields.iter().enumerate().enumerate().map(|(enum_n, (field_n, field))| {
let mut elem = if enum_n != 0 { let mut elem = if enum_n != 0 {
quote!{ #either_ident::Second(ev) } quote!{ #either_ident::Second(ev) }
} else { } else {
quote!{ ev } quote!{ ev }
}; };
for _ in 0 .. data_struct.fields.iter().filter(|f| !is_ignored(f)).count() - 1 - enum_n { for _ in 0 .. data_struct_fields.len() - 1 - enum_n {
elem = quote!{ #either_ident::First(#elem) }; elem = quote!{ #either_ident::First(#elem) };
} }
@ -434,10 +373,7 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// The [`ConnectionHandler`] associated type. // The [`ConnectionHandler`] associated type.
let protocols_handler_ty = { let protocols_handler_ty = {
let mut ph_ty = None; let mut ph_ty = None;
for field in data_struct.fields.iter() { for field in data_struct_fields.iter() {
if is_ignored(field) {
continue;
}
let ty = &field.ty; let ty = &field.ty;
let field_info = quote! { <#ty as #trait_to_impl>::ConnectionHandler }; let field_info = quote! { <#ty as #trait_to_impl>::ConnectionHandler };
match ph_ty { match ph_ty {
@ -454,11 +390,7 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
let new_handler = { let new_handler = {
let mut out_handler = None; let mut out_handler = None;
for (field_n, field) in data_struct.fields.iter().enumerate() { for (field_n, field) in data_struct_fields.iter().enumerate() {
if is_ignored(field) {
continue;
}
let field_name = match field.ident { let field_name = match field.ident {
Some(ref i) => quote! { self.#i }, Some(ref i) => quote! { self.#i },
None => quote! { self.#field_n }, None => quote! { self.#field_n },
@ -505,7 +437,7 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
// List of statements to put in `poll()`. // List of statements to put in `poll()`.
// //
// We poll each child one by one and wrap around the output. // We poll each child one by one and wrap around the output.
let poll_stmts = data_struct.fields.iter().enumerate().filter(|f| !is_ignored(f.1)).enumerate().map(|(enum_n, (field_n, field))| { let poll_stmts = data_struct_fields.iter().enumerate().enumerate().map(|(enum_n, (field_n, field))| {
let field_name = match field.ident { let field_name = match field.ident {
Some(ref i) => quote!{ self.#i }, Some(ref i) => quote!{ self.#i },
None => quote!{ self.#field_n }, None => quote!{ self.#field_n },
@ -516,7 +448,7 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
} else { } else {
quote!{ event } quote!{ event }
}; };
for _ in 0 .. data_struct.fields.iter().filter(|f| !is_ignored(f)).count() - 1 - enum_n { for _ in 0 .. data_struct_fields.len() - 1 - enum_n {
wrapped_event = quote!{ #either_ident::First(#wrapped_event) }; wrapped_event = quote!{ #either_ident::First(#wrapped_event) };
} }
@ -527,10 +459,7 @@ fn build_struct(ast: &DeriveInput, data_struct: &DataStruct) -> TokenStream {
let provided_handler_and_new_handlers = { let provided_handler_and_new_handlers = {
let mut out_handler = None; let mut out_handler = None;
for (f_n, f) in data_struct.fields.iter().enumerate() { for (f_n, f) in data_struct_fields.iter().enumerate() {
if is_ignored(f) {
continue;
}
let f_name = match f.ident { let f_name = match f.ident {
Some(ref i) => quote! { self.#i }, Some(ref i) => quote! { self.#i },

27
swarm-derive/tests/test.rs
View File

@ -419,3 +419,30 @@ fn no_event_with_either() {
require_net_behaviour::<Foo>(); require_net_behaviour::<Foo>();
} }
} }
#[test]
fn mixed_field_order() {
struct Foo {}
#[derive(NetworkBehaviour)]
#[behaviour(event_process = true)]
pub struct Behaviour {
#[behaviour(ignore)]
_foo: Foo,
_ping: libp2p::ping::Ping,
#[behaviour(ignore)]
_foo2: Foo,
_identify: libp2p::identify::Identify,
#[behaviour(ignore)]
_foo3: Foo,
}
impl<T> libp2p::swarm::NetworkBehaviourEventProcess<T> for Behaviour {
fn inject_event(&mut self, _evt: T) {}
}
#[allow(dead_code)]
fn behaviour() {
require_net_behaviour::<Behaviour>();
}
}