macros: fix diagnostics of last statement (tokio-rs#5762)

icedrocket · web-flow · commit 076d77c18679 · 2023-06-04T15:10:59.000+02:00
diff --git a/tokio-macros/src/entry.rs b/tokio-macros/src/entry.rs
@@ -354,10 +354,10 @@ fn parse_knobs(mut input: ItemFn, is_test: bool, config: FinalConfig) -> TokenSt
         },
     };
     if let Some(v) = config.worker_threads {
-        rt = quote! { #rt.worker_threads(#v) };
+        rt = quote_spanned! {last_stmt_start_span=> #rt.worker_threads(#v) };
     }
     if let Some(v) = config.start_paused {
-        rt = quote! { #rt.start_paused(#v) };
+        rt = quote_spanned! {last_stmt_start_span=> #rt.start_paused(#v) };
     }
 
     let header = if is_test {

Original file line number	Diff line number	Diff line change
`@@ -354,10 +354,10 @@ fn parse_knobs(mut input: ItemFn, is_test: bool, config: FinalConfig) -> TokenSt`
`354`	`354`	`},`
`355`	`355`	`};`
`356`	`356`	`if let Some(v) = config.worker_threads {`
`357`		`- rt = quote! { #rt.worker_threads(#v) };`
	`357`	`+ rt = quote_spanned! {last_stmt_start_span=> #rt.worker_threads(#v) };`
`358`	`358`	`}`
`359`	`359`	`if let Some(v) = config.start_paused {`
`360`		`- rt = quote! { #rt.start_paused(#v) };`
	`360`	`+ rt = quote_spanned! {last_stmt_start_span=> #rt.start_paused(#v) };`
`361`	`361`	`}`
`362`	`362`
`363`	`363`	`let header = if is_test {`