diff --git a/lib/rails_erd/cli.rb b/lib/rails_erd/cli.rb index a1de4e3..9cebb90 100644 --- a/lib/rails_erd/cli.rb +++ b/lib/rails_erd/cli.rb @@ -164,7 +164,7 @@ def start def initialize(path, options) @path, @options = path, options - require "rails_erd/diagram/graphviz" if options.generator == :graphviz + require "rails_erd/diagram/graphviz" if options[:generator] == :graphviz end def start diff --git a/lib/rails_erd/domain.rb b/lib/rails_erd/domain.rb index 6ec975d..9ef2669 100644 --- a/lib/rails_erd/domain.rb +++ b/lib/rails_erd/domain.rb @@ -135,6 +135,20 @@ def rails_models ActiveStorage::VariantRecord ActionText::RichText ActionText::EncryptedRichText + SolidQueue::Message + SolidCache::Entry + SolidQueue::Semaphore + SolidQueue::RecurringTask + SolidQueue::Process + SolidQueue::Pause + SolidQueue::Job + SolidQueue::ScheduledExecution + SolidQueue::RecurringExecution + SolidQueue::ReadyExecution + SolidQueue::FailedExecution + SolidQueue::ClaimedExecution + SolidQueue::BlockedExecution + SolidQueue::Execution ).map{ |model| Object.const_get(model) rescue nil }.compact end