prost-build: use parse_syn() over syn::parse_str::<syn::Arm>

tokio-rs · Apr 20, 2024 · 9b28f19 · 9b28f19
1 parent 95065da
commit 9b28f19
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/prost-build/src/code_generator/enums.rs b/prost-build/src/code_generator/enums.rs
@@ -42,20 +42,18 @@ impl CodeGenerator<'_> {
         let enum_variants = resolve_enum_variants(self, &variant_mappings, &fq_proto_enum_name);
         let enum_name_syn = enum_name.parse_syn::<syn::Ident>();
         let arms_1 = variant_mappings.iter().map(|variant| {
-            syn::parse_str::<syn::Arm>(&format!(
+            format!(
                 "{}::{} => \"{}\"",
                 enum_name_syn, variant.generated_variant_name, variant.proto_name
-            ))
-            .expect("unable to parse enum arm")
-            .to_token_stream()
+            )
+            .parse_syn::<syn::Arm>()
         });
         let arms_2 = variant_mappings.iter().map(|variant| {
-            syn::parse_str::<syn::Arm>(&format!(
+            format!(
                 "\"{}\" => Some(Self::{})",
                 variant.proto_name, variant.generated_variant_name
-            ))
-            .expect("unable to parse enum arm")
-            .to_token_stream()
+            )
+            .parse_syn::<syn::Arm>()
         });
 
         return Some(quote! {

diff --git a/prost-build/src/code_generator/mod.rs b/prost-build/src/code_generator/mod.rs
@@ -12,7 +12,7 @@ use prost_types::{
     DescriptorProto, EnumDescriptorProto, FieldDescriptorProto, FieldOptions, FileDescriptorProto,
     OneofDescriptorProto, ServiceDescriptorProto, SourceCodeInfo,
 };
-use quote::{quote, ToTokens};
+use quote::quote;
 use syn::{Attribute, TypePath};
 
 use crate::ast::{Comments, Method, Service};