Add optional context args in Instrument.record/add/set

Author: fcollonval

opentelemetry-api/src/opentelemetry/metrics/_internal/instrument.py

@@ -33,6 +33,7 @@
 
 # pylint: disable=unused-import; needed for typing and sphinx
 from opentelemetry import metrics
+from opentelemetry.context import Context
 from opentelemetry.metrics._internal.observation import Observation
 from opentelemetry.util.types import Attributes
 
@@ -173,6 +174,7 @@ def add(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         pass
 
@@ -192,18 +194,20 @@ def add(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
-        return super().add(amount, attributes=attributes)
+        return super().add(amount, attributes=attributes, context=context)
 
 
 class _ProxyCounter(_ProxyInstrument[Counter], Counter):
     def add(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         if self._real_instrument:
-            self._real_instrument.add(amount, attributes)
+            self._real_instrument.add(amount, attributes, context)
 
     def _create_real_instrument(self, meter: "metrics.Meter") -> Counter:
         return meter.create_counter(self._name, self._unit, self._description)
@@ -217,6 +221,7 @@ def add(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         pass
 
@@ -236,18 +241,20 @@ def add(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
-        return super().add(amount, attributes=attributes)
+        return super().add(amount, attributes=attributes, context=context)
 
 
 class _ProxyUpDownCounter(_ProxyInstrument[UpDownCounter], UpDownCounter):
     def add(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         if self._real_instrument:
-            self._real_instrument.add(amount, attributes)
+            self._real_instrument.add(amount, attributes, context)
 
     def _create_real_instrument(self, meter: "metrics.Meter") -> UpDownCounter:
         return meter.create_up_down_counter(
@@ -328,6 +335,7 @@ def record(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         pass
 
@@ -347,18 +355,20 @@ def record(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
-        return super().record(amount, attributes=attributes)
+        return super().record(amount, attributes=attributes, context=context)
 
 
 class _ProxyHistogram(_ProxyInstrument[Histogram], Histogram):
     def record(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         if self._real_instrument:
-            self._real_instrument.record(amount, attributes)
+            self._real_instrument.record(amount, attributes, context)
 
     def _create_real_instrument(self, meter: "metrics.Meter") -> Histogram:
         return meter.create_histogram(
@@ -406,6 +416,7 @@ def set(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         pass
 
@@ -425,8 +436,9 @@ def set(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
-        return super().set(amount, attributes=attributes)
+        return super().set(amount, attributes=attributes, context=context)
 
 
 class _ProxyGauge(
@@ -437,9 +449,10 @@ def set(
         self,
         amount: Union[int, float],
         attributes: Optional[Attributes] = None,
+        context: Optional[Context] = None,
     ) -> None:
         if self._real_instrument:
-            self._real_instrument.set(amount, attributes)
+            self._real_instrument.set(amount, attributes, context)
 
     def _create_real_instrument(self, meter: "metrics.Meter") -> Gauge:
         return meter.create_gauge(self._name, self._unit, self._description)