forked from afair/postgresql_cursor
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'v0.5'
- Loading branch information
Showing
18 changed files
with
797 additions
and
397 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,4 @@ | ||
source 'https://rubygems.org' | ||
|
||
# Specify your gem's dependencies in postgresql_cursor.gemspec | ||
gemspec |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,41 @@ | ||
PATH | ||
remote: . | ||
specs: | ||
postgresql_cursor (0.5.0) | ||
activerecord (>= 3.2.0) | ||
|
||
GEM | ||
remote: https://rubygems.org/ | ||
specs: | ||
activemodel (4.1.1) | ||
activesupport (= 4.1.1) | ||
builder (~> 3.1) | ||
activerecord (4.1.1) | ||
activemodel (= 4.1.1) | ||
activesupport (= 4.1.1) | ||
arel (~> 5.0.0) | ||
activesupport (4.1.1) | ||
i18n (~> 0.6, >= 0.6.9) | ||
json (~> 1.7, >= 1.7.7) | ||
minitest (~> 5.1) | ||
thread_safe (~> 0.1) | ||
tzinfo (~> 1.1) | ||
arel (5.0.1.20140414130214) | ||
builder (3.2.2) | ||
i18n (0.6.9) | ||
json (1.8.1) | ||
minitest (5.3.3) | ||
pg (0.17.1) | ||
rake (10.3.1) | ||
thread_safe (0.3.4) | ||
tzinfo (1.2.1) | ||
thread_safe (~> 0.1) | ||
|
||
PLATFORMS | ||
ruby | ||
|
||
DEPENDENCIES | ||
minitest | ||
pg | ||
postgresql_cursor! | ||
rake |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,185 @@ | ||
#PostgreSQLCursor for handling large Result Sets | ||
|
||
[![Gem Version](https://badge.fury.io/rb/postgresql_cursor.svg)](http://badge.fury.io/rb/postgresql_cursor) | ||
|
||
PostgreSQLCursor extends ActiveRecord to allow for efficient processing of queries | ||
returning a large number of rows, and allows you to sort your result set. | ||
|
||
In PostgreSQL, a | ||
[cursor](http://www.postgresql.org/docs/9.4/static/plpgsql-cursors.html) | ||
runs a query, from which you fetch a block of | ||
(say 1000) rows, process them, and continue fetching until the result | ||
set is exhausted. By fetching a smaller chunk of data, this reduces the | ||
amount of memory your application uses and prevents the potential crash | ||
of running out of memory. | ||
|
||
Version 0.5.0 has been refactored to install more smoothly into ActiveRecord. | ||
It supports Rails and ActiveRecord 3.2.x and up. | ||
|
||
##Use Cursors | ||
|
||
PostgreSQLCursor was developed to take advantage of PostgreSQL's cursors. Cursors allow the program | ||
to declare a cursor to run a given query returning "chunks" of rows to the application program while | ||
retaining the position of the full result set in the database. This overcomes all the disadvantages | ||
of using find_each and find_in_batches. | ||
|
||
Also, with PostgreSQL, you have on option to have raw hashes of the row returned instead of the | ||
instantiated models. An informal benchmark showed that returning instances is a factor of 4 times | ||
slower than returning hashes. If you are can work with the data in this form, you will find better | ||
performance. | ||
|
||
With PostgreSQL, you can work with cursors as follows: | ||
|
||
```ruby | ||
Product.where("id>0").order("name").each_row { |hash| Product.process(hash) } | ||
|
||
Product.where("id>0").each_instance { |product| product.process! } | ||
Product.where("id>0").each_instance(block_size:100_000) { |product| product.process } | ||
|
||
Product.each_row { |hash| Product.process(hash) } | ||
Product.each_instance { |product| product.process } | ||
|
||
Product.each_row_by_sql("select * from products") { |hash| Product.process(hash) } | ||
Product.each_instance_by_sql("select * from products") { |product| product.process } | ||
``` | ||
|
||
###PostgreSQLCursor is an Enumerable | ||
|
||
If you do not pass in a block, the cursor is returned, which mixes in the Enumerable | ||
libary. With that, you can pass it around, or chain in the awesome enumerable things | ||
like `map` and `reduce`. Furthermore, the cursors already act as `lazy`, but you can | ||
also chain in `lazy` when you want to keep the memory footprint small for rest of the process. | ||
|
||
```ruby | ||
Product.each_row.map {|r| r["id"].to_i } #=> [1, 2, 3, ...] | ||
Product.each_instance.map {|r| r.id }.each {|id| p id } #=> [1, 2, 3, ...] | ||
Product.each_instance.lazy.inject(0) {|sum,r| sum + r.quantity } #=> 499500 | ||
``` | ||
|
||
All these methods take an options hash to control things more: | ||
|
||
block_size:n The number of rows to fetch from the database each time (default 1000) | ||
while:value Continue looping as long as the block returns this value | ||
until:value Continue looping until the block returns this value | ||
connection:conn Use this connection instead of the current Product connection | ||
fraction:float A value to set for the cursor_tuple_fraction variable. | ||
PostgreSQL uses 0.1 (optimize for 10% of result set) | ||
This library uses 1.0 (Optimize for 100% of the result set) | ||
Do not override this value unless you understand it. | ||
|
||
Notes: | ||
|
||
* Use cursors *only* for large result sets. They have more overhead with the database | ||
than ActiveRecord selecting all matching records. | ||
* Aliases each_hash and each_hash_by_sql are provided for each_row and each_row_by_sql | ||
if you prefer to express what types are being returned. | ||
|
||
###Hashes vs. Instances | ||
|
||
The each_row method returns the Hash of strings for speed (as this allows you to process a lot of rows). | ||
Hashes are returned with String values, and you must take care of any type conversion. | ||
|
||
When you use each_instance, ActiveRecord lazily casts these strings into | ||
Ruby types (Time, Fixnum, etc.) only when you read the attribute. | ||
|
||
If you find you need the types cast for your attributes, consider using each_instance | ||
insead. ActiveRecord's read casting algorithm will only cast the values you need and | ||
has become more efficient over time. | ||
|
||
###Select and Pluck | ||
|
||
To limit the columns returned to just those you need, use `.select(:id, :name)` | ||
query method. | ||
|
||
```ruby | ||
Product.select(:id, :name).each_row { |product| product.process } | ||
``` | ||
|
||
Pluck is a great alternative instead of using a cursor. It does not instantiate | ||
the row, and builds an array of result values, and translates the values into ruby | ||
values (numbers, Timestamps. etc.). Using the cursor would still allow you to lazy | ||
load them in batches for very large sets. | ||
|
||
You can also use the `pluck_rows` or `pluck_instances` if the results | ||
won't eat up too much memory. | ||
|
||
```ruby | ||
Product.newly_arrived.pluck(:id) #=> [1, 2, 3, ...] | ||
Product.newly_arrived.each_row { |hash| } | ||
Product.select(:id).each_row.map {|r| r["id"].to_i } # cursor instead of pluck | ||
Product.pluck_rows(:id) #=> ["1", "2", ...] | ||
Product.pluck_instances(:id, :quantity) #=> [[1, 503], [2, 932], ...] | ||
``` | ||
|
||
###Associations and Eager Loading | ||
|
||
ActiveRecord performs some magic when eager-loading associated row. It | ||
will usually not join the tables, and prefers to load the data in | ||
separate queries. | ||
|
||
This library hooks onto the `to_sql` feature of the query builder. As a | ||
result, it can't do the join if ActiveRecord decided not to join, nor | ||
can it construct the association objects eagerly. | ||
|
||
##Background: Why PostgreSQL Cursors? | ||
|
||
ActiveRecord is designed and optimized for web performance. In a web transaction, only a "page" of | ||
around 20 rows is returned to the user. When you do this | ||
|
||
```ruby | ||
Product.find_each { |product| product.process } | ||
``` | ||
|
||
The database returns all matching result set rows to ActiveRecord, which instantiates each row with | ||
the data returned. This function returns an array of all these rows to the caller. | ||
|
||
Asyncronous, Background, or Offline processing may require processing a large amount of data. | ||
When there is a very large number of rows, this requires a lot more memory to hold the data. Ruby | ||
does not return that memory after processing the array, and the causes your process to "bloat". If you | ||
don't have enough memory, it will cause an exception. | ||
|
||
###ActiveRecord.find_each and find_in_batches | ||
|
||
To solve this problem, ActiveRecord gives us two alternative methods that work in "chunks" of your data: | ||
|
||
```ruby | ||
Product.where("id>0").find_each { |model| Product.process } | ||
|
||
Product.where("id>0").find_in_batches do |batch| | ||
batch.each { |model| Product.process } | ||
end | ||
``` | ||
|
||
Optionally, you can specify a :batch_size option as the size of the "chunk", and defaults to 1000. | ||
|
||
There are drawbacks with these methods: | ||
|
||
* You cannot specify the order, it will be ordered by the primary key (usually id) | ||
* The primary key must be numeric | ||
* The query is rerun for each chunk (1000 rows), starting at the next id sequence. | ||
* You cannot use overly complex queries as that will be rerun and incur more overhead. | ||
|
||
|
||
##Meta | ||
###Author | ||
Allen Fair, [@allenfair](https://twitter/com/allenfair), http://github.com/afair | ||
|
||
Thanks to: | ||
|
||
* Iulian Dogariu, http://github.com/iulianu (Fixes) | ||
* Julian Mehnle, julian@mehnle.net (Suggestions) | ||
* ...And all the other contributers! | ||
|
||
###Note on Patches/Pull Requests | ||
|
||
* Fork the project. | ||
* Make your feature addition or bug fix. | ||
* Add tests for it. This is important so I don't break it in a | ||
future version unintentionally. | ||
* Commit, do not mess with rakefile, version, or history. | ||
(if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull) | ||
* Send me a pull request. Bonus points for topic branches. | ||
|
||
###Copyright | ||
|
||
Copyright (c) 2010-2014 Allen Fair. See (MIT) LICENSE for details. |
This file was deleted.
Oops, something went wrong.
Oops, something went wrong.