class ActiveRecord::Reflection::AssociationReflection

def primary_key_column 

def primary_key_column
  @primary_key_column ||= klass.columns.find { |c| c.name == klass.primary_key }
end