-
-
Save bf4/7e02f97b60c9e0bd09ce0f2ae8bdb662 to your computer and use it in GitHub Desktop.
Live stream a database dump (or any other STDOUT) using Rails 4. Why would you want this? If you have a large database dump and want to avoid storing it in memory as Rails streams it. This allows pipe the dump directly into the http response instead of storing it as a file, sending it, and then deleting it. Let me know what you think! I've teste…
This file contains hidden or 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
| class DatabaseController < ApplicationController | |
| def database_dump | |
| database = Rails.configuration.database_configuration[Rails.env]["database"] | |
| send_file_headers!(:type => 'application/octet-stream', :filename => "#{database}_#{Time.now.to_s(:human)}.backup") | |
| pipe = IO.popen("pg_dump '#{database}' -F c") | |
| stream = response.stream | |
| while (line = pipe.gets) | |
| stream.write line | |
| sleep 0.0001 # HACK: Prevent server instance from sleeping forever if client disconnects during download | |
| end | |
| rescue IOError | |
| # Client Disconnected | |
| ensure | |
| pipe.close | |
| response.stream.close | |
| end | |
| # Code that allows us to only mix in the live methods if we're accessing the desired action | |
| def dispatch(name, *args) | |
| extend ActionController::Live if Array(@@live_actions).include?(name.to_s) | |
| super | |
| end | |
| @@live_actions = [] | |
| def self.live(actions) | |
| @@live_actions |= Array(actions).collect(&:to_s) | |
| end | |
| live :database_dump | |
| end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment