feat: Allow setting max_results for connection fields

.pre-commit-config.yaml

@@ -18,7 +18,7 @@ repos:
       - id: check-xml
       - id: check-symlinks
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.8.6
+    rev: v0.9.1
     hooks:
       - id: ruff-format
       - id: ruff

pyproject.toml

@@ -36,7 +36,7 @@ django = ">=4.2"
 asgiref = ">=3.8"
 django-choices-field = { version = ">=2.2.2", optional = true }
 django-debug-toolbar = { version = ">=3.4", optional = true }
-strawberry-graphql = ">=0.257.0"
+strawberry-graphql = ">=0.258.0"
 django-tree-queries = "^0.19.0"
 
 [tool.poetry.group.dev.dependencies]
@@ -54,7 +54,7 @@ pytest-django = "^4.1.0"
 pytest-mock = "^3.5.1"
 pytest-snapshot = "^0.9.0"
 pytest-watch = "^4.2.0"
-ruff = "^0.8.5"
+ruff = "^0.9.1"
 django-polymorphic = "^3.1.0"
 setuptools = "^75.1.0"
 psycopg2 = "^2.9.9"

strawberry_django/fields/field.py

@@ -767,6 +767,7 @@ def connection(
     metadata: Mapping[Any, Any] | None = None,
     directives: Sequence[object] | None = (),
     extensions: Sequence[FieldExtension] = (),
+    max_results: int | None = None,
     filters: type | None = UNSET,
     order: type | None = UNSET,
     only: TypeOrSequence[str] | None = None,
@@ -795,6 +796,7 @@ def connection(
     metadata: Mapping[Any, Any] | None = None,
     directives: Sequence[object] | None = (),
     extensions: Sequence[FieldExtension] = (),
+    max_results: int | None = None,
     filters: type | None = UNSET,
     order: type | None = UNSET,
     only: TypeOrSequence[str] | None = None,
@@ -821,6 +823,7 @@ def connection(
     metadata: Mapping[Any, Any] | None = None,
     directives: Sequence[object] | None = (),
     extensions: Sequence[FieldExtension] = (),
+    max_results: int | None = None,
     filters: type | None = UNSET,
     order: type | None = UNSET,
     only: TypeOrSequence[str] | None = None,
@@ -892,7 +895,10 @@ def connection(
         https://relay.dev/graphql/connections.htm
 
     """
-    extensions = [*extensions, StrawberryDjangoConnectionExtension()]
+    extensions = [
+        *extensions,
+        StrawberryDjangoConnectionExtension(max_results=max_results),
+    ]
     f = field_cls(
         python_name=None,
         django_name=field_name,

strawberry_django/optimizer.py

@@ -570,6 +570,7 @@ def _optimize_prefetch_queryset(
                     related_field_id=related_field_id,
                     offset=slice_metadata.start,
                     limit=slice_metadata.end - slice_metadata.start,
+                    max_results=connection_extension.max_results,
                 )
             else:
                 mark_optimized = False

strawberry_django/pagination.py

@@ -171,6 +171,7 @@ def apply_window_pagination(
     related_field_id: str,
     offset: int = 0,
     limit: Optional[int] = UNSET,
+    max_results: Optional[int] = None,
 ) -> _QS:
     """Apply pagination using window functions.
 
@@ -211,7 +212,11 @@ def apply_window_pagination(
 
     if limit is UNSET:
         settings = strawberry_django_settings()
-        limit = settings["PAGINATION_DEFAULT_LIMIT"]
+        limit = (
+            max_results
+            if max_results is not None
+            else settings["PAGINATION_DEFAULT_LIMIT"]
+        )
 
     # Limit == -1 means no limit. sys.maxsize is set by relay when paginating
     # from the end to as a way to mimic a "not limit" as well