heroku-mongo-backup 0.1.0

data/lib/heroku_mongo_backup.rb

@@ -0,0 +1,136 @@
+# coding: UTF-8
+
+require 'mongo'
+require 'bson'
+require 'json'
+require 'zlib'
+require 'uri'
+require 'yaml'
+require 'rubygems'
+require 's3'
+
+module Heroku::Mongo
+  class Backup
+    def chdir
+      Dir.chdir("tmp")
+      begin
+        Dir.mkdir("dump")
+      rescue
+      end
+      Dir.chdir("dump")
+    end
+  
+    def store
+      backup = {}
+    
+      @db.collections.each do |col|
+        backup['system.indexes.db.name'] = col.db.name if col.name == "system.indexes"
+      
+        records = []
+      
+        col.find().each do |record|
+          records << record
+        end
+
+        backup[col.name] = records
+      end
+    
+      marshal_dump = Marshal.dump(backup)
+    
+      file = File.new(@file_name, 'w')
+      file = Zlib::GzipWriter.new(file)
+      file.write marshal_dump
+      file.close
+    end
+  
+    def load
+      file = Zlib::GzipReader.open(@file_name)
+      obj = Marshal.load file.read
+      file.close
+
+      obj.each do |col_name, records|
+        next if col_name =~ /^system\./
+      
+        @db.drop_collection(col_name)
+        dest_col = @db.create_collection(col_name)
+      
+        records.each do |record|
+          dest_col.insert record
+        end
+      end
+    
+      # Load indexes here
+      col_name = "system.indexes"
+      dest_index_col = @db.collection(col_name)
+      obj[col_name].each do |index|
+        if index['_id']
+          index['ns'] = index['ns'].sub(obj['system.indexes.db.name'], dest_index_col.db.name)
+          dest_index_col.insert index
+        end
+      end
+    end
+  
+    def connect
+      uri = URI.parse(@url)
+      connection = ::Mongo::Connection.new(uri.host, uri.port)
+      @db = connection.db(uri.path.gsub(/^\//, ''))
+      @db.authenticate(uri.user, uri.password) if uri.user
+    end
+  
+    def s3_connect
+      bucket            = YAML.load_file("config/assets.yml")['s3_bucket']
+      access_key_id     = ENV['S3_KEY']
+      secret_access_key = ENV['S3_SECRET']
+
+      service = S3::Service.new(:access_key_id => access_key_id,
+                                :secret_access_key => secret_access_key)
+      @bucket = service.buckets.find(bucket)
+    end
+  
+    def s3_upload
+      object = @bucket.objects.build("backups/#{@file_name}")
+      object.content = open(@file_name)
+      object.save
+    end
+  
+    def s3_download
+      open(@file_name, 'w') do |file|
+        object = @bucket.objects.find("backups/#{@file_name}")
+        file.write object.content
+      end
+    end
+  
+    def initialize
+      # Backup settings
+      @file_name = Time.now.strftime("%Y-%m-%d_%H-%M-%S.gz")
+    
+      # MongoDB config
+      local_db_name = YAML.load_file("config/mongoid.yml")['development']['database']
+    
+      development_uri = "mongodb://localhost:27017/#{local_db_name}"
+      production_uri = YAML.load_file("config/mongoid.yml")['production']['uri']
+    
+      #@url = development_uri
+      @url = production_uri
+    
+      puts "Using databased: #{@url}"
+    
+      self.connect
+      self.s3_connect
+    end
+  
+    def backup
+      self.chdir    
+      self.store
+      self.s3_upload
+    end
+  
+    def restore file_name
+      @file_name = file_name
+    
+      self.chdir
+      self.s3_download
+      self.load
+    end
+  end
+end

data/lib/tasks/heroku_mongo_backup.rake

@@ -0,0 +1,17 @@
+# coding: UTF-8
+
+namespace :mongo do
+  desc "Mongo backup and restore commands.\n\tBackup: rake mongo:backup\n\tRestore: rake mongo:restore FILE=<backup-file.gz>"
+  
+  task :backup => [:environment] do
+    Heroku::Mongo::Backup.new.backup
+  end
+  
+  task :restore => [:environment] do
+    if ENV['FILE']
+      Heroku::Mongo::Backup.new.restore ENV['FILE']
+    else
+      puts "Please provide backup file to restore from. Thanks!"
+    end
+  end
+end

metadata

@@ -0,0 +1,68 @@
+--- !ruby/object:Gem::Specification
+name: heroku-mongo-backup
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Alex Kraves
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2011-09-13 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70318711197760 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70318711197760
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: &70318711197280 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: *70318711197280
+description: Rake task for backing up mongo database on heroku and push it to S3.
+email: mail@alexkravets.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/heroku_mongo_backup.rb
+- lib/tasks/heroku_mongo_backup.rake
+homepage: https://github.com/alexkravets/heroku-mongo-backup
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: nowarning
+rubygems_version: 1.8.6
+signing_key: 
+specification_version: 3
+summary: Rake task for backing up mongo database on heroku and push it to S3.
+test_files: []