MongoDB -- MongoClient连接池用法

来源：互联网发布：知天下事编辑：程序博客网时间：2024/06/05 18:54

主要的调整在于2.10.0版本中引入了MongoClient类，同时在其API中也说明了Mongo类会在将来的版本中被MongoClient替换（Note: This class has been superseded by MongoClient, and may be deprecated in a future release.）。故在这次调整中，也对原先的Mongodb部分做了相应的修改。

MongoClient被设计成线程安全、可以被多线程共享的。通常访问数据库集群的应用只需要一个实例，所以这次调整我们设计成单例形式。如果出于某些原因，你决定使用多个实例，请注意：所有资源使用限制（最大连接数等等）对每个MongoClient都适用；销毁一个实例时，请确认调用MongoClient.close()方法来清理资源。

下面给出相关的代码：

    /**       * 项目名：SpiderCrawler       * 文件名：MongoDBDao.java       * 作者：zhouyh       * 时间：2014-8-30 下午03:46:55       * 描述：TODO(用一句话描述该文件做什么)        */      package com.newsTest.dao;      import java.util.ArrayList;      import com.mongodb.DB;      import com.mongodb.DBCollection;      import com.mongodb.DBObject;      /**      * 类名： MongoDBDao      * 包名： com.newsTest.dao      * 作者： zhouyh      * 时间： 2014-8-30 下午03:46:55      * 描述： TODO(这里用一句话描述这个类的作用)       */      public interface MongoDBDao {          /**          *           * 方法名：getDb          * 作者：zhouyh          * 创建时间：2014-8-30 下午03:53:40          * 描述：获取指定的mongodb数据库          * @param dbName          * @return          */          public DB getDb(String dbName);          /**          *           * 方法名：getCollection          * 作者：zhouyh          * 创建时间：2014-8-30 下午03:54:43          * 描述：获取指定mongodb数据库的collection集合          * @param dbName    数据库名          * @param collectionName    数据库集合          * @return          */          public DBCollection getCollection(String dbName, String collectionName);          /**          *           * 方法名：inSert          * 作者：zhouyh          * 创建时间：2014-8-30 下午04:07:35          * 描述：向指定的数据库中添加给定的keys和相应的values          * @param dbName          * @param collectionName          * @param keys          * @param values          * @return          */          public boolean inSert(String dbName, String collectionName, String[] keys, Object[] values);          /**          *           * 方法名：delete          * 作者：zhouyh          * 创建时间：2014-8-30 下午04:09:00          * 描述：删除数据库dbName中，指定keys和相应values的值          * @param dbName          * @param collectionName          * @param keys          * @param values          * @return          */          public boolean delete(String dbName, String collectionName, String[] keys, Object[] values);          /**          *           * 方法名：find          * 作者：zhouyh          * 创建时间：2014-8-30 下午04:11:11          * 描述：从数据库dbName中查找指定keys和相应values的值          * @param dbName          * @param collectionName          * @param keys          * @param values          * @param num          * @return          */          public ArrayList<DBObject> find(String dbName, String collectionName, String[] keys, Object[] values, int num);          /**          *           * 方法名：update          * 作者：zhouyh          * 创建时间：2014-8-30 下午04:17:54          * 描述：更新数据库dbName，用指定的newValue更新oldValue          * @param dbName          * @param collectionName          * @param oldValue          * @param newValue          * @return          */          public boolean update(String dbName, String collectionName, DBObject oldValue, DBObject newValue);          /**          *           * 方法名：isExit          * 作者：zhouyh          * 创建时间：2014-8-30 下午04:19:21          * 描述：判断给定的keys和相应的values在指定的dbName的collectionName集合中是否存在          * @param dbName          * @param collectionName          * @param keys          * @param values          * @return          */          public boolean isExit(String dbName, String collectionName, String key, Object value);      }

/**  * 项目名：SpiderCrawler  * 文件名：MongoDBDaoImpl.java  * 作者：zhouyh  * 时间：2014-8-30 下午04:21:11  * 描述：TODO(用一句话描述该文件做什么)   */package com.newsTest.dao.impl;import java.net.UnknownHostException;import java.util.ArrayList;import com.newsTest.dao.MongoDBDao;import com.mongodb.BasicDBObject;import com.mongodb.DB;import com.mongodb.DBCollection;import com.mongodb.DBCursor;import com.mongodb.DBObject;import com.mongodb.MongoClient;import com.mongodb.MongoClientOptions;import com.mongodb.MongoException;import com.mongodb.WriteResult;/** * 类名： MongoDBDaoImpl * 包名： com.newsTest.dao.impl * 作者： zhouyh * 时间： 2014-8-30 下午04:21:11 * 描述： TODO(这里用一句话描述这个类的作用)  */public class MongoDBDaoImpl implements MongoDBDao{    /**     * MongoClient的实例代表数据库连接池，是线程安全的，可以被多线程共享，客户端在多线程条件下仅维持一个实例即可     * Mongo是非线程安全的，目前mongodb API中已经建议用MongoClient替代Mongo     */    private MongoClient mongoClient = null;    /**     *      * 私有的构造函数     * 作者：zhouyh     */    private MongoDBDaoImpl(){        if(mongoClient == null){            MongoClientOptions.Builder build = new MongoClientOptions.Builder();                    build.connectionsPerHost(50);   //与目标数据库能够建立的最大connection数量为50            build.autoConnectRetry(true);   //自动重连数据库启动            build.threadsAllowedToBlockForConnectionMultiplier(50); //如果当前所有的connection都在使用中，则每个connection上可以有50个线程排队等待            /*             * 一个线程访问数据库的时候，在成功获取到一个可用数据库连接之前的最长等待时间为2分钟             * 这里比较危险，如果超过maxWaitTime都没有获取到这个连接的话，该线程就会抛出Exception             * 故这里设置的maxWaitTime应该足够大，以免由于排队线程过多造成的数据库访问失败             */            build.maxWaitTime(1000*60*2);            build.connectTimeout(1000*60*1);    //与数据库建立连接的timeout设置为1分钟            MongoClientOptions myOptions = build.build();                   try {                //数据库连接实例                mongoClient = new MongoClient("127.0.0.1", myOptions);                      } catch (UnknownHostException e) {                // TODO 这里写异常处理的代码                e.printStackTrace();            } catch (MongoException e){                e.printStackTrace();            }        }    }    /********单例模式声明开始，采用饿汉式方式生成，保证线程安全********************/    //类初始化时，自行实例化，饿汉式单例模式    private static final MongoDBDaoImpl mongoDBDaoImpl = new MongoDBDaoImpl();    /**     *      * 方法名：getMongoDBDaoImplInstance     * 作者：zhouyh     * 创建时间：2014-8-30 下午04:29:26     * 描述：单例的静态工厂方法     * @return     */    public static MongoDBDaoImpl getMongoDBDaoImplInstance(){        return mongoDBDaoImpl;    }    /************************单例模式声明结束*************************************/    @Override    public boolean delete(String dbName, String collectionName, String[] keys,            Object[] values) {        DB db = null;        DBCollection dbCollection = null;        if(keys!=null && values!=null){            if(keys.length != values.length){   //如果keys和values不对等，直接返回false                return false;            }else{                try {                    db = mongoClient.getDB(dbName); //获取指定的数据库                    dbCollection = db.getCollection(collectionName);    //获取指定的collectionName集合                    BasicDBObject doc = new BasicDBObject();    //构建删除条件                    WriteResult result = null;  //删除返回结果                    String resultString = null;                    for(int i=0; i<keys.length; i++){                        doc.put(keys[i], values[i]);    //添加删除的条件                    }                    result = dbCollection.remove(doc);  //执行删除操作                    resultString = result.getError();                    if(null != db){                        try {                            db.requestDone();   //请求结束后关闭db                            db = null;                        } catch (Exception e) {                            // TODO: handle exception                            e.printStackTrace();                        }                    }                    return (resultString!=null) ? false : true; //根据删除执行结果进行判断后返回结果                } catch (Exception e) {                    // TODO: handle exception                    e.printStackTrace();                } finally{                    if(null != db){                        db.requestDone();   //关闭db                        db = null;                    }                }            }        }        return false;    }    @Override    public ArrayList<DBObject> find(String dbName, String collectionName,            String[] keys, Object[] values, int num) {        ArrayList<DBObject> resultList = new ArrayList<DBObject>(); //创建返回的结果集        DB db = null;        DBCollection dbCollection = null;        DBCursor cursor = null;        if(keys!=null && values!=null){            if(keys.length != values.length){                return resultList;  //如果传来的查询参数对不对，直接返回空的结果集            }else{                try {                    db = mongoClient.getDB(dbName); //获取数据库实例                    dbCollection = db.getCollection(collectionName);    //获取数据库中指定的collection集合                    BasicDBObject queryObj = new BasicDBObject();   //构建查询条件                    for(int i=0; i<keys.length; i++){   //填充查询条件                        queryObj.put(keys[i], values[i]);                    }                                   cursor = dbCollection.find(queryObj);   //查询获取数据                    int count = 0;                    if(num != -1){  //判断是否是返回全部数据，num=-1返回查询全部数据，num!=-1则返回指定的num数据                        while(count<num && cursor.hasNext()){                            resultList.add(cursor.next());                            count++;                        }                        return resultList;                    }else{                        while(cursor.hasNext()){                            resultList.add(cursor.next());                        }                        return resultList;                    }                } catch (Exception e) {                    // TODO: handle exception                } finally{                                  if(null != cursor){                        cursor.close();                    }                    if(null != db){                        db.requestDone();   //关闭数据库请求                    }                }            }        }        return resultList;    }    @Override    public DBCollection getCollection(String dbName, String collectionName) {        // TODO Auto-generated method stub        return mongoClient.getDB(dbName).getCollection(collectionName);    }    @Override    public DB getDb(String dbName) {        // TODO Auto-generated method stub        return mongoClient.getDB(dbName);    }    @Override    public boolean inSert(String dbName, String collectionName, String[] keys,            Object[] values) {        DB db = null;        DBCollection dbCollection = null;        WriteResult result = null;        String resultString = null;        if(keys!=null && values!=null){            if(keys.length != values.length){                return false;            }else{                db = mongoClient.getDB(dbName); //获取数据库实例                dbCollection = db.getCollection(collectionName);    //获取数据库中指定的collection集合                BasicDBObject insertObj = new BasicDBObject();                for(int i=0; i<keys.length; i++){   //构建添加条件                    insertObj.put(keys[i], values[i]);                }                try {                    result = dbCollection.insert(insertObj);                    resultString = result.getError();                } catch (Exception e) {                    // TODO: handle exception                    e.printStackTrace();                }finally{                    if(null != db){                        db.requestDone();   //请求结束后关闭db                    }                }                               return (resultString != null) ? false : true;            }        }        return false;    }    @Override    public boolean isExit(String dbName, String collectionName, String key,            Object value) {        // TODO Auto-generated method stub        DB db = null;        DBCollection dbCollection = null;        if(key!=null && value!=null){            try {                db = mongoClient.getDB(dbName); //获取数据库实例                dbCollection = db.getCollection(collectionName);    //获取数据库中指定的collection集合                BasicDBObject obj = new BasicDBObject();    //构建查询条件                obj.put(key, value);                if(dbCollection.count(obj) > 0) {                    return true;                }else{                    return false;                }            } catch (Exception e) {                // TODO: handle exception                e.printStackTrace();            } finally{                if(null != db){                    db.requestDone();   //关闭db                    db = null;                }            }        }        return false;    }    @Override    public boolean update(String dbName, String collectionName,            DBObject oldValue, DBObject newValue) {        DB db = null;        DBCollection dbCollection = null;        WriteResult result = null;        String resultString = null;        if(oldValue.equals(newValue)){            return true;        }else{            try {                db = mongoClient.getDB(dbName); //获取数据库实例                dbCollection = db.getCollection(collectionName);    //获取数据库中指定的collection集合                result = dbCollection.update(oldValue, newValue);                resultString = result.getError();                return (resultString!=null) ? false : true;            } catch (Exception e) {                // TODO: handle exception                e.printStackTrace();            } finally{                if(null != db){                    db.requestDone();   //关闭db                    db = null;                }            }        }        return false;    }    /**     * 方法名：main     * 作者：zhouyh     * 创建时间：2014-8-30 下午04:21:11     * 描述：TODO(这里用一句话描述这个方法的作用)     * @param args     */    public static void main(String[] args) {        // TODO Auto-generated method stub    }}

0 0