added $unset to UpdateBuilder

This commit is contained in:
Thomas Risberg
2010-12-06 10:26:26 -05:00
parent 4d49068117
commit 0fda8f7520
2 changed files with 11 additions and 0 deletions

View File

@@ -30,6 +30,11 @@ public class UpdateBuilder {
return this; return this;
} }
public UpdateBuilder unset(String key) {
criteria.put("$unset", Collections.singletonMap(key, 1));
return this;
}
public UpdateBuilder inc(String key, long inc) { public UpdateBuilder inc(String key, long inc) {
criteria.put("$inc", Collections.singletonMap(key, inc)); criteria.put("$inc", Collections.singletonMap(key, inc));
return this; return this;

View File

@@ -43,5 +43,11 @@ public class UpdateBuilderTests {
ub.build().toString()); ub.build().toString());
} }
@Test
public void testUnset() {
UpdateBuilder ub = new UpdateBuilder()
.unset("directory");
Assert.assertEquals("{ \"$unset\" : { \"directory\" : 1}}", ub.build().toString());
}
} }