feat(context): add utils method to remove keys from context #66

src/api/propagation.ts

@@ -28,7 +28,12 @@ import {
   registerGlobal,
   unregisterGlobal,
 } from '../internal/global-utils';
-import { getBaggage, createBaggage, setBaggage } from '../baggage/index';
+import {
+  getBaggage,
+  createBaggage,
+  setBaggage,
+  deleteBaggage,
+} from '../baggage/index';
 
 const API_NAME = 'propagation';
 
@@ -107,6 +112,8 @@ export class PropagationAPI {
 
   public setBaggage = setBaggage;
 
+  public deleteBaggage = deleteBaggage;
+
   private _getGlobalPropagator(): TextMapPropagator {
     return getGlobal(API_NAME) || NOOP_TEXT_MAP_PROPAGATOR;
   }

src/api/trace.ts

@@ -27,6 +27,8 @@ import {
 import { Tracer } from '../trace/tracer';
 import { TracerProvider } from '../trace/tracer_provider';
 import {
+  deleteSpan,
+  deleteSpanContext,
   getSpan,
   getSpanContext,
   setSpan,
@@ -89,6 +91,10 @@ export class TraceAPI {
 
   public isSpanContextValid = isSpanContextValid;
 
+  public deleteSpan = deleteSpan;
+
+  public deleteSpanContext = deleteSpanContext;
+
   public getSpan = getSpan;
 
   public getSpanContext = getSpanContext;

src/baggage/index.ts

@@ -41,6 +41,8 @@ export function createBaggage(
 }
 
 /**
+ * Get baggage from context if its present
+ *
  * @param {Context} Context that manage all context values
  * @returns {Baggage} Extracted baggage from the context
  */
@@ -49,13 +51,24 @@ export function getBaggage(context: Context): Baggage | undefined {
 }
 
 /**
+ * Set a baggage in a new context and returns it
+ *
  * @param {Context} Context that manage all context values
  * @param {Baggage} baggage that will be set in the actual context
  */
 export function setBaggage(context: Context, baggage: Baggage): Context {
   return context.setValue(BAGGAGE_KEY, baggage);
 }
 
+/**
+ * Delete baggage from the given context
+ *
+ * @param {Context} Context context to delete baggage from
+ */
+export function deleteBaggage(context: Context): Context {
+  return context.deleteValue(BAGGAGE_KEY);
+}
+
 /**
  * Create a serializable BaggageEntryMetadata object from a string.
  *

src/trace/context-utils.ts

@@ -19,6 +19,7 @@ import { Context } from '../context/types';
 import { Span } from './span';
 import { SpanContext } from './span_context';
 import { NonRecordingSpan } from './NonRecordingSpan';
+import { INVALID_SPAN_CONTEXT } from './spancontext-utils';
 
 /**
  * span key
@@ -44,6 +45,15 @@ export function setSpan(context: Context, span: Span): Context {
   return context.setValue(SPAN_KEY, span);
 }
 
+/**
+ * Remove current span stored in the context
+ *
+ * @param context context to delete span from
+ */
+export function deleteSpan(context: Context): Context {
+  return context.deleteValue(SPAN_KEY);
+}
+
 /**
  * Wrap span context in a NoopSpan and set as span in a new
  * context
@@ -66,3 +76,12 @@ export function setSpanContext(
 export function getSpanContext(context: Context): SpanContext | undefined {
   return getSpan(context)?.spanContext();
 }
+
+/**
+ * Replace the span context from the context with an invalid span context
+ *
+ * @param context context to remove the span context from
+ */
+export function deleteSpanContext(context: Context): Context {
+  return setSpan(context, new NonRecordingSpan(INVALID_SPAN_CONTEXT));
+}