shorten the persistence loading stuff a bit

aasm · Apr 24, 2013 · 7ab505a · 7ab505a
1 parent 55f6ccd
commit 7ab505a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/lib/aasm/persistence.rb b/lib/aasm/persistence.rb
@@ -7,20 +7,20 @@ def load_persistence(base)
         hierarchy = base.ancestors.map {|klass| klass.to_s}
 
         if hierarchy.include?("ActiveRecord::Base")
-          require_files_for('active_record_persistence')
+          require_files_for(:active_record)
           base.send(:include, AASM::Persistence::ActiveRecordPersistence)
         elsif hierarchy.include?("Mongoid::Document")
-          require_files_for('mongoid_persistence')
+          require_files_for(:mongoid)
           base.send(:include, AASM::Persistence::MongoidPersistence)
         end
       end
 
     private
 
       def require_files_for(persistence)
-        require File.join(File.dirname(__FILE__), 'persistence', 'base')
-        require File.join(File.dirname(__FILE__), 'persistence', 'read_state')
-        require File.join(File.dirname(__FILE__), 'persistence', persistence)
+        ['base', 'read_state', "#{persistence}_persistence"].each do |file_name|
+          require File.join(File.dirname(__FILE__), 'persistence', file_name)
+        end
       end
 
     end # class << self