Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Passing a mongo document serializer to MongoClient.remove
  • Loading branch information
jyemin committed Dec 12, 2012
1 parent fb16fe7 commit 497e3fe
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions src/main/org/mongodb/impl/MongoCollectionImpl.java
Expand Up @@ -20,6 +20,7 @@
import org.mongodb.MongoClient;
import org.mongodb.MongoCollection;
import org.mongodb.MongoCursor;
import org.mongodb.MongoDocument;
import org.mongodb.MongoNamespace;
import org.mongodb.ReadPreference;
import org.mongodb.WriteConcern;
Expand All @@ -37,6 +38,7 @@
import org.mongodb.result.RemoveResult;
import org.mongodb.serialization.Serializer;
import org.mongodb.serialization.Serializers;
import org.mongodb.serialization.serializers.MongoDocumentSerializer;

class MongoCollectionImpl<T> implements MongoCollection<T> {
private final String name;
Expand Down Expand Up @@ -104,13 +106,14 @@ public T findAndRemove(final MongoFindAndRemove findAndRemove) {

@Override
public InsertResult insert(final MongoInsert<T> insert) {
return getClient().getOperations().insert(getNamespace(), insert.writeConcernIfAbsent(getWriteConcern()), serializer);
return getClient().getOperations().insert(getNamespace(), insert.writeConcernIfAbsent(getWriteConcern()), getSerializer());
}

@Override
public RemoveResult remove(final MongoRemove remove) {
// TODO: need a serializer to pass in here
return getClient().getOperations().remove(getNamespace(), remove.writeConcernIfAbsent(getWriteConcern()), null);
return getClient().getOperations().remove(getNamespace(), remove.writeConcernIfAbsent(getWriteConcern()),
getMongoDocumentSerializer());
}

@Override
Expand Down Expand Up @@ -152,4 +155,7 @@ public MongoNamespace getNamespace() {
return new MongoNamespace(getDatabase().getName(), getName());
}

private Serializer<MongoDocument> getMongoDocumentSerializer() {
return new MongoDocumentSerializer(baseSerializers);
}
}

0 comments on commit 497e3fe

Please sign in to comment.