mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-09-20 17:30:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fixed #29339 -- Added result caching to RawQuerySet.

Browse source
This commit is contained in:
Adnan Umer 2018-04-18 22:25:59 +05:00 • committed by Tim Graham
parent c1c163b427
commit ec0319ff82
4 changed files with 25 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
django/db/models/query.py
View file

@ -1277,6 +1277,7 @@ class RawQuerySet:
self.query = query or sql.RawQuery(sql=raw_query, using=self.db, params=params)
self.params = params or ()
self.translations = translations or {}
self._result_cache = None
def resolve_model_init_order(self):
"""Resolve the init field names and value positions."""
@ -1288,7 +1289,15 @@ class RawQuerySet:
model_init_names = [f.attname for f in model_init_fields]
return model_init_names, model_init_order, annotation_fields
def _fetch_all(self):
if self._result_cache is None:
self._result_cache = list(self.iterator())
def __iter__(self):
self._fetch_all()
return iter(self._result_cache)
def iterator(self):
# Cache some things for performance reasons outside the loop.
db = self.db
compiler = connections[db].ops.compiler('SQLCompiler')(