Class: MongoMasker::Main

Inherits:
Object
  • Object
show all
Defined in:
lib/mongo_masker.rb

Instance Method Summary collapse

Constructor Details

#initialize(config = nil) ⇒ Main

Returns a new instance of Main.



9
10
11
12
 
# File 'lib/mongo_masker.rb', line 9

def initialize(config = nil)
  configure config unless config.nil?
  @sequence = 0
end

Instance Method Details

#configure(config) ⇒ Object 



30
31
32
 
# File 'lib/mongo_masker.rb', line 30

def configure(config)
  @config = config.is_a?(String) ? load_from_yaml(config) : config
end

#db ⇒ Object 



14
15
16
 
# File 'lib/mongo_masker.rb', line 14

def db
  @db ||= Mongo::Client.new(@config['db_url'] || 'mongodb://mongodb:27017/development')
end

#mask(config = nil) ⇒ Object 



18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/mongo_masker.rb', line 18

def mask(config = nil)
  configure config unless config.nil?
  raise 'Please provide mask' if @config.nil?

  track_time do
    @config['models'].each do |model|
      mask_document model
    end
    puts 'Done!' unless @config['silent']
  end
end

#seq ⇒ Object 



34
35
36
 
# File 'lib/mongo_masker.rb', line 34

def seq
  @sequence += 1
end