Module: KMDB::ModuleMethods

Included in:
KMDB
Defined in:
lib/kmdb.rb

Instance Method Summary collapse

Instance Method Details

#connect ⇒ Object 



11
12
13
14
15
16
17
18
19
20
 
# File 'lib/kmdb.rb', line 11

def connect
  url = ENV.fetch('DATABASE_URL')
  puts url
  ActiveRecord::Base.establish_connection(url)

  if ENV.fetch('KMDB_AR_LOG', 'NO') == 'YES'
    ActiveRecord::Base.logger = ActiveSupport::Logger.new(STDOUT)
  end
  self
end

#env ⇒ Object 



7
8
9
 
# File 'lib/kmdb.rb', line 7

def env
  ENV['RACK_ENV'] || ENV['KMDB_ENV'] || 'development'
end

#migrate ⇒ Object 



22
23
24
25
26
 
# File 'lib/kmdb.rb', line 22

def migrate
  ActiveRecord::Migration.verbose = true
  ActiveRecord::Migrator.migrate MIGRATIONS_DIR
  self
end

#transaction(&block) ⇒ Object 



28
29
30
31
32
 
# File 'lib/kmdb.rb', line 28

def transaction(&block)
  ActiveRecord::Base.transaction do
    yield ActiveRecord::Base.connection
  end
end