feat: implement provider status

Author: federicobond

openfeature/provider/__init__.py

@@ -1,4 +1,5 @@
 import typing
+from enum import Enum
 
 from openfeature.evaluation_context import EvaluationContext
 from openfeature.flag_evaluation import FlagResolutionDetails
@@ -7,6 +8,14 @@
 from .metadata import Metadata
 
 
+class ProviderStatus(Enum):
+    NOT_READY = "NOT_READY"
+    READY = "READY"
+    ERROR = "ERROR"
+    STALE = "STALE"
+    FATAL = "FATAL"
+
+
 class FeatureProvider(typing.Protocol):  # pragma: no cover
     def initialize(self, evaluation_context: EvaluationContext) -> None:
         ...

openfeature/provider/registry.py

@@ -2,17 +2,20 @@
 
 from openfeature.evaluation_context import EvaluationContext
 from openfeature.exception import GeneralError
-from openfeature.provider import FeatureProvider
+from openfeature.provider import FeatureProvider, ProviderStatus
 from openfeature.provider.no_op_provider import NoOpProvider
 
 
 class ProviderRegistry:
     _default_provider: FeatureProvider
     _providers: typing.Dict[str, FeatureProvider]
+    _provider_status: typing.Dict[FeatureProvider, ProviderStatus]
 
     def __init__(self) -> None:
         self._default_provider = NoOpProvider()
         self._providers = {}
+        self._provider_status = {}
+        self._set_provider_status(self._default_provider, ProviderStatus.NOT_READY)
 
     def set_provider(self, domain: str, provider: FeatureProvider) -> None:
         if provider is None:
@@ -22,9 +25,9 @@ def set_provider(self, domain: str, provider: FeatureProvider) -> None:
             old_provider = providers[domain]
             del providers[domain]
             if old_provider not in providers.values():
-                old_provider.shutdown()
+                self._shutdown_provider(old_provider)
         if provider not in providers.values():
-            provider.initialize(self._get_evaluation_context())
+            self._initialize_provider(provider)
         providers[domain] = provider
 
     def get_provider(self, domain: typing.Optional[str]) -> FeatureProvider:
@@ -36,9 +39,9 @@ def set_default_provider(self, provider: FeatureProvider) -> None:
         if provider is None:
             raise GeneralError(error_message="No provider")
         if self._default_provider:
-            self._default_provider.shutdown()
+            self._shutdown_provider(self._default_provider)
         self._default_provider = provider
-        provider.initialize(self._get_evaluation_context())
+        self._initialize_provider(provider)
 
     def get_default_provider(self) -> FeatureProvider:
         return self._default_provider
@@ -50,10 +53,31 @@ def clear_providers(self) -> None:
 
     def shutdown(self) -> None:
         for provider in {self._default_provider, *self._providers.values()}:
-            provider.shutdown()
+            self._shutdown_provider(provider)
 
     def _get_evaluation_context(self) -> EvaluationContext:
         # imported here to avoid circular imports
         from openfeature.api import get_evaluation_context
 
         return get_evaluation_context()
+
+    def _initialize_provider(self, provider: FeatureProvider) -> None:
+        try:
+            if hasattr(provider, "initialize"):
+                provider.initialize(self._get_evaluation_context())
+            self._set_provider_status(provider, ProviderStatus.READY)
+        except Exception:
+            self._set_provider_status(provider, ProviderStatus.ERROR)
+
+    def _shutdown_provider(self, provider: FeatureProvider) -> None:
+        try:
+            if hasattr(provider, "shutdown"):
+                provider.shutdown()
+            self._set_provider_status(provider, ProviderStatus.NOT_READY)
+        except Exception:
+            self._set_provider_status(provider, ProviderStatus.FATAL)
+
+    def _set_provider_status(
+        self, provider: FeatureProvider, status: ProviderStatus
+    ) -> None:
+        self._provider_status[provider] = status