MongoDB Helper的简单封装

lyl801013

db.properties
#mongodb数据库配置文件
#数据库服务器所在的ip地址ip=127.0.0.1
#mongodb服务端口号port=27017
#要连接的库dbName=yc
#用户名uname=
#密码password=
#默认的连接集合collectionName=voting

DBHelper
package com.yc.votingsys.dao;

import java.net.UnknownHostException;import java.util.ArrayList;import java.util.HashMap;import java.util.List;import java.util.Map;import java.util.Set;
import com.mongodb.BasicDBObject;import com.mongodb.DB;import com.mongodb.DBCollection;import com.mongodb.DBCursor;import com.mongodb.DBObject;import com.mongodb.Mongo;import com.mongodb.MongoException;import com.mongodb.WriteResult;import com.yc.votingsys.utils.LogUtil;import com.yc.votingsys.utils.StringUtil;

public class DBHelper {private static Mongo mongo=null;private DB db=null;private DBCollection collection=null;
/** * 连接到服务器 */static{try {mongo= new Mongo(MyPro.getInstance().getProperty("id"),Integer.parseInt( MyPro.getInstance().getProperty("port")));} catch (UnknownHostException e) {LogUtil.log.error(e.toString());e.printStackTrace();} catch (MongoException e) {LogUtil.log.error(e.toString());e.printStackTrace();}  }
/** * 获取数据库连接 * @param dbName：数据库名 * @param uname：登陆数据库的用户名 * @param pwd：登陆密码 * @return：成功返回true */public boolean getDb(String dbName,String uname,String pwd){db=mongo.getDB(MyPro.getInstance().getProperty("dbName"));
//校验用户密码是否正确 if(!StringUtil.isNull(uname) && !StringUtil.isNull(pwd)){ //如果给定了用户名和密码if (!db.authenticate(uname, pwd.toCharArray())){  LogUtil.log.error("用户名或密码错误，连接MongoDB数据库失败....");return false;}else{  LogUtil.log.error("连接MongoDB数据库成功...."); return true;}}else{ //如果没有给定，则读取配置文件中配置的用户名和密码String unames=MyPro.getInstance().getProperty("uname");String pwds=MyPro.getInstance().getProperty("password");
//校验用户密码是否正确 if(!StringUtil.isNull(unames) && !StringUtil.isNull(pwds)){if (!db.authenticate(unames, pwds.toCharArray())){  LogUtil.log.error("用户名或密码错误，连接MongoDB数据库失败....");return false;}else{  LogUtil.log.error("连接MongoDB数据库成功...."); return true;}}else{return true;}}}
/** * 关闭数据库连接 * @param mongo：服务器 * @param db：数据库 */public void closeAll(Mongo mongo,DB db){if(db!=null){db.requestDone();}//if(mongo!=null){//mongo.close();//}}
/** * 获取指定的集合 * @param collectionName：集合名称 * @param dbName：数据库名 * @param uname：登陆数据库的用户名 * @param pwd：登陆密码 * @return：返回获取到的集合 */public DBCollection getDBCollection(String collectionName,String dbName,String uname,String pwd){if(getDb(dbName,uname,pwd)){ //连接数据库成功db.requestStart(); //启动if(collectionName==null){collectionName=MyPro.getInstance().getProperty("collectionName");}collection=db.getCollection(collectionName); //获取指定的集合}else{throw new RuntimeException("数据库连接失败，请检查用户名和密码...");}return collection;}
/** * 添加对象 * @param map * @param collection * @return */public int addObject(Map<String,Object> map,String collectionName){WriteResult result=null;try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}result=collection.save(  new BasicDBObject(map) );} catch (Exception e) {LogUtil.log.error(e); e.printStackTrace();} finally{this.closeAll(mongo, db);}return result.getN();}
/** * 添加对象 * @param map * @param collection * @return */public int addObjects(Map<String,String> map,String collectionName){WriteResult result=null;try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}
result=collection.save(  new BasicDBObject(map) );} catch (Exception e) {LogUtil.log.error(e); e.printStackTrace();} finally{this.closeAll(mongo, db);}return result.getN();}
/** * 查询单个结果 * @param params * @param collectionName * @return */public Map<String,Object> findOne(Map<String,String> params,String collectionName){Map<String,Object> map=new HashMap<String,Object>();try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}
DBObject object;if(params!=null){object=collection.findOne( new BasicDBObject(params) );}else{object=collection.findOne();}
Set<String> keys=object.keySet();for(String key:keys){map.put(key, object.get(key));}}catch (MongoException e) {LogUtil.log.error(e); e.printStackTrace();}finally{this.closeAll(mongo, db);}return map;}public Object findOneToObject(Map<String,String> params,String collectionName){DBObject object = null;try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}if(params!=null){object=collection.findOne( new BasicDBObject(params) );}else{object=collection.findOne();}}catch (MongoException e) {LogUtil.log.error(e); e.printStackTrace();}finally{this.closeAll(mongo, db);}return object;}
public List<Object> find(Map<String,Object> params,String collectionName){List<Object> list=new ArrayList<Object>();try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}
DBCursor cursor;if(params!=null){cursor=collection.find(new BasicDBObject(params) );}else{cursor=collection.find();}
DBObject object;while(cursor.hasNext()){object=cursor.next();list.add(object);}}catch (MongoException e) {LogUtil.log.error(e); e.printStackTrace();}finally{this.closeAll(mongo, db);}return list;}
/** * 获取总记录条数 * @param params：参数列表 * @return：总记录数 */public int getTotal(Map<String,Object> params,String collectionName){int total=0;try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}if(params!=null){total=(int) collection.count( new BasicDBObject(params) );}else{total=(int) collection.count();}}catch (MongoException e) {LogUtil.log.error(e); e.printStackTrace();}finally{this.closeAll(mongo, db);}return total;}/** * 更新信息 * @param map：条件 * @param params：修改的&#20540; * @param collectionName * @return */public int update(Map<String,Object> map,Map<String,Map<String,Object>> params,String collectionName){WriteResult result = null;try {if(collectionName==null){collection=this.getDBCollection(null, null, null, null);}else{collection=this.getDBCollection(collectionName, null, null, null);}BasicDBObject object=new BasicDBObject();if(params!=null){Set<String> set=params.keySet();for(String key:set){object.append(key,params.get(key));}}if(map!=null){result=collection.update(new BasicDBObject(map),object);}else{result=collection.update(null,new BasicDBObject(params));}}catch (MongoException e) {LogUtil.log.error(e); e.printStackTrace();}finally{this.closeAll(mongo, db);}return result.getN();}}
         版权声明：本文为博主原创文章，未经博主允许不得转载。