refactor: benchmark

.gitignore

@@ -5,3 +5,5 @@
 *.pyd
 *.swp
 *.log
+
+.env

Dockerfile

@@ -0,0 +1,23 @@
+FROM python:3.11-slim as base
+
+RUN apt update && apt install -y git && rm -rf /var/lib/apt/lists/*
+
+RUN pip install pip --upgrade
+
+WORKDIR /app
+
+CMD ["uvicorn", "octopus.asgi:application", "--host", "0.0.0.0", "--port", "8000"]
+
+# graphene-django main version
+FROM base as sync
+
+COPY project/sync /app/
+
+RUN pip install -r requirements.txt
+
+# graphene-django PR version
+FROM base as async
+
+COPY project/async /app/
+
+RUN pip install -r requirements.txt

Makefile

@@ -0,0 +1,19 @@
+up:
+	docker compose up -d --build
+.PHONY: up
+
+down:
+	docker compose down
+.PHONY: down
+
+sync_benchmark_main:
+	cd tests && ./run-main-sync.sh
+.PHONY: sync_benchmark_main
+
+sync_benchmark_PR:
+	cd tests && ./run-PR-sync.sh
+.PHONY: sync_benchmark_PR
+
+async_benchmark_PR:
+	cd tests && ./run-PR-async.sh
+.PHONY: async_benchmark_PR

docker-compose.yml

@@ -0,0 +1,47 @@
+version: "3.8"
+
+services:
+  postgres:
+    image: postgres:16.1
+    container_name: graphene-django-benchmark-postgres
+    environment:
+      POSTGRES_DB: benchmark
+      POSTGRES_USER: user
+      POSTGRES_PASSWORD: password
+    healthcheck:
+      test: [ "CMD-SHELL", "pg_isready -d benchmark -U user"]
+      interval: 1s
+      timeout: 1s
+      retries: 10
+
+  population:
+    build:
+      context: .
+      target: sync
+    container_name: graphene-django-benchmark-population
+    command: bash -c "python manage.py migrate && python manage.py generate"
+    depends_on:
+      postgres:
+        condition: service_healthy
+
+  service-graphene-django-sync:
+    build:
+      context: .
+      target: sync
+    container_name: graphene-django-benchmark-service-graphene-django-sync
+    depends_on:
+      population:
+        condition: service_completed_successfully
+    ports:
+      - "8000:8000"
+
+  service-graphene-django-async:
+    build:
+      context: .
+      target: async
+    container_name: graphene-django-benchmark-service-graphene-django-async
+    depends_on:
+      population:
+        condition: service_completed_successfully
+    ports:
+      - "8001:8000"

project/api/schema_async/queries.py → project/async/api/schema_async/queries.py

@@ -51,7 +51,7 @@ async def resolve_octopuses(self, info, **kwargs):
             .prefetch_related('seafood')
             .all()
         )
-        async for i in qs[:10]:
+        async for i in qs[:500]:
             yield i
 
     async def resolve_octopus(self, info, id):

project/async/api/schema_sync/__init__.py

@@ -0,0 +1,10 @@
+import graphene
+
+from api.schema_sync.queries import OctopusQuery
+
+
+class Query(OctopusQuery):
+    pass
+
+
+schema_sync = graphene.Schema(query=Query)