Upgrade cache in module trace file

onflow · Sep 9, 2023 · 576148c · 576148c
1 parent 6104d0c
commit 576148c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/module/trace/trace.go b/module/trace/trace.go
@@ -41,7 +41,7 @@ type Tracer struct {
 	tracer      trace.Tracer
 	shutdown    func(context.Context) error
 	log         zerolog.Logger
-	spanCache   *lru.Cache
+	spanCache   *lru.Cache[flow.Identifier, trace.Span]
 	chainID     string
 	sensitivity uint
 }
@@ -88,7 +88,7 @@ func NewTracer(
 		log.Debug().Err(err).Msg("tracing error")
 	}))
 
-	spanCache, err := lru.New(int(DefaultEntityCacheSize))
+	spanCache, err := lru.New[flow.Identifier, trace.Span](int(DefaultEntityCacheSize))
 	if err != nil {
 		return nil, err
 	}