Jeffrey Martin
07cbe426e2
ApplicationRecord is a new superclass for all app models, analogous to app controllers subclassing ApplicationController instead of ActionController::Base. This gives apps a single spot to configure app-wide model behavior. https://edgeguides.rubyonrails.org/upgrading_ruby_on_rails.html#active-record-models-now-inherit-from-applicationrecord-by-default Deprecated Relation#uniq use Relation#distinct instead. https://edgeguides.rubyonrails.org/5_0_release_notes.html#active-record-deprecations
48 lines
980 B
Ruby
48 lines
980 B
Ruby
module Msf::DBManager::Ref
|
|
#
|
|
# Find or create a reference matching this name
|
|
#
|
|
def find_or_create_ref(opts)
|
|
ret = {}
|
|
|
|
::ApplicationRecord.connection_pool.with_connection {
|
|
if opts[:id] && !opts[:id].to_s.empty?
|
|
return Mdm::Ref.find(opts[:id])
|
|
end
|
|
|
|
if opts[:ref]
|
|
return get_ref(opts[:name])
|
|
end
|
|
|
|
ref = ::Mdm::Ref.where(name: opts[:name]).first_or_initialize
|
|
|
|
begin
|
|
framework.events.on_db_ref(ref) if ref
|
|
rescue ::Exception => e
|
|
wlog("Exception in on_db_ref event handler: #{e.class}: #{e}")
|
|
wlog("Call Stack\n#{e.backtrace.join("\n")}")
|
|
end
|
|
|
|
if ref and ref.changed?
|
|
ref.save!
|
|
end
|
|
ref
|
|
}
|
|
end
|
|
|
|
def get_ref(name)
|
|
::ApplicationRecord.connection_pool.with_connection {
|
|
::Mdm::Ref.find_by_name(name)
|
|
}
|
|
end
|
|
|
|
#
|
|
# Find a reference matching this name
|
|
#
|
|
def has_ref?(name)
|
|
::ApplicationRecord.connection_pool.with_connection {
|
|
Mdm::Ref.find_by_name(name)
|
|
}
|
|
end
|
|
end
|