return the registered slot from register_slot

census-instrumentation · Mar 20, 2019 · 20b8c8b · 20b8c8b
1 parent ae9f878
commit 20b8c8b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/contrib/opencensus-context/opencensus/common/runtime_context/__init__.py b/contrib/opencensus-context/opencensus/common/runtime_context/__init__.py
@@ -102,7 +102,9 @@ def register_slot(cls, name, default=None):
         with cls._lock:
             if name in cls._slots:
                 raise ValueError('slot {} already registered'.format(name))
-            cls._slots[name] = cls.Slot(name, default)
+            slot = cls.Slot(name, default)
+            cls._slots[name] = slot
+            return slot
 
 
 class _AsyncRuntimeContext(_RuntimeContext):