org.mongodb.Document Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongoFS Show documentation
Show all versions of mongoFS Show documentation
An extension to the MongoDB Java Driver library that goes beyond what the GridFS feature supports.
Compressed file storage, zip files, temporary files
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