Java MongoDB封装代码,Java代码库,德仔网

代码语言
代码分类
【Java】 Java MongoDB封装
作者:金马超 / 发布于2015/8/3/ 668
驱动包版本最高是2.10.1 2.10.1以上版本中删除了ObjectId.massageToObjectId(key); 这个方法, 所以最高是2.10.1. 可以在 http://search.maven.org/ 中找到这个版本配置文件只需在classpath下即可,名字任取,传参数即可.

	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日
	* @版权 &copy; 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,批量更新.
	 * 因massageToObjectId()删除,所以MongoDB驱动包版本最高为2.10.1(含)
	 * 2.10.1以上版本无此方法,故需要自行确定ObjectId.
	 * 待有时间找到该方法的替代方法.
	 * @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();
	 }
	
	 }
	
	}

	

	SystemMessage.java

	package com.liloo.util;
	
	import java.util.MissingResourceException;
	import java.util.ResourceBundle;
	
	/**
	* 获取配置文件参数
	* @作者 Liloo
	* @E-mail liloo@liloo.top
	* @时间 2015年7月29日
	* @版权 &copy; Liloo 版权所有.
	*/
	public class SystemMessage {
	
	 private SystemMessage() {
	
	 }
	
	 /**
	 * 从配置文件获取参数
	 * @param bundle_name
	 * @param key
	 * @return
	 */
	 public static String getString(String bundle_name,String key) {
	
	 try {
	
	 return ResourceBundle.getBundle(bundle_name).getString(key);
	
	 } catch (MissingResourceException e) {
	 return '!' + key + '!';
	 }
	
	 }
	
	}
试试其它关键字
　MongoDB　
同语言下
可能有用的
金马超贡献的其它代码(3)