
平台论坛博客文库

› 论坛 › 程序设计 › Java › Java MongoDB封装

Java MongoDB封装 [复制链接]

wyfem

丰衣足食

论坛徽章:: 0

电梯直达

1楼 [收藏(0)] [报告]

发表于 2015-07-30 10:30 |只看该作者 |倒序浏览

驱动包版本最高是2.10.1
2.10.1以上版本中删除了ObjectId.massageToObjectId(key); 这个方法,
所以最高是2.10.1.
可以在 http://search.maven.org/ 中找到这个版本

配置文件只需在classpath下即可,名字任取,传参数即可.

MongoManager.java

package com.liloo.db;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.LinkedBlockingQueue;
import org.apache.log4j.Logger;
import org.bson.types.ObjectId;
import com.liloo.util.SystemMessage;
import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
/**
* @作者 Liloo
* @E-mail liloo@vip.qq.com
* @时间 2015年5月14日
* @版权 © Liloo 版权所有.
*/
public class MongoManager {
private static Logger log = Logger.getLogger(MongoManager.class);
private static final MongoManager instance = new MongoManager();
private static MongoClient mongo = null;
private static final String host = SystemMessage.getString("systemConstant", "mongo_host");
private static final Integer port = Integer.valueOf(SystemMessage.getString("systemConstant", "mongo_port"));
/**
* 私有化
*/
private MongoManager() {
}
/**
* 单例
* @return
*/
public static MongoManager getInstance() {
return instance;
}
/**
* 初始化MongoDB
*/
public void init() {
try {
mongo = new MongoClient(host, port);
log.info("MongoDB init success!");
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 获取DB对象
* @return
*/
public DB getDB() {
try {
if (mongo == null) {
init();
log.debug("Get DB : " + SystemMessage.getString("systemConstant", "DB_name"));
}
return mongo.getDB(SystemMessage.getString("systemConstant", "DB_name"));
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 获取集合对象
* @param name
* @return
*/
private DBCollection getCollection(String name) {
try {
return getDB().getCollection(name);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 插入MongoDB
* @param name
* @param obj
*/
public void insert(String name, DBObject obj) {
try {
long begin = System.currentTimeMillis();
getCollection(name).insert(obj);
long end = System.currentTimeMillis();
log.debug("Insert Complete! Cost " + (end - begin) + "/ms");
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 删除指定条件的数据
* @param name
* @param obj
*/
public void delete(String name, DBObject obj) {
try {
getCollection(name).remove(obj);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 清空集合
* @param collection
* @throws Exception
*/
public void deleteAll(String collection) {
try {
List<DBObject> rs = findAll(collection);
if (rs != null && !rs.isEmpty()) {
for (int i = 0; i < rs.size(); i++) {
getCollection(collection).remove(rs.get(i));
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 如果更新的数据不存在插入一条数据
* @param collection
* @param setFields
* @param whereFields
*/
public void updateOrInsert(String name, DBObject set, DBObject where) {
try {
getCollection(name).update(where, set, true, false);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 只更新存在的数据,不会新增. 批量更新.
* @param name
* @param setFields
* @param whereFields
*/
public void updateExistDataWithBatch(String name, DBObject set, DBObject where) {
try {
getCollection(name).update(where, new BasicDBObject("$set", set), false, true);
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 按照ObjectId,批量更新.<br>
* 因massageToObjectId()删除,所以MongoDB驱动包版本最高为2.10.1(含)<br>
* 2.10.1以上版本无此方法,故需要自行确定ObjectId.<br>
* 待有时间找到该方法的替代方法.
* @param name
* @param ids
* @param set
*/
public void updateBatchByObjectId(String name, String ids, DBObject set) {
try {
if (ids == null || ids == "")
return;
String[] id = ids.split(",");
for (int i = 0; i < id.length; i++) {
BasicDBObject dest = new BasicDBObject();
BasicDBObject doc = new BasicDBObject();
dest.put("_id", ObjectId.massageToObjectId(id[i]));
doc.put("$set", set);
getCollection(name).update(dest, doc, false, true);
}
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 查询全部
* @param name
* @return
*/
public List<DBObject> findAll(String name) {
try {
return getCollection(name).find().toArray();
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 查询1条记录
* @param name
* @param obj
* @return
*/
public DBObject findOne(String name, DBObject obj) {
try {
DBCollection coll = getCollection(name);
return coll.findOne(obj);
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 查询指定条数的记录
* @param name
* @param obj
* @param limit
* @return
*/
public List<DBObject> find(String name, DBObject obj, int limit) {
try {
DBCollection coll = getCollection(name);
DBCursor c = coll.find(obj).limit(limit);
if (c != null){
List<DBObject> list = new ArrayList<DBObject>();
list = c.toArray();
return list;
}
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 查询符合的全部数据
* @param name
* @param where
* @return
*/
public List<DBObject> find(String name, DBObject where) {
try {
DBCursor c = getCollection(name).find(where);
if (c != null) {
List<DBObject> list = new ArrayList<DBObject>();
list = c.toArray();
return list;
}
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 返回Queue的查询
* @param name
* @param where
* @return
* @throws Exception
*/
public LinkedBlockingQueue<DBObject> findQueue(String name, DBObject where) {
try {
LinkedBlockingQueue<DBObject> queue = new LinkedBlockingQueue<DBObject>();
DBCursor c = getCollection(name).find(where);
if (c != null) {
for (DBObject obj : c) {
obj = c.next();
queue.offer(obj);
}
return queue;
}
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
/**
* 关闭Mongo链接
*/
public void close() {
try {
if (mongo != null) {
mongo.close();
log.info("MongoClient has benn closed...");
}
} catch (Exception e) {
e.printStackTrace();
}
}
}