changes to template

Author: doppleware

gringotts/api/vault_service.py

@@ -55,6 +55,11 @@ async def authenticate_vault_key(payload: AuthenticationRequest, db_session: Asy
     return token_response
 
 
+
+
+
+
+
 @router.get("/appraisal", response_model=VaultBalanceResponse, status_code=status.HTTP_200_OK)
 async def get_vault_appraisal(vault_id: str, muggle_currency_code: Optional[str] = None,
                               db_session: AsyncSession = Depends(get_db),
@@ -81,6 +86,15 @@ async def start_new_vault_appraisal(payload: VaultAppraisalRequest,
     await request_vault_appraisal(payload.vault_id)
 
 
+
+
+
+
+
+
+
+
+
 async def _ensure_authorized_to_vault(db_session, owner_access: VaultOwnerAccess, vault_id):
     try:
         await authorize_vault_owner_vault_access(db_session,owner_access.vault_owner,vault_id)

gringotts/domain/vault_authorization.py

@@ -43,7 +43,19 @@ async def authorize_vault_owner_vault_access(db_session: AsyncSession, vault_own
         await _ensure_owns_requested_vault(owner, vault_id)
 
 
+
+
+
+
+
+
+
+
+
+
+
 async def authenticate_vault_owner_and_key(db_session: AsyncSession, vault_owner: str, vault_key: str ):
+   
     with tracer.start_as_current_span("Authenticate vault owner and key"):
         # Owner is known
         owner = await _ensure_owner_exists(db_session, vault_owner)
@@ -78,14 +90,17 @@ async def _ensure_key_matches_records(db_session, vault_key: str):
 
 async def _ensure_owner_has_a_vault(owner):
     # Vault requested is the one that belongs to the requestor
-    vault_id = owner.vault_id
-    if not vault_id:
-        raise CreatureNotAuthenticatedException(f"Specified vault_id {vault_id} doesn't exist")
-    return vault_id
+    with tracer.start_as_current_span("Ensure_owner_has_a_vault"):
+        vault_id = owner.vault_id
+        await asyncio.sleep(40)
+        if not vault_id:
+            raise CreatureNotAuthenticatedException(f"Specified vault_id {vault_id} doesn't exist")
+        return vault_id
 
 
 async def _ensure_owner_exists(db_session, vault_owner:str):
     owner: VaultOwner = await VaultOwner.find(username=vault_owner, db_session=db_session)
+    print("owner" + owner.name)
     if not owner:
         raise CreatureNotAuthenticatedException(f"Creature {vault_owner} isn't a registered owner")
     return owner

gringotts/main.py

@@ -29,7 +29,7 @@
 
 resource = Resource.create(attributes={SERVICE_NAME: 'vault_service'})
 resource = DigmaConfiguration().trace_this_package()\
-                    .set_environment("Dev").resource.merge(resource)
+                    .set_environment("Prod").resource.merge(resource)
 exporter = OTLPSpanExporter(endpoint=get_settings().otlp_exporter_url, insecure=True)
 provider = TracerProvider(resource=resource)
 provider.add_span_processor(BatchSpanProcessor(exporter))

observability/tracing/docker-compose.trace.yml

@@ -45,7 +45,7 @@ services:
     depends_on:
       - jaeger
     environment:
-      - OTLP_EXPORTER_DIGMA_COLLECTOR_API=digma-collector-api:5050
+      - OTLP_EXPORTER_DIGMA_COLLECTOR_API=host.docker.internal:5050
 
 networks:
   default: