lib/appfuel/storage/db/mapper.rb in appfuel-0.4.5 vs lib/appfuel/storage/db/mapper.rb in appfuel-0.5.0

@@ -14,13 +14,15 @@
       # Determine Domain Mapentry and DbModel from entity expression.
       #
       # @param expr [SpCore::Domain::Expr]
       # @return [table_name, column] [Array]
       def db_table_column(expr, entry = nil)
-        entry ||= find(expr.domain_name, expr.domain_attr)
-        db  = storage_class_from_entry(entry, :db)
+        storage_map = storage_map(:db, expr.domain_name)
 
-        [db.table_name, entry.storage_attr]
+        db     = fetch_storage_class(storage_map.storage_key)
+        column = storage_map.storage_attr(expr.domain_attr)
+
+        [db.table_name, column]
       end
 
       # Converts an entity expression into a valid active record expresion
       # expression.
       #