File size: 1,243 Bytes
94337ad ce96dae 94337ad ce96dae 94337ad ce96dae 94337ad ce96dae 94337ad ce96dae 94337ad ce96dae 94337ad ce96dae |
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 |
# app/api/traffic_daemon.py
import asyncio
import pandas as pd
import random
import os
import httpx
API_URL = "http://localhost:8000/predict"
SOURCE_DATA = "data/processed/current_data.csv"
MIN_SLEEP = 2
MAX_SLEEP = 8
MIN_BATCH = 1
MAX_BATCH = 5
STARTUP_DELAY = 7
async def traffic_loop():
await asyncio.sleep(STARTUP_DELAY)
if not os.path.exists(SOURCE_DATA):
print("Traffic daemon: source data not found, disabled.")
return
df = pd.read_csv(SOURCE_DATA)
print("Traffic daemon started.")
async with httpx.AsyncClient(timeout=60.0) as client:
while True:
try:
batch_size = random.randint(MIN_BATCH, MAX_BATCH)
sample = df.sample(batch_size)
csv_bytes = sample.to_csv(index=False).encode("utf-8")
resp = await client.post(
API_URL,
files={"file": ("sample.csv", csv_bytes, "text/csv")}
)
if resp.status_code != 200:
print("Traffic daemon warning:", resp.status_code)
except Exception as e:
print("Traffic daemon error:", e)
await asyncio.sleep(random.uniform(MIN_SLEEP, MAX_SLEEP))
|