Skip to content

Commit 8244950

Browse files
committed
RHINENG-17708 unleash client setup
1 parent 16d09c7 commit 8244950

File tree

2 files changed

+24
-0
lines changed

2 files changed

+24
-0
lines changed

ros/lib/config.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,11 @@ def build_endpoint_url(ep):
122122
AWS_REGION_NAME = os.getenv("AWS_REGION_NAME", None)
123123
AWS_LOG_GROUP = os.getenv("AWS_LOG_GROUP", None)
124124

125+
# Feature flags
126+
UNLEASH_CACHE_DIR = os.getenv("UNLEASH_CACHE_DIR", "/tmp/.unleashcache")
127+
UNLEASH_TOKEN = os.getenv("UNLEASH_TOKEN", 'ros:dev.token')
128+
UNLEASH_URL = os.getenv("UNLEASH_URL", "http://localhost:3063/api")
129+
125130
DB_URI = f"postgresql://{DB_USER}:{DB_PASSWORD}"\
126131
f"@{DB_HOST}:{DB_PORT}/{DB_NAME}"
127132
if DB_SSL_CERTPATH:
@@ -155,6 +160,7 @@ def build_endpoint_url(ep):
155160
os.getenv("CACHE_TIMEOUT_FOR_DELETED_SYSTEM", "86400"))
156161
CACHE_KEYWORD_FOR_DELETED_SYSTEM = '_del_'
157162
POLL_TIMEOUT_SECS = 1.0
163+
UNLEASH_ROS_V2_FLAG = 'ros.v2'
158164

159165

160166
KESSEL_ENABLED = str_to_bool(os.getenv("KESSEL_ENABLED", "False"))

ros/lib/unleash.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
from UnleashClient import UnleashClient
2+
from .config import UNLEASH_URL, UNLEASH_TOKEN, APP_NAME, UNLEASH_CACHE_DIR, get_logger
3+
import logging
4+
5+
logger = get_logger(__name__)
6+
7+
logging.getLogger('UnleashClient').setLevel(logging.WARNING)
8+
logging.getLogger('apscheduler').setLevel(logging.WARNING)
9+
10+
unleash_client = UnleashClient(
11+
url=UNLEASH_URL,
12+
app_name=APP_NAME,
13+
custom_headers={"Authorization": UNLEASH_TOKEN},
14+
cache_directory=UNLEASH_CACHE_DIR
15+
)
16+
17+
unleash_client.initialize_client()
18+
logger.info(f"Unleash client initialized: {unleash_client.is_initialized}")

0 commit comments

Comments
 (0)