Skip to content

Commit d8355fe

Browse files
committed
fix: naming in trace
1 parent 00787b0 commit d8355fe

File tree

2 files changed

+24
-21
lines changed

2 files changed

+24
-21
lines changed

namespace.go

+5-5
Original file line numberDiff line numberDiff line change
@@ -336,9 +336,9 @@ func (n Namespace[T]) deduplicateLoadFromOrigin(ctx context.Context, ns types.TN
336336

337337
defer n.revalidating.Delete(revalidateKey)
338338

339-
ctx, span = telemetry.NewSpan(ctx, "namespace.refreshFromOrigin")
339+
_, span2 := telemetry.NewSpan(ctx, "namespace.refreshFromOrigin")
340340
value, err := refreshFromOrigin(key)
341-
span.End()
341+
span2.End()
342342

343343
// Send the result through the channel
344344
future <- deduplicateEntry[T]{value, err}
@@ -364,13 +364,13 @@ func (n Namespace[T]) deduplicateLoadFromOriginMany(ctx context.Context, ns type
364364
n.revalidating.Store(revalidateKey, future)
365365
defer n.revalidating.Delete(revalidateKey)
366366

367-
ctx, span = telemetry.NewSpan(ctx, "namespace.refreshFromOrigin")
368-
telemetry.WithAttributes(span,
367+
_, span2 := telemetry.NewSpan(ctx, "namespace.refreshFromOrigin")
368+
telemetry.WithAttributes(span2,
369369
telemetry.AttributeKV{Key: "keys", Value: keys},
370370
telemetry.AttributeKV{Key: "namespace", Value: string(n.ns)},
371371
)
372372
values, err := refreshFromOrigin(keys)
373-
span.End()
373+
span2.End()
374374

375375
// Send the result through the channel
376376
future <- deduplicateManyEntry[T]{values, err}

tiered.go

+19-16
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ func getStaleFreshTime(now time.Time, freshDuration time.Duration, staleDuration
5252
}
5353

5454
func (t tieredCache[T]) Get(ctx context.Context, ns types.TNamespace, key string) (*types.TValue, bool, error) {
55-
ctx, span := telemetry.NewSpan(ctx, "tiered.get-many")
55+
ctx, span := telemetry.NewSpan(ctx, "tiered.get")
5656
defer span.End()
5757

5858
if len(t.stores) == 0 {
@@ -80,7 +80,7 @@ func (t tieredCache[T]) Get(ctx context.Context, ns types.TNamespace, key string
8080
continue
8181
}
8282

83-
_, span := telemetry.NewSpan(ctx, store.Name()+".set")
83+
_, span := telemetry.NewSpan(ctx, lowerStore.Name()+".set")
8484
defer span.End()
8585
telemetry.WithAttributes(span,
8686
telemetry.AttributeKV{Key: "key", Value: key},
@@ -167,7 +167,7 @@ func (t tieredCache[T]) GetMany(ctx context.Context, ns types.TNamespace, keys [
167167
continue
168168
}
169169

170-
_, span := telemetry.NewSpan(ctx, store.Name()+".set-many")
170+
_, span := telemetry.NewSpan(ctx, lowerStore.Name()+".set-many")
171171
defer span.End()
172172
telemetry.WithAttributes(span,
173173
telemetry.AttributeKV{Key: "keys", Value: keysToFind},
@@ -204,6 +204,7 @@ func (t tieredCache[T]) GetMany(ctx context.Context, ns types.TNamespace, keys [
204204
func (t tieredCache[T]) Set(ctx context.Context, ns types.TNamespace, key string, value *T, opts *types.SetOptions) error {
205205
ctx, span := telemetry.NewSpan(ctx, "tiered.set")
206206
defer span.End()
207+
207208
telemetry.WithAttributes(span,
208209
telemetry.AttributeKV{Key: "key", Value: key},
209210
telemetry.AttributeKV{Key: "namespace", Value: string(ns)},
@@ -215,9 +216,9 @@ func (t tieredCache[T]) Set(ctx context.Context, ns types.TNamespace, key string
215216

216217
fresh, stale := getStaleFreshTime(time.Now(), t.fresh, t.stale, opts)
217218
for _, store := range t.stores {
218-
_, span := telemetry.NewSpan(ctx, store.Name()+".set")
219-
defer span.End()
220-
telemetry.WithAttributes(span,
219+
_, span2 := telemetry.NewSpan(ctx, store.Name()+".set")
220+
defer span2.End()
221+
telemetry.WithAttributes(span2,
221222
telemetry.AttributeKV{Key: "key", Value: key},
222223
telemetry.AttributeKV{Key: "namespace", Value: string(ns)},
223224
telemetry.AttributeKV{Key: "fresh", Value: fresh.String()},
@@ -230,10 +231,10 @@ func (t tieredCache[T]) Set(ctx context.Context, ns types.TNamespace, key string
230231
StaleUntil: stale,
231232
Key: key,
232233
}); err != nil {
233-
telemetry.RecordError(span, err)
234+
telemetry.RecordError(span2, err)
234235
return fault.Wrap(err, fmsg.With(store.Name()+" failed to set key: "+key))
235236
}
236-
span.End()
237+
span2.End()
237238
}
238239

239240
return nil
@@ -263,16 +264,17 @@ func (t *tieredCache[T]) SetMany(ctx context.Context, ns types.TNamespace, value
263264
}
264265

265266
for _, store := range t.stores {
266-
_, span := telemetry.NewSpan(ctx, store.Name()+".set-many")
267-
defer span.End()
268-
telemetry.WithAttributes(span,
267+
_, span2 := telemetry.NewSpan(ctx, store.Name()+".set-many")
268+
defer span2.End()
269+
telemetry.WithAttributes(span2,
269270
telemetry.AttributeKV{Key: "namespace", Value: string(ns)},
270271
)
271272

272273
if err := store.SetMany(t.ns, valuesToSet, opts); err != nil {
273-
telemetry.RecordError(span, err)
274+
telemetry.RecordError(span2, err)
274275
return fault.Wrap(err, fmsg.With(store.Name()+" failed to set keys: "))
275276
}
277+
span2.End()
276278
}
277279

278280
return nil
@@ -287,17 +289,18 @@ func (t *tieredCache[T]) Remove(ctx context.Context, ns types.TNamespace, keys [
287289
}
288290

289291
for _, store := range t.stores {
290-
_, span := telemetry.NewSpan(ctx, store.Name()+".remove")
291-
defer span.End()
292-
telemetry.WithAttributes(span,
292+
_, span2 := telemetry.NewSpan(ctx, store.Name()+".remove")
293+
defer span2.End()
294+
telemetry.WithAttributes(span2,
293295
telemetry.AttributeKV{Key: "key", Value: keys},
294296
telemetry.AttributeKV{Key: "namespace", Value: string(ns)},
295297
)
296298

297299
if err := store.Remove(t.ns, keys); err != nil {
298-
telemetry.RecordError(span, err)
300+
telemetry.RecordError(span2, err)
299301
return fault.Wrap(err, fmsg.With(store.Name()+" failed to remove key(s): "+strings.Join(keys, ",")))
300302
}
303+
span2.End()
301304
}
302305

303306
return nil

0 commit comments

Comments
 (0)