feat: implement analytics logic with celery, improve tests, implement

locust and docker

Author: volychev

.env.example

@@ -1,9 +1,17 @@
 # Common
 DEBUG=True
+HOST=0.0.0.0
+PORT=8000
+CELERY_RESULT_EXPIRES=43200
 
 # PostgreSQL
 POSTGRES_USER=
 POSTGRES_PASSWORD=
 POSTGRES_HOST=
 POSTGRES_PORT=
-POSTGRES_DATABASE=
+POSTGRES_DB=
+
+# Redis
+REDIS_HOST=
+REDIS_PORT=
+REDIS_DB=

.gitignore

@@ -1,3 +1,5 @@
+*.db
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

Dockerfile

@@ -1,9 +1,7 @@
 FROM python:3.13.11
-WORKDIR /usr/src/python_template
+WORKDIR /usr/src/telemetry_api
 COPY . .
 
 RUN python -m pip install --upgrade pip
 RUN python -m pip install poetry
 RUN python -m poetry install --without dev
-
-# CMD [ "some", "command" ]

docker-compose.yml

@@ -0,0 +1,58 @@
+services:
+  telemetry-api:
+    build: .
+    env_file: ".env"
+    container_name: "telemetry-api"
+    command: >
+      poetry run uvicorn telemetry_api.main:app
+      --host ${HOST}
+      --port ${PORT}
+      --reload
+    ports:
+      - "${PORT}:${PORT}"
+    depends_on:
+      - "telemetry-postgres"
+      - "telemetry-redis"
+    networks:
+      - app-net
+    restart: always
+
+  telemetry-postgres:
+    image: "postgres:16"
+    env_file: ".env"
+    container_name: "telemetry-postgres"
+    volumes:
+      - "telemetry-postgres:/var/lib/postgresql/data/"
+    networks:
+      - app-net
+    restart: always
+
+  telemetry-redis:
+    image: "redis:7-alpine"
+    container_name: "telemetry-redis"
+    command: "redis-server --save 60 1 --loglevel warning"
+    volumes:
+      - "telemetry-redis:/data"
+    networks:
+      - app-net
+    restart: always
+
+  telemetry-celery-worker:
+    build: .
+    env_file: ".env"
+    container_name: "telemetry-celery-worker"
+    command: "poetry run celery -A telemetry_api.worker.celery_app:celery_instance worker --loglevel=INFO"
+    depends_on:
+      - "telemetry-postgres"
+      - "telemetry-redis"
+    networks:
+      - app-net
+    restart: always
+
+volumes:
+  telemetry-postgres:
+  telemetry-redis:
+
+networks:
+  app-net:
+    external: true

locustfile.py

@@ -0,0 +1,110 @@
+import uuid
+
+from locust import HttpUser, between, task
+
+
+class TelemetryApiUser(HttpUser):
+    wait_time = between(0.5, 3.0)
+
+    def on_start(self):
+        self.user_id = self._create_user()
+        self.device_id = self._create_device(self.user_id)
+        self._add_measurement(self.device_id)
+
+    def _create_user(self) -> int:
+        unique_suffix = uuid.uuid4().hex[:8]\
+        payload = {
+            "username": f"user-{unique_suffix}",
+            "email": f"{unique_suffix}@example.com",
+        }
+
+        with self.client.post(
+            "/users/",
+            json=payload,
+            name="POST /users/",
+            catch_response=True
+        ) as response:
+            if response.status_code != 201:
+                response.failure(f"Unexpected status: {response.status_code}, body={response.text}")
+                return -1
+
+            data = response.json()
+            return int(data["id"])
+
+    def _create_device(self, user_id: int) -> int:
+        payload = {
+            "name": f"device-{uuid.uuid4().hex[:6]}",
+            "user_id": user_id,
+        }
+
+        with self.client.post(
+            "/devices/",
+            json=payload,
+            name="POST /devices/",
+            catch_response=True
+        ) as response:
+            if response.status_code != 201:
+                response.failure(f"Unexpected status: {response.status_code}, body={response.text}")
+                return -1
+
+            data = response.json()
+            return int(data["id"])
+
+    def _add_measurement(self, device_id: int):
+        payload = {
+            "x": 1.0,
+            "y": 2.0,
+            "z": 3.0,
+        }
+
+        with self.client.post(
+            f"/analytics/{device_id}/data",
+            json=payload,
+            name="POST /analytics/{device_id}/data",
+            catch_response=True,
+        ) as response:
+            if response.status_code != 201:
+                response.failure(f"Unexpected status: {response.status_code}, body={response.text}")
+
+    @task(5)
+    def add_measurement(self):
+        self._add_measurement(self.device_id)
+
+    @task(3)
+    def get_device_analytics(self):
+        params = {
+            "device_id": self.device_id,
+            "limit": 25,
+            "offset": 0,
+        }
+
+        with self.client.get("/analytics/", params=params, name="GET /analytics/?device_id", catch_response=True) as response:
+            if response.status_code != 200:
+                response.failure(f"Unexpected status: {response.status_code}, body={response.text}")
+                return
+
+            payload = response.json()
+            if "data" not in payload:
+                response.failure("Response without data field")
+
+    @task(2)
+    def get_users(self):
+        with self.client.get(
+            "/users/",
+            params={"limit": 10, "offset": 0},
+            name="GET /users/",
+            catch_response=True,
+        ) as response:
+            if response.status_code != 200:
+                response.failure(f"Unexpected status: {response.status_code}, body={response.text}")
+
+    @task(2)
+    def get_devices(self):
+        with self.client.get(
+            "/devices/",
+            params={"limit": 10, "offset": 0},
+            name="GET /devices/",
+            catch_response=True,
+        ) as response:
+            if response.status_code != 200:
+                response.failure(f"Unexpected status: {response.status_code}, body={response.text}")

pyproject.toml

@@ -17,6 +17,8 @@ aiosqlite = "^0.22.1"
 asyncpg = "^0.31.0"
 greenlet = "^3.5.0"
 email-validator = "^2.3.0"
+celery = {extras = ["redis"], version = "^5.6.3"}
+redis = "^5.2.1"
 
 [tool.poetry.group.dev.dependencies]
 black = "^25.1.0"
@@ -32,6 +34,7 @@ setuptools = "^75.8.0"
 pytest = "^9.0.3"
 pytest-asyncio = "^1.3.0"
 httpx = "^0.28.1"
+locust = "^2.37.10"
 
 [build-system]
 requires = ["poetry-core"]

telemetry_api/api/__init__.py

@@ -5,10 +5,7 @@
 
 def create_api() -> FastAPI:
     """
-    Создание инстенса приложения с подключением всех роутеров
-
-    Return:
-        FastAPI - инстанс приложения
+    Создаёт и настраивает экземпляр FastAPI-приложения.
     """
 
     api = FastAPI(