package com.x.mongodb;
import java.net.UnknownHostException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.bson.types.ObjectId;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoException;
public final class MongoDBUtil {
private static final String HOST = "127.0.0.1";
private static final String dbName = "test";
private static Mongo mongo;
private static DB db;
static {
try {
mongo = new Mongo(HOST);
db = mongo.getDB(dbName);
// db.authenticate(username, passwd)
} catch (UnknownHostException e) {
e.printStackTrace();
} catch (MongoException e) {
e.printStackTrace();
}
}
private MongoDBUtil() {
}
/**
* 添加操作
* <br>------------------------------<br>
* @param map
* @param collectionName
*/
public static void add(Map<String, Object> map, String collectionName) {
DBObject dbObject = new BasicDBObject(map);
getCollection(collectionName).insert(dbObject);
}
/**
* 添加操作
* <br>------------------------------<br>
* @param list
* @param collectionName
*/
public static void add(List<Map<String, Object>> list, String collectionName) {
for (Map<String, Object> map : list) {
add(map, collectionName);
}
}
/**
* 删除操作
* <br>------------------------------<br>
* @param map
* @param collectionName
*/
public static void delete(Map<String, Object> map, String collectionName) {
DBObject dbObject = new BasicDBObject(map);
getCollection(collectionName).remove(dbObject);
}
/**
* 删除操作,根据主键
* <br>------------------------------<br>
* @param id
* @param collectionName
*/
public static void delete(String id, String collectionName) {
Map<String, Object> map = new HashMap<String, Object>();
map.put("_id", new ObjectId(id));
delete(map, collectionName);
}
/**
* 删除全部
* <br>------------------------------<br>
* @param collectionName
*/
public static void deleteAll(String collectionName) {
getCollection(collectionName).drop();
}
/**
* 修改操作</br>
* 会用一个新文档替换现有文档,文档key结构会发生改变</br>
* 比如原文档{"_id":"123","name":"zhangsan","age":12}当根据_id修改age
* value为{"age":12}新建的文档name值会没有,结构发生了改变
* <br>------------------------------<br>
* @param whereMap
* @param valueMap
* @param collectionName
*/
public static void update(Map<String, Object> whereMap, Map<String, Object> valueMap, String collectionName) {
executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){
public DBObject doCallback(DBObject valueDBObject) {
return valueDBObject;
}
});
}
/**
* 修改操作,使用$set修改器</br>
* 用来指定一个键值,如果键不存在,则自动创建,会更新原来文档, 不会生成新的, 结构不会发生改变
* <br>------------------------------<br>
* @param whereMap
* @param valueMap
* @param collectionName
*/
public static void updateSet(Map<String, Object> whereMap, Map<String, Object> valueMap, String collectionName) {
executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){
public DBObject doCallback(DBObject valueDBObject) {
return new BasicDBObject("$set", valueDBObject);
}
});
}
/**
* 修改操作,使用$inc修改器</br>
* 修改器键的值必须为数字</br>
* 如果键存在增加或减少键的值, 如果不存在创建键
* <br>------------------------------<br>
* @param whereMap
* @param valueMap
* @param collectionName
*/
public static void updateInc(Map<String, Object> whereMap, Map<String, Integer> valueMap, String collectionName) {
executeUpdate(collectionName, whereMap, valueMap, new UpdateCallback(){
public DBObject doCallback(DBObject valueDBObject) {
return new BasicDBObject("$inc", valueDBObject);
}
});
}
/**
* 修改
* <br>------------------------------<br>
* @param collectionName
* @param whereMap
* @param valueMap
* @param updateCallback
*/
private static void executeUpdate(String collectionName, Map whereMap, Map valueMap, UpdateCallback updateCallback) {
DBObject whereDBObject = new BasicDBObject(whereMap);
DBObject valueDBObject = new BasicDBObject(valueMap);
valueDBObject = updateCallback.doCallback(valueDBObject);
getCollection(collectionName).update(whereDBObject, valueDBObject);
}
interface UpdateCallback {
DBObject doCallback(DBObject valueDBObject);
}
/**
* 获取集合(表)
* <br>------------------------------<br>
* @param collectionName
* @return
*/
public static DBCollection getCollection(String collectionName) {
return db.getCollection(collectionName);
}
}