diff options
author | Lizzy Fleckenstein <eliasfleckenstein@web.de> | 2023-02-06 16:30:55 +0100 |
---|---|---|
committer | Lizzy Fleckenstein <eliasfleckenstein@web.de> | 2023-02-06 16:50:23 +0100 |
commit | dbe6934449b4db1162079cbc39d89d51728bdda1 (patch) | |
tree | f1bef1cff37aea91c74d5702847610bbb5cb3c58 /derive-macro/src/handle_enum.rs | |
parent | 97385f084e01bf935f8040a1616ed465f9d7db11 (diff) | |
download | generate-random-dbe6934449b4db1162079cbc39d89d51728bdda1.tar.xz |
cargo fmt
Diffstat (limited to 'derive-macro/src/handle_enum.rs')
-rw-r--r-- | derive-macro/src/handle_enum.rs | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/derive-macro/src/handle_enum.rs b/derive-macro/src/handle_enum.rs index b2b14c0..8165ab5 100644 --- a/derive-macro/src/handle_enum.rs +++ b/derive-macro/src/handle_enum.rs @@ -1,19 +1,23 @@ +use super::generate_fields; use proc_macro2::{Ident, Literal, TokenStream}; use quote::quote; use syn::{DataEnum, Variant}; -use super::generate_fields; fn variant_weight(variant: &Variant) -> Literal { for attr in variant.attrs.iter() { if attr.path.is_ident("weight") { - return attr.parse_args::<Literal>().expect("expected literal for `#[weight(...)]`") + return attr + .parse_args::<Literal>() + .expect("expected literal for `#[weight(...)]`"); } } Literal::u64_suffixed(1) } pub fn generate(name: &Ident, ty: DataEnum) -> TokenStream { - let mut variant_weights = ty.variants.into_iter() + let mut variant_weights = ty + .variants + .into_iter() .map(|variant| (variant_weight(&variant), variant)); let mut arms = TokenStream::new(); |