tmp 2

Signed-off-by: kvmw <mshamsi@broadcom.com>

Author: kvmw

greeter-messages/app.py

@@ -1,11 +1,12 @@
 import os
-import asyncio
 from flask import Flask, request
-from eureka_client import EurekaServerClient
-
+from eureka import init_client
 
 app = Flask(__name__)
 
+port = int(os.getenv('PORT', '8080'))
+init_client(port)
+
 @app.route('/greeting')
 def greeting():
     salutation = request.args.get('salutation', 'Hello')
@@ -14,7 +15,4 @@ def greeting():
     return f"{salutation}, {name}!"
 
 if __name__ == '__main__':
-    port = int(os.getenv('PORT', '8080'))
-    eureka_client = EurekaServerClient(port)
-    asyncio.run(eureka_client.start())
     app.run(host='0.0.0.0', port=port)

greeter-messages/eureka.py

@@ -0,0 +1,70 @@
+import os
+import urllib.request
+from cfenv import AppEnv
+from oauth2 import OAuth2Client
+import py_eureka_client.eureka_client as eureka_client
+import py_eureka_client.http_client as http_client
+from urllib.error import URLError
+from typing import Union
+
+
+class _OAuth2HttpClient(http_client.HttpClient):
+    """
+    A custom http client that uses OAuth2 to authenticate requests.
+    """
+
+    def __init__(self, credentials: dict):
+        self.oauth_client = OAuth2Client(
+            client_id=credentials['client_id'],
+            client_secret=credentials['client_secret'],
+            access_token_uri=credentials['access_token_uri']
+        )
+
+    async def urlopen(self, request: Union[str, http_client.HttpRequest] = None,
+                      data: bytes = None, timeout: float = None) -> http_client.HttpResponse:
+        if isinstance(request, http_client.HttpRequest):
+            req = request
+        elif isinstance(request, str):
+            req = http_client.HttpRequest(request, headers={'Accept-Encoding': 'gzip'})
+        else:
+            raise URLError("Invalid URL")
+
+        req = req._to_urllib_request()
+        req.add_header("Connection", "close")
+        req.add_header("Authorization", f"Bearer {self.oauth_client.get_access_token()}")
+        res = urllib.request.urlopen(req, data=data, timeout=timeout)
+        return http_client.HttpResponse(res)
+
+def _on_err(err_type: str, err: Exception):
+    if err_type in (eureka_client.ERROR_REGISTER, eureka_client.ERROR_DISCOVER):
+        eureka_client.stop()
+    else:
+        print(f"{err_type}::{err}")
+
+def init_client(port: int):
+    """
+    Initializes the eureka client.
+    """
+
+    env = AppEnv()
+
+    service_registry = env.get_service(label="p.service-registry")
+    if not service_registry:
+        raise ValueError("No 'p.service-registry' service instance found.")
+
+    credentials = service_registry.credentials
+
+    http_client.set_http_client(_OAuth2HttpClient(credentials))
+
+    eureka_client.init(
+                app_name=env.name,
+                instance_host=env.uris[0],
+                instance_ip=os.getenv('CF_INSTANCE_INTERNAL_IP'),
+                instance_id=f"{env.uris[0]}:{env.app.get('instance_id') or '0'}",
+                instance_unsecure_port_enabled=True,
+                instance_port=port,
+                instance_secure_port_enabled=True,
+                instance_secure_port=port,
+                eureka_server=credentials['uri'],
+                on_error=_on_err
+            )