mindol1004/spring-data-mongodb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added quick fix to populate generated _id in saved object

Browse Source
This commit is contained in:
Thomas Risberg
2010-12-02 12:28:29 -05:00
parent 50e48b570b
commit 5d5688fd1b
1 changed files with 35 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
spring-data-mongodb/src/main/java/org/springframework/data/document/mongodb/MongoTemplate.java
View File

@@ -17,9 +17,14 @@
package org.springframework.data.document.mongodb;
import java.beans.PropertyDescriptor;
import java.util.ArrayList;
import java.util.List;
import org.bson.types.ObjectId;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.BeanWrapper;
import org.springframework.beans.PropertyAccessorFactory;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.dao.DataRetrievalFailureException;
import org.springframework.dao.InvalidDataAccessApiUsageException;
@@ -210,27 +215,30 @@ public class MongoTemplate extends AbstractDocumentStoreTemplate<DB> implements
}
public void save(String collectionName, Object objectToSave) {
BasicDBObject dbDoc = new BasicDBObject();
this.mongoConverter.write(objectToSave, dbDoc);
saveDBObject(collectionName, dbDoc);
save(collectionName, objectToSave, this.mongoConverter);
}
public <T> void save(String collectionName, T objectToSave, MongoWriter<T> writer) {
BasicDBObject dbDoc = new BasicDBObject();
this.mongoConverter.write(objectToSave, dbDoc);
saveDBObject(collectionName, dbDoc);
writer.write(objectToSave, dbDoc);
Object _id = saveDBObject(collectionName, dbDoc);
populateIdIfNecessary(objectToSave, _id);
}
protected void saveDBObject(String collectionName, BasicDBObject dbDoc) {
protected Object saveDBObject(String collectionName, BasicDBObject dbDoc) {
if (dbDoc.keySet().size() > 0 ) {
WriteResult wr = null;
try {
wr = getConnection().getCollection(collectionName).save(dbDoc);
return dbDoc.get("_id");
} catch (MongoException e) {
throw new DataRetrievalFailureException(wr.getLastError().getErrorMessage(), e);
}
}
else {
return null;
}
}
public <T> List<T> queryForCollection(Class<T> targetClass) {
@@ -344,7 +352,26 @@ public class MongoTemplate extends AbstractDocumentStoreTemplate<DB> implements
return dbo;
}
private void populateIdIfNecessary(Object savedObject, Object id) {
//TODO Needs proper conversion support and should be integrated with reader implementation somehow
BeanWrapper bw = PropertyAccessorFactory.forBeanPropertyAccess(savedObject);
PropertyDescriptor idPd = BeanUtils.getPropertyDescriptor(savedObject.getClass(), "id");
if (idPd == null) {
idPd = BeanUtils.getPropertyDescriptor(savedObject.getClass(), "_id");
}
if (idPd != null) {
Object v = bw.getPropertyValue(idPd.getName());
if (v == null) {
if (id instanceof ObjectId) {
bw.setPropertyValue(idPd.getName(), id.toString());
}
else if (id.getClass().isAssignableFrom(idPd.getPropertyType())) {
bw.setPropertyValue(idPd.getName(), id);
}
}
}
}
public void afterPropertiesSet() throws Exception {
if (this.getDefaultCollectionName() != null) {