java调用mongodb的工具类-改动

 

import com.mongodb.MongoClient;
import com.mongodb.MongoClientOptions;
import com.mongodb.MongoClientOptions.Builder;
import com.mongodb.WriteConcern;
import com.mongodb.client.FindIterable;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Filters;
import org.apache.commons.lang3.StringUtils;
import org.bson.Document;

import java.util.ArrayList;
import java.util.List;

public class MongoUtil {
    private static MongoClient mongoClient;
    private static MongoDatabase dbBase;

    static {
        String ip = "10.30.225.5";
        int port =27017;
        mongoClient = new MongoClient(ip, port);
        // 就需要在客户端提供用户名和密码：
        // boolean auth = db.authenticate(myUserName, myPassword);
        Builder options = new MongoClientOptions.Builder();
        options.cursorFinalizerEnabled(true);
        // options.autoConnectRetry(true);// 自动重连true
        // options.maxAutoConnectRetryTime(10); // the maximum auto connect retry time
        options.connectionsPerHost(200);// 连接池设置为300个连接,默认为100
        options.connectTimeout(30000);// 连接超时，推荐>3000毫秒
        options.maxWaitTime(5000); //
        options.socketTimeout(0);// 套接字超时时间，0无限制
        options.threadsAllowedToBlockForConnectionMultiplier(5000);// 线程队列数，如果连接线程排满了队列就会抛出“Out of semaphores to get db”错误。
        options.writeConcern(WriteConcern.SAFE);//
        options.build();
    }

    public static MongoDatabase getDbBase(){
        if(dbBase == null){
            return mongoClient.getDatabase("dbBaseTest");
        }
        return dbBase;
    }

    /**
     * 统计总数
     * @param collName
     * @return
     */
   public static int getCount(String collName) {
       dbBase = getDbBase();
       int count = (int) dbBase.getCollection(collName).count();
       return count;
   }

    /**
     * 符合条件的数据
     * @param collName 集合名
     */
    public static Document get(String collName,String key,String value){
        try {
            dbBase = getDbBase();
            Document doc = dbBase.getCollection(collName).find(Filters.eq(key, value)).first();
            return doc;
        }catch(Exception e){
            System.err.println( "！！数据查询异常：" + e.getMessage() );
            return null;
        }
    }

    public static List queryList(String collName){
        return queryList(collName,"","");
    }
    /**
     * 符合条件的数据列表
     * @param collName 集合名
     */
    public static List queryList(String collName,String key,String value){
        try {
            dbBase = getDbBase();
            MongoCollection list = dbBase.getCollection(collName);
            FindIterable findIterable;
            if(StringUtils.isNoneBlank(key)){
                findIterable = list.find(Filters.eq(key, value));
            }else{
                findIterable = list.find();
            }
            MongoCursor mongoCursor = findIterable.iterator();
            List docs = new ArrayList<>();
            while (mongoCursor.hasNext()) {
                docs.add(mongoCursor.next());
            }
            return docs;
        }catch(Exception e){
            System.err.println( "！！数据查询异常：" + e.getMessage() );
            return null;
        }
    }

    /**
     * 新增一条
     * @param collName 集合名
     * @param map 数据列表
     */
    public static void add(String collName,Document map) {
        try {
            dbBase = getDbBase();
            dbBase.getCollection(collName).insertOne(map);
            System.out.println("数据插入成功");
        }catch(Exception e){
            System.err.println( "！！数据新增异常：" + e.getMessage() );
        }
    }

    /**
     * 新增多条
     * @param collName 集合名
     * @param list 数据列表
     */
    public static void addBatch(String collName,List list) {
        try {
            dbBase = getDbBase();
            dbBase.getCollection(collName).insertMany(list);
            System.out.println("数据批量插入成功");
        }catch(Exception e){
            System.err.println( "！！数据新增异常：" + e.getMessage() );
        }
    }

    /**
     * 删除符合条件( key=value)的第一个文档
     * @param collName 集合名
     * @param key 文档中的key
     * @param value 文档中的value
     */
    public static void delete(String collName,String key,String value){
        try {
            dbBase = getDbBase();
            dbBase.getCollection(collName).deleteOne(Filters.eq(key, value));
        }catch(Exception e){
            System.err.println( "！！数据删除异常：" + e.getMessage() );
        }
    }

    /**
     * 删除该集合
     * @param collName
     */
    public static void dropColl(String collName){
        dbBase = getDbBase();
        dbBase.getCollection(collName).drop();
    }
    /**
     *  删除所有符合条件( key=value)的文档
     * @param collName 集合名
     * @param key 文档中的key
     * @param value 文档中的vaule
     */
    public static void deleteBatch(String collName,String key,String value){
        try {
            dbBase = getDbBase();
            dbBase.getCollection(collName).deleteMany(Filters.eq(key, value));
        }catch(Exception e){
            System.err.println( "！！数据删除异常：" + e.getMessage() );
        }
    }

}