File size: 1,708 Bytes
6ca4b94 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import os
import socket
import json
import logging
import httpx
from tenacity import retry, stop_after_attempt, wait_exponential
from src.schemas.search_request import SearchRequest
from src.config import config
logger = logging.getLogger(__name__)
SPACE_ID = config.app.id
API_ENDPOINT = config.tracking_db.api_endpoint
ANON_KEY = config.tracking_db.anon_key
JWT_TOKEN = config.tracking_db.jwt_token
@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=1, max=5))
async def send_tracking_event_to_db(search_data: SearchRequest):
headers = {
"apikey": ANON_KEY.get_secret_value(),
"Authorization": f"Bearer {JWT_TOKEN.get_secret_value()}",
"Content-Type": "application/json",
"Prefer": "return=minimal"
}
data = search_data.model_dump()
session_id = data.pop('session_id')
payload = {
"space_id": SPACE_ID,
"session_id": session_id,
"environment": "dev" if os.getenv("DEV_MODE", "").lower() == "true" else "prod",
"server_host": socket.gethostname(),
"data": data
}
try:
async with httpx.AsyncClient(timeout=10.0) as client:
response = await client.post(API_ENDPOINT, headers=headers, json=payload)
response.raise_for_status()
logger.info("Successfully sent tracking event to DB!")
except httpx.TimeoutException:
logger.warning("Tracking request timed out - continuing anyway")
except httpx.HTTPStatusError as e:
logger.error(f"HTTP {e.response.status_code} error tracking event")
except Exception as e:
logger.error(f"Unexpected error in tracking: {e} with payload: {json.dumps(payload)}") |