package com.oreilly.springdata.batch.item.mongodb; import java.util.HashSet; import java.util.Properties; import java.util.Set; import org.springframework.batch.item.file.transform.FieldSet; import org.springframework.core.convert.converter.Converter; import com.mongodb.BasicDBObject; import com.mongodb.DBObject; public class FieldSetConverter implements Converter<FieldSet, DBObject> { public DBObject convert(FieldSet fieldSet) { DBObject dbo = new BasicDBObject(); Properties props = fieldSet.getProperties(); Set<String> keys = new HashSet(props.keySet()); for (String key : keys) { if (key.compareToIgnoreCase("id") == 0) { dbo.put("_id", props.get(key)); } else { dbo.put(key, props.get(key)); } } return dbo; } }