org.mongodb.Document Maven / Gradle / Ivy
package org.mongodb;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.util.Collection;
import java.util.Date;
import me.lightspeed7.mongofs.MongoFileConstants;
import org.bson.BSON;
import org.bson.types.ObjectId;
import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
public class Document {
private DBObject surrogate;
public Document(final String key, final Object id) {
this();
getSurrogate().put(key, unwrap(id));
}
public Document() {
setSurrogate(new BasicDBObject());
}
public Document(final DBObject incoming) {
this.setSurrogate(incoming);
}
public Document append(final String key, final Object value) {
getSurrogate().put(key, unwrap(value));
return this;
}
public Object put(final String key, final Object value) {
return getSurrogate().put(key, unwrap(value));
}
public Object get(final String key) {
return wrap(getSurrogate().get(key));
}
public String getString(final MongoFileConstants constant) {
return this.getString(constant.name());
}
@SuppressWarnings("deprecation")
public boolean containsKey(final String key) {
return getSurrogate().containsKey(key);
}
public Collection keySet() {
return getSurrogate().keySet();
}
//
// Private helpers
// ///////////////////////////////
private Object unwrap(final Object value) {
if (value == null || !value.getClass().isAssignableFrom(Document.class)) {
return value;
}
if (value.getClass().isAssignableFrom(Document.class) //
&& value.getClass().isAssignableFrom(Serializable.class)) {
return value;
}
try {
Field field = value.getClass().getDeclaredField("surrogate");
field.setAccessible(true);
return field.get(value);
} catch (Exception e) {
throw new IllegalStateException("surrogate.surrogate not valid", e);
}
}
private Object wrap(final Object value) {
if (value != null && value.getClass().isAssignableFrom(DBObject.class)) {
return new Document((DBObject) value);
}
return value;
}
/**
* Returns the value of a field as an int.
*
* @param key
* the field to look for
* @return the field value (or default)
*/
public int getInt(final String key) {
Object o = get(key);
if (o == null) {
throw new NullPointerException("no value for: " + key);
}
return BSON.toInt(o);
}
/**
* Returns the value of a field as an int.
*
* @param key
* the field to look for
* @param def
* the default to return
* @return the field value (or default)
*/
public int getInt(final String key, final int def) {
Object foo = get(key);
if (foo == null) {
return def;
}
return BSON.toInt(foo);
}
/**
* Returns the value of a field as a long.
*
* @param key
* the field to return
* @return the field value
*/
public long getLong(final String key) {
Object foo = get(key);
return ((Number) foo).longValue();
}
/**
* Returns the value of a field as an long.
*
* @param key
* the field to look for
* @param def
* the default to return
* @return the field value (or default)
*/
public long getLong(final String key, final long def) {
Object foo = get(key);
if (foo == null) {
return def;
}
return ((Number) foo).longValue();
}
/**
* Returns the value of a field as a double.
*
* @param key
* the field to return
* @return the field value
*/
public double getDouble(final String key) {
Object foo = get(key);
return ((Number) foo).doubleValue();
}
/**
* Returns the value of a field as an double.
*
* @param key
* the field to look for
* @param def
* the default to return
* @return the field value (or default)
*/
public double getDouble(final String key, final double def) {
Object foo = get(key);
if (foo == null) {
return def;
}
return ((Number) foo).doubleValue();
}
/**
* Returns the value of a field as a string
*
* @param key
* the field to look up
* @return the value of the field, converted to a string
*/
public String getString(final String key) {
Object foo = get(key);
if (foo == null) {
return null;
}
return foo.toString();
}
/**
* Returns the value of a field as a string
*
* @param key
* the field to look up
* @param def
* the default to return
* @return the value of the field, converted to a string
*/
public String getString(final String key, final String def) {
Object foo = get(key);
if (foo == null) {
return def;
}
return foo.toString();
}
/**
* Returns the value of a field as a boolean.
*
* @param key
* the field to look up
* @return the value of the field, or false if field does not exist
*/
public boolean getBoolean(final String key) {
return getBoolean(key, false);
}
/**
* Returns the value of a field as a boolean
*
* @param key
* the field to look up
* @param def
* the default value in case the field is not found
* @return the value of the field, converted to a string
*/
public boolean getBoolean(final String key, final boolean def) {
Object foo = get(key);
if (foo == null) {
return def;
}
if (foo instanceof Number) {
return ((Number) foo).intValue() > 0;
}
if (foo instanceof Boolean) {
return ((Boolean) foo).booleanValue();
}
throw new IllegalArgumentException("can't coerce to bool:" + foo.getClass());
}
/**
* Returns the object id or null if not set.
*
* @param field
* The field to return
* @return The field object value or null if not found (or if null :-^).
*/
public ObjectId getObjectId(final String field) {
return (ObjectId) get(field);
}
/**
* Returns the object id or def if not set.
*
* @param field
* The field to return
* @param def
* the default value in case the field is not found
* @return The field object value or def if not set.
*/
public ObjectId getObjectId(final String field, final ObjectId def) {
final Object foo = get(field);
return (foo != null) ? (ObjectId) foo : def;
}
/**
* Returns the date or null if not set.
*
* @param field
* The field to return
* @return The field object value or null if not found.
*/
public Date getDate(final String field) {
return (Date) get(field);
}
/**
* Returns the date or def if not set.
*
* @param field
* The field to return
* @param def
* the default value in case the field is not found
* @return The field object value or def if not set.
*/
public Date getDate(final String field, final Date def) {
final Object foo = get(field);
return (foo != null) ? (Date) foo : def;
}
public Integer getInteger(final Object key) {
return (Integer) get(key.toString());
}
public int getInteger(final Object key, final int defaultValue) {
Object value = get(key.toString());
return value == null ? defaultValue : (Integer) value;
}
public DBObject getSurrogate() {
return surrogate;
}
public void setSurrogate(final DBObject surrogate) {
this.surrogate = surrogate;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy