Class: Dumper::Database::MongoDB

Inherits:
Base
  • Object
show all
Defined in:
lib/dumper/database/mongodb.rb

Constant Summary collapse

DUMP_TOOL =
'mongodump'
FILE_EXT =
'tar.gz'

Instance Attribute Summary

Attributes inherited from Base

#config, #filename, #tmpdir

Instance Method Summary collapse

Methods inherited from Base

#dump_path, #dump_tool_path, #file_ext, #finalize

Methods included from Utility::ObjectFinder

#find_instance_in_object_space

Instance Method Details

#additional_optionsObject



18
19
20
# File 'lib/dumper/database/mongodb.rb', line 18

def additional_options
  "--out='#{tmpdir}'"
end

#commandObject



7
8
9
# File 'lib/dumper/database/mongodb.rb', line 7

def command
  "cd #{tmpdir} && #{dump_tool_path} #{connection_options} #{additional_options} && tar -czf #{filename} --exclude='#{filename}' ."
end

#connection_optionsObject



11
12
13
14
15
16
# File 'lib/dumper/database/mongodb.rb', line 11

def connection_options
  [ :database, :host, :port, :username, :password ].map do |option|
    next if @config[option].blank?
    "--#{option}='#{@config[option]}'".gsub('--database', '--db')
  end.compact.join(' ')
end

#set_config_for(rails_env = nil) ⇒ Object



22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
# File 'lib/dumper/database/mongodb.rb', line 22

def set_config_for(rails_env=nil)
  return unless defined?(Mongo::DB) &&
    (mongo = find_instance_in_object_space(Mongo::DB)) ||
    defined?(Mongoid) && Mongoid::Config.respond_to?(:sessions) &&
    (mongoid = Mongoid::Config.sessions[:default])

  @config = {
    :host => mongo ? mongo.connection.host : mongoid[:hosts].first.split(/:/).first,
    :port => mongo ? mongo.connection.port : mongoid[:hosts].first.split(/:/).last,
    :database => mongo ? mongo.name : mongoid[:database],
    :dump_tool => dump_tool_path
  }.tap do |h|
    if auth = mongo ? mongo.connection.auths.first : mongoid
      h.update(:username => auth['username'], :password => auth['password'])
    end
  end
end