@@ -26,9 +26,10 @@ class CacheType(Enum):
2626DEFAULT_RESOLVER_TYPE = ResolverType .RPC
2727DEFAULT_RETRY_BACKOFF = 1000
2828DEFAULT_RETRY_BACKOFF_MAX = 120000
29- DEFAULT_RETRY_GRACE_ATTEMPTS = 5
29+ DEFAULT_RETRY_GRACE_PERIOD_SECONDS = 5
3030DEFAULT_STREAM_DEADLINE = 600000
3131DEFAULT_TLS = False
32+ DEFAULT_TLS_CERT : typing .Optional [str ] = None
3233
3334ENV_VAR_CACHE_SIZE = "FLAGD_MAX_CACHE_SIZE"
3435ENV_VAR_CACHE_TYPE = "FLAGD_CACHE"
@@ -41,10 +42,11 @@ class CacheType(Enum):
4142ENV_VAR_RESOLVER_TYPE = "FLAGD_RESOLVER"
4243ENV_VAR_RETRY_BACKOFF_MS = "FLAGD_RETRY_BACKOFF_MS"
4344ENV_VAR_RETRY_BACKOFF_MAX_MS = "FLAGD_RETRY_BACKOFF_MAX_MS"
44- ENV_VAR_RETRY_GRACE_ATTEMPTS = "FLAGD_RETRY_GRACE_ATTEMPTS "
45- ENV_VAR_SELECTOR = "FLAGD_SELECTOR "
45+ ENV_VAR_RETRY_GRACE_PERIOD_SECONDS = "FLAGD_RETRY_GRACE_PERIOD "
46+ ENV_VAR_SELECTOR = "FLAGD_SOURCE_SELECTOR "
4647ENV_VAR_STREAM_DEADLINE_MS = "FLAGD_STREAM_DEADLINE_MS"
4748ENV_VAR_TLS = "FLAGD_TLS"
49+ ENV_VAR_TLS_CERT = "FLAGD_SERVER_CERT_PATH"
4850
4951T = typing .TypeVar ("T" )
5052
@@ -83,12 +85,13 @@ def __init__( # noqa: PLR0913
8385 offline_poll_interval_ms : typing .Optional [int ] = None ,
8486 retry_backoff_ms : typing .Optional [int ] = None ,
8587 retry_backoff_max_ms : typing .Optional [int ] = None ,
86- retry_grace_attempts : typing .Optional [int ] = None ,
88+ retry_grace_period : typing .Optional [int ] = None ,
8789 deadline_ms : typing .Optional [int ] = None ,
8890 stream_deadline_ms : typing .Optional [int ] = None ,
8991 keep_alive_time : typing .Optional [int ] = None ,
9092 cache : typing .Optional [CacheType ] = None ,
9193 max_cache_size : typing .Optional [int ] = None ,
94+ cert_path : typing .Optional [str ] = None ,
9295 ):
9396 self .host = env_or_default (ENV_VAR_HOST , DEFAULT_HOST ) if host is None else host
9497
@@ -117,14 +120,16 @@ def __init__( # noqa: PLR0913
117120 else retry_backoff_max_ms
118121 )
119122
120- self .retry_grace_attempts : int = (
123+ self .retry_grace_period : int = (
121124 int (
122125 env_or_default (
123- ENV_VAR_RETRY_GRACE_ATTEMPTS , DEFAULT_RETRY_GRACE_ATTEMPTS , cast = int
126+ ENV_VAR_RETRY_GRACE_PERIOD_SECONDS ,
127+ DEFAULT_RETRY_GRACE_PERIOD_SECONDS ,
128+ cast = int ,
124129 )
125130 )
126- if retry_grace_attempts is None
127- else retry_grace_attempts
131+ if retry_grace_period is None
132+ else retry_grace_period
128133 )
129134
130135 self .resolver = (
@@ -201,6 +206,12 @@ def __init__( # noqa: PLR0913
201206 else max_cache_size
202207 )
203208
209+ self .cert_path = (
210+ env_or_default (ENV_VAR_TLS_CERT , DEFAULT_TLS_CERT )
211+ if cert_path is None
212+ else cert_path
213+ )
214+
204215 self .selector = (
205216 env_or_default (ENV_VAR_SELECTOR , None ) if selector is None else selector
206217 )
0 commit comments