java调用mongodb的工具类-改动

最新推荐文章于 2024-12-23 09:18:26 发布

原创最新推荐文章于 2024-12-23 09:18:26 发布 · 183 阅读

本内容遵循CC 4.0 BY-SA版权协议

本文详细介绍了使用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<Document> queryList(String collName){
return queryList(collName,"","");
}
/**
* 符合条件的数据列表
* @param collName 集合名
*/
public static List<Document> queryList(String collName,String key,String value){
try {
dbBase = getDbBase();
MongoCollection<Document> list = dbBase.getCollection(collName);
FindIterable<Document> findIterable;
if(StringUtils.isNoneBlank(key)){
findIterable = list.find(Filters.eq(key, value));
}else{
findIterable = list.find();
}
MongoCursor<Document> mongoCursor = findIterable.iterator();
List<Document> 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<Document> 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() );
}
}

}