Add option to stringify symbols when looking them up from an original object. An alternate approach could be to have two different classes, one for objects with symbol keys, and one for string keys

git-svn-id: https://svn.apache.org/repos/asf/lucene/solr/trunk@532569 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Erik Hatcher 2007-04-26 02:09:50 +00:00
parent d1f038a1c8
commit 5cd57a4a57
1 changed files with 3 additions and 2 deletions

View File

@ -11,8 +11,9 @@
# limitations under the License.
class Solr::Importer::Mapper
def initialize(mapping)
def initialize(mapping, options={})
@mapping = mapping
@options = options
end
def field_data(orig_data, field_name)
@ -26,7 +27,7 @@ class Solr::Importer::Mapper
when Proc
field_mapping.call(orig_data)
when Symbol
field_data(orig_data, field_mapping)
field_data(orig_data, @options[:stringify_symbols] ? field_mapping.to_s : field_mapping)
when Enumerable
field_mapping.collect {|orig_field_name| mapped_field_value(orig_data, orig_field_name)}.flatten
else