class Mihari::Database

Public Class Methods

close() click to toggle source
# File lib/mihari/database.rb, line 120
def close
  ActiveRecord::Base.clear_active_connections!
  ActiveRecord::Base.connection.close
end
connect() click to toggle source
# File lib/mihari/database.rb, line 98
def connect
  case adapter
  when "postgresql", "mysql2"
    ActiveRecord::Base.establish_connection(Mihari.config.database)
  else
    ActiveRecord::Base.establish_connection(
      adapter: adapter,
      database: Mihari.config.database
    )
  end

  # ActiveRecord::Base.logger = Logger.new STDOUT
  ActiveRecord::Migration.verbose = false

  InitialSchema.migrate(:up)
  AddeSourceToArtifactSchema.migrate(:up)
  EnrichmentsSchema.migrate(:up)
  EnrichmentCreatedAtSchema.migrate(:up)
rescue StandardError
  # Do nothing
end
destroy!() click to toggle source
# File lib/mihari/database.rb, line 125
def destroy!
  return unless ActiveRecord::Base.connected?

  InitialSchema.migrate(:down)
  AddeSourceToArtifactSchema.migrate(:down)
  EnrichmentsSchema.migrate(:down)
  EnrichmentCreatedAtSchema.migrate(:down)
end