@@ -3,10 +3,10 @@ use std::future::Future;
33use anyhow:: Result ;
44use rustc_hash:: FxHashSet ;
55use serde:: { Deserialize , Serialize } ;
6- use tracing:: { Instrument , Level , Span } ;
6+ use tracing:: Instrument ;
77use turbo_rcstr:: RcStr ;
88use turbo_tasks:: {
9- FxIndexSet , NonLocalValue , ReadRef , ResolvedVc , TryJoinIterExt , Vc ,
9+ FxIndexSet , NonLocalValue , ReadRef , ResolvedVc , TryJoinIterExt , ValueToString , Vc ,
1010 debug:: ValueDebugFormat ,
1111 graph:: { AdjacencyMap , GraphTraversal , Visit , VisitControlFlow } ,
1212 trace:: TraceRawVcs ,
@@ -121,23 +121,14 @@ pub async fn find_server_entries(
121121 include_traced : bool ,
122122) -> Result < Vc < ServerEntries > > {
123123 async move {
124- let emit_spans = tracing:: enabled!( Level :: INFO ) ;
125124 let graph = AdjacencyMap :: new ( )
126125 . skip_duplicates ( )
127126 . visit (
128127 vec ! [ FindServerEntriesNode :: Internal (
129128 entry,
130- if emit_spans {
131- // INVALIDATION: we don't need to invalidate when the span name changes
132- Some ( entry. ident_string( ) . untracked( ) . await ?)
133- } else {
134- None
135- } ,
129+ entry. ident( ) . to_string( ) . await ?,
136130 ) ] ,
137- FindServerEntries {
138- include_traced,
139- emit_spans,
140- } ,
131+ FindServerEntries { include_traced } ,
141132 )
142133 . await
143134 . completed ( ) ?
@@ -170,7 +161,6 @@ pub async fn find_server_entries(
170161struct FindServerEntries {
171162 /// Whether to walk ChunkingType::Traced references
172163 include_traced : bool ,
173- emit_spans : bool ,
174164}
175165
176166#[ derive(
@@ -187,12 +177,9 @@ struct FindServerEntries {
187177) ]
188178enum FindServerEntriesNode {
189179 ClientReference ,
190- ServerComponentEntry (
191- ResolvedVc < NextServerComponentModule > ,
192- Option < ReadRef < RcStr > > ,
193- ) ,
194- ServerUtilEntry ( ResolvedVc < NextServerUtilityModule > , Option < ReadRef < RcStr > > ) ,
195- Internal ( ResolvedVc < Box < dyn Module > > , Option < ReadRef < RcStr > > ) ,
180+ ServerComponentEntry ( ResolvedVc < NextServerComponentModule > , ReadRef < RcStr > ) ,
181+ ServerUtilEntry ( ResolvedVc < NextServerUtilityModule > , ReadRef < RcStr > ) ,
182+ Internal ( ResolvedVc < Box < dyn Module > > , ReadRef < RcStr > ) ,
196183}
197184
198185impl Visit < FindServerEntriesNode > for FindServerEntries {
@@ -221,7 +208,6 @@ impl Visit<FindServerEntriesNode> for FindServerEntries {
221208 FindServerEntriesNode :: ServerUtilEntry ( module, _) => Vc :: upcast ( * * module) ,
222209 FindServerEntriesNode :: ServerComponentEntry ( module, _) => Vc :: upcast ( * * module) ,
223210 } ;
224- let emit_spans = self . emit_spans ;
225211 async move {
226212 // Pass include_traced to reuse the same cached `primary_chunkable_referenced_modules`
227213 // task result, but the traced references will be filtered out again afterwards.
@@ -249,13 +235,7 @@ impl Visit<FindServerEntriesNode> for FindServerEntries {
249235 {
250236 return Ok ( FindServerEntriesNode :: ServerComponentEntry (
251237 server_component_asset,
252- if emit_spans {
253- // INVALIDATION: we don't need to invalidate when the span name
254- // changes
255- Some ( server_component_asset. ident_string ( ) . untracked ( ) . await ?)
256- } else {
257- None
258- } ,
238+ server_component_asset. ident ( ) . to_string ( ) . await ?,
259239 ) ) ;
260240 }
261241
@@ -264,24 +244,13 @@ impl Visit<FindServerEntriesNode> for FindServerEntries {
264244 {
265245 return Ok ( FindServerEntriesNode :: ServerUtilEntry (
266246 server_util_module,
267- if emit_spans {
268- // INVALIDATION: we don't need to invalidate when the span name
269- // changes
270- Some ( module. ident_string ( ) . untracked ( ) . await ?)
271- } else {
272- None
273- } ,
247+ module. ident ( ) . to_string ( ) . await ?,
274248 ) ) ;
275249 }
276250
277251 Ok ( FindServerEntriesNode :: Internal (
278252 * module,
279- if emit_spans {
280- // INVALIDATION: we don't need to invalidate when the span name changes
281- Some ( module. ident_string ( ) . untracked ( ) . await ?)
282- } else {
283- None
284- } ,
253+ module. ident ( ) . to_string ( ) . await ?,
285254 ) )
286255 } ) ;
287256
@@ -292,21 +261,18 @@ impl Visit<FindServerEntriesNode> for FindServerEntries {
292261 }
293262
294263 fn span ( & mut self , node : & FindServerEntriesNode ) -> tracing:: Span {
295- if !self . emit_spans {
296- return Span :: current ( ) ;
297- }
298264 match node {
299265 FindServerEntriesNode :: ClientReference => {
300266 tracing:: info_span!( "client reference" )
301267 }
302268 FindServerEntriesNode :: Internal ( _, name) => {
303- tracing:: info_span!( "module" , name = display( name. as_ref ( ) . unwrap ( ) ) )
269+ tracing:: info_span!( "module" , name = display( name) )
304270 }
305271 FindServerEntriesNode :: ServerUtilEntry ( _, name) => {
306- tracing:: info_span!( "server util" , name = display( name. as_ref ( ) . unwrap ( ) ) )
272+ tracing:: info_span!( "server util" , name = display( name) )
307273 }
308274 FindServerEntriesNode :: ServerComponentEntry ( _, name) => {
309- tracing:: info_span!( "layout segment" , name = display( name. as_ref ( ) . unwrap ( ) ) )
275+ tracing:: info_span!( "layout segment" , name = display( name) )
310276 }
311277 }
312278 }
0 commit comments