Opened 5 years ago

Last modified 2 weeks ago

#16614 new New feature

Support server-side cursors for queryset iteration in database backends

Reported by: Dan McGee Owned by:
Component: Database layer (models, ORM) Version: master
Severity: Normal Keywords: memory cursors database
Cc: dpmcgee@…, nikolai@…, trbs@…, benth, Simon Charette, macek@…, riccardo@…, axel.rau@…, clokep@…, josh.smeaton@…, olivier.tabone@…, mail@… Triage Stage: Accepted
Has patch: yes Needs documentation: yes
Needs tests: no Patch needs improvement: no
Easy pickings: no UI/UX: no

Description

This is related to concerns raised in #5423 as well as documentation issues noted in #13869.

Attached is a very rough first cut of a possible patch that adds server-side iteration for all backends that need it, as well as turning it on by default in the results_iter() codepaths, which is only used in the iterator() methods of the various QuerySet classes defined in django.db.models.query.

Observations:

Thoughts and feedback are welcome- I can imagine only enabling this by default for PostgreSQL only, as MySQL's implementation leaves something to be desired. I could also see never doing this by default and allowing it to be configured in DATABASE settings.

This was mildy tested with a random dumpdata operation on a random project using PostgreSQL. The max memory used by the dumpdata after applying this patch and the one from FS#5423, piping to /dev/null, went from 50MB to 26MB.

Attachments (1)

support-server-side-cursors.patch (6.7 KB) - added by Dan McGee 5 years ago.

Download all attachments as: .zip

Change History (26)

Changed 5 years ago by Dan McGee

comment:1 Changed 5 years ago by Dan McGee

Cc: dpmcgee@… added

comment:2 Changed 5 years ago by nikolai@…

Cc: nikolai@… added

comment:3 Changed 5 years ago by Aymeric Augustin

Needs documentation: set
Needs tests: set
Triage Stage: UnreviewedAccepted

Based on slides 55-63 of http://thebuild.com/presentations/unbreaking-django.pdf I think it's a good idea.

comment:4 Changed 5 years ago by Anssi Kääriäinen

I did a very similar implementation based on discussions in https://groups.google.com/group/django-developers/browse_thread/thread/f19040e2e3229d7a

The main difference is that my version adds a queryset method .chunked(), which will use named cursor for PostgreSQL and will not fetch all the results in one go for SQLite3. SQLite3 has a problem if not fetching all the results in one go as updates done while iterating the results will be seen in the results.

The idea for the .chunked() method is that it will be documented as having backend-specific limitations which the .iterator() approach does not have. The abovementioned SQLite3 limitation is one, PostgreSQL has at least two:

  • As long as the cursor is open (you have a reference to the iterator in the code), server side resources will be tied. This might be important for cases where you open a lot of cursors and then iterate them in a template.
  • The named cursor is not iterable at all outside of a transaction (or you need to use WITH HOLD cursors, which will tie the resources even longer). This means named cursor will not be usable in autocommit mode.

I did not yet include anything for MySQL. There was some discussions about MySQL and it seems it could have some deadlocking problems.

The above limitations are not backwards compatible with current behavior of .iterator(). So, there might be some reason not to expose this as a default / settings behavior, but as a different queryset method you can use when you really need it. The conditions when you need this are exceptional, in normal HTML generation you will almost never want to use named cursors.

The patch is at: http://www.google.com/url?sa=D&q=https://github.com/akaariai/django/commit/8990e20df50ce110fe6ddbbdfed7a98987bb5835&usg=AFQjCNGtHz9sSHT0tOIWKXAZAypfvPIHyw

comment:5 Changed 4 years ago by anonymous

Has there been any movement on this issue? Do the core developers have any plans to merge this patch in the near future?

comment:6 Changed 4 years ago by Luke Plant

akaariai is now a core developer, and could do this if he is still interested. The idea as described in comment 4 sounds solid to me. It would also be fine to have this implemented for some backends and not others IMO.

The link for the patch no longer works. Hopefully Anssi has a record of it somewhere.

comment:7 Changed 4 years ago by Anssi Kääriäinen

I renamed the repo to django-old when Django was moved to github. Here is a working link: https://github.com/akaariai/django-old/commit/8990e20df50ce110fe6ddbbdfed7a98987bb5835

I can take care of final polish & commit, but I am not too interested in doing a lot of work on this right now. To get this committed making sure the patch works on current git HEAD, and writing some docs & tests is the way to get this into core.

comment:8 Changed 4 years ago by trbs

Cc: trbs@… added

comment:9 Changed 4 years ago by benth

Cc: benth added

comment:11 Changed 4 years ago by Simon Charette

Cc: Simon Charette added

comment:12 Changed 3 years ago by Vlada Macek

Cc: macek@… added

Is it really in stage Accepted? :)

I think the missing server-side cursor support keeps the Django ORM down...

comment:13 Changed 3 years ago by Riccardo Attilio Galli

Cc: riccardo@… added

comment:14 Changed 21 months ago by Asif Saifuddin Auvi

Owner: changed from nobody to Asif Saifuddin Auvi
Status: newassigned
Version: 1.3master

comment:15 Changed 14 months ago by Tim Graham

Owner: Asif Saifuddin Auvi deleted
Status: assignednew

comment:16 Changed 9 months ago by Axel Rau

Cc: axel.rau@… added

comment:17 Changed 6 months ago by Kevin Turner

I've updated akaariai's patch from 2012 to server_side_cursor_16614.

(not yet ready to submit, still wants docs and tests)

comment:18 Changed 6 months ago by Kevin Turner

also, it renames the DatabaseFeatures can_use_chunked_reads to has_safe_chunked_reads, is that a private interface that can be renamed freely, or is it a public interface we need to cautious around?

comment:19 Changed 6 months ago by Carl Meyer

Technically that's not public API and we can change it if we need to, but it does cause problems for third-party database backends (and I think we've said that we'll mention such changes in the release notes?). So we should only change it if it's really valuable to do so. I'm not immediately seeing a gain in clarity from that change that would justify it, but maybe I'm missing some subtle reason why the new name is more accurate? Unless that's so, I'd lean towards leaving it unchanged.

comment:20 Changed 6 months ago by Kevin Turner

Okay, I've added a test, which is better than no tests, but still leaves some to be desired. That is, this test makes sure that a call to chunked() does not fail horribly, but it doesn't test the salient difference between chunked and iterator, that chunked uses a server-side cursor and doesn't load large result sets in to memory all at once.

Other things that it seems like might be valuable to test:

  • what if you have two chunked() queries at once, and alternate taking objects from each of them, do they successfully not get in each other's way?
  • that may be the same as: if you took the _named_cursor_idx out of the chunked_cursor name, what would break?

Considering the API, is it necessary to add a new QuerySet.chunked(), or can we change QuerySet.iterator() to have this behaviour by default? (because it's what I always assumed that iterator did until I ran out of RAM.)

It does change the run-time characteristics to some degree, but I'm having a hard time coming up with a situation where it would really break an existing use case.

comment:21 Changed 2 months ago by Patrick Cloke

Cc: clokep@… added

comment:22 Changed 6 weeks ago by Josh Smeaton

Cc: josh.smeaton@… added

comment:23 Changed 5 weeks ago by Olivier Tabone

Cc: olivier.tabone@… added

comment:24 Changed 5 weeks ago by Josh Smeaton

For what it's worth I'm keen on the postgres implementation using SSC from the iterator method. That is, no need to implement another queryset method to support this. Testing will need to be done with and without transactions though.

comment:25 Changed 2 weeks ago by François Freitag

Cc: mail@… added
Needs tests: unset

I have written a draft for this feature based on previous patch by Anssi and Kevin.
PR

Note: See TracTickets for help on using tickets.
Back to Top