java封装Mongodb3.2.1工具类
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了java封装Mongodb3.2.1工具类相关的知识，希望对你有一定的参考价值。
由于最近项目要使用mongodb来处理一些日志，提前学习了一下mongodb的一些基本用法，大概写了一些常用的。
开发环境为:WIN7-64,JDK7-64,MAVEN3.3.9-64,IDEA2017-64.
程序基本结构为:
下面贴出核心代码示例：
pom.xml
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd"> <modelVersion>4.0.0</modelVersion> <groupId>TestWebProjectMaven</groupId> <artifactId>TestWebProjectMaven</artifactId> <packaging>war</packaging> <version>1.0-SNAPSHOT</version> <name>TestWebProjectMaven Maven Webapp</name> <!-- 设定主仓库 --> <repositories> <!-- nexus私服 --> <repository> <id>nexus-repos</id> <name>Team Nexus Repository</name> <url>http://192.168.200.205:8081/nexus/content/groups/public/</url> <releases> <enabled>true</enabled> </releases> <snapshots> <enabled>true</enabled> </snapshots> </repository> </repositories> <!-- 设定插件仓库 --> <pluginRepositories> <pluginRepository> <id>nexus-repos</id> <name>Team Nexus Repository</name> <url>http://192.168.200.205:8081/nexus/content/groups/public/</url> <releases> <enabled>true</enabled> </releases> <snapshots> <enabled>true</enabled> </snapshots> </pluginRepository> </pluginRepositories> <dependencies> <dependency> <groupId>junit</groupId> <artifactId>junit</artifactId> <version>3.8.1</version> <scope>test</scope> </dependency> <dependency> <groupId>org.springframework</groupId> <artifactId>spring-webmvc</artifactId> <version>4.1.6.RELEASE</version> </dependency> <dependency> <groupId>org.springframework</groupId> <artifactId>spring-context</artifactId> <version>4.1.6.RELEASE</version> </dependency> <dependency> <groupId>org.mongodb</groupId> <artifactId>mongo-java-driver</artifactId> <version>3.2.1</version> </dependency> <dependency> <groupId>org.jetbrains</groupId> <artifactId>annotations-java5</artifactId> <version>RELEASE</version> </dependency> <dependency> <groupId>commons-configuration</groupId> <artifactId>commons-configuration</artifactId> <version>1.10</version> </dependency> </dependencies> <build> <finalName>TestWebProjectMaven</finalName> <!-- 设置properties文件编译到target目录中,不然读取不到配置文件 --> <resources> <resource> <directory>src/main/java</directory> <includes> <include>**/*.properties</include> </includes> </resource> <resource> <directory>src/main/resources</directory> </resource> </resources> </build> </project>
mongodb.properties
MONGODB_IP=192.168.200.234 MONGODB_PORT=10143 MONGODB_DATABASE_NAME=runoob MONGODB_COLLECTION_NAME=test
MongodbUtil
package org.mbox.util;
import com.mongodb.MongoClient;
import com.mongodb.client.*;
import com.mongodb.client.model.Filters;
import com.mongodb.client.model.UpdateOptions;
import com.mongodb.client.result.DeleteResult;
import org.apache.commons.configuration.CompositeConfiguration;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.bson.types.ObjectId;
import org.mbox.model.PageVO;
import java.beans.IntrospectionException;
import java.beans.PropertyDescriptor;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.*;
/**
* Created by pc on 2017/6/5.
*/
public class MongodbUtil {
private static MongoClient MONGODB_CLIENT = null;
private static String MONGODB_IP = null;
private static Integer MONGODB_PORT = null;
private static String MONGODB_DATABASE_NAME = null;
private static String MONGODB_COLLECTION_NAME = null;
static{
CompositeConfiguration compositeConfiguration = new CompositeConfiguration();
try {
compositeConfiguration.addConfiguration(new PropertiesConfiguration("mongodb.properties"));
} catch (ConfigurationException e) {
e.printStackTrace();
}
MONGODB_IP = compositeConfiguration.getString("MONGODB_IP");
MONGODB_PORT = compositeConfiguration.getInt("MONGODB_PORT");
MONGODB_DATABASE_NAME = compositeConfiguration.getString("MONGODB_DATABASE_NAME");
MONGODB_COLLECTION_NAME = compositeConfiguration.getString("MONGODB_COLLECTION_NAME");
MONGODB_CLIENT = new MongoClient(MONGODB_IP,MONGODB_PORT);
}
private MongodbUtil() {
}
/**
* 初始化mongodb数据源
* @return
*/
public static MongoDatabase getMongodbDatabase(){
return MONGODB_CLIENT.getDatabase(MONGODB_DATABASE_NAME);
}
/**
* 关闭MongoClient连接
*/
public static void closeMongodbClient(){
if(null != MONGODB_CLIENT){
MONGODB_CLIENT.close();
MONGODB_CLIENT = null;
}
}
/**
* 获取mongodb的表对象
* @return
*/
public static MongoCollection<Document> getMongoCollection(){
return getMongodbDatabase().getCollection(MONGODB_COLLECTION_NAME);
}
/**
* 通过map插入一条数据到表中
* @param map
*/
public static void insertOneCollectionByMap(Map<String,Object> map){
getMongoCollection().insertOne(handleMap(map));
}
/**
* 通过集合map一次性插入多条数据到表中
* @param listMap
*/
public static void insertManyCollectionByMap(List<Map<String,Object>> listMap){
List<Document> list = new ArrayList<Document>();
for(Map<String,Object> map : listMap){
Document document = handleMap(map);
list.add(document);
}
getMongoCollection().insertMany(list);
}
/**
* 通过实体对象插入一条数据到表中
* @param obj
*/
public static void insertOneCollectionByModel(Object obj){
getMongoCollection().insertOne(handleModel(obj));
}
/**
* 通过集合实体对象一次性插入多条数据到表中
* @param listObj
*/
public static void insertManyCollectionByModel(List<Object> listObj){
List<Document> list = new ArrayList<Document>();
for(Object obj : listObj){
Document document = handleModel(obj);
list.add(document);
}
getMongoCollection().insertMany(list);
}
/**
* 通过手工拼接条件获取查询结果集
* 下面是拼接queryDocument例子
* document = new Document();
* 要注意value中的数据类型
* document.append("num",new Document("$eq",20));//相等
* document.append("num",new Document("$age",20));//不相等
* document.append("num",new Document("$gt",20));//大于
* document.append("num",new Document("$gte",21));//大于等于
* document.append("num",new Document("$lte",21));//小于等于
* document.append("num",new Document("$lt",21));//小于
* 下面是或的写法
* List<Document> documentList = new ArrayList<Document>();
* documentList.add(new Document("num",1));
* documentList.add(new Document("num",2));
* document.append("$or",documentList);
* @param queryDocument
* @param sortDocument
* @param pageVO
* @return
*/
public static String queryCollectionByCondition(Document queryDocument,Document sortDocument,PageVO pageVO){
if(null == queryDocument || null == sortDocument || null == pageVO){
return null;
}else{
String returnList = getQueryCollectionResult(queryDocument,sortDocument,pageVO);
return returnList;
}
}
/**
* 通过不定条件map查询出表中的所有数据,只限于等于的条件
* @param map
* @param sortDocument
* @param pageVO
* @return
*/
public static String queryCollectionByMap(Map<String,Object> map,Document sortDocument,PageVO pageVO){
String sql = getQueryCollectionResult(handleMap(map),sortDocument,pageVO);
return sql;
}
/**
* 通过不定条件实体对象obj查询出表中的所有数据,只限于等于的条件
* @param obj
* @param sortDocument
* @param pageVO
* @return
*/
public static String queryCollectionByModel(Object obj,Document sortDocument,PageVO pageVO){
String sql = getQueryCollectionResult(handleModel(obj),sortDocument,pageVO);
return sql;
}
/**
* 接收Document组装的查询对象,处理子集查询结果并以JSON的形式返回前端
* @param queryDocument
* @param sortDocument
* @param pageVO
* @return
*/
private static String getQueryCollectionResult(Document queryDocument,Document sortDocument,PageVO pageVO){
FindIterable<Document> findIterable = getMongoCollection().find(queryDocument)
.sort(sortDocument).skip((pageVO.getPageNum()-1)*pageVO.getPageSize()).limit(pageVO.getPageSize());
MongoCursor<Document> mongoCursor = findIterable.iterator();
StringBuilder sql = new StringBuilder();
Integer lineNum = 0;
while(mongoCursor.hasNext()){
sql.append("{");
Document documentVal = mongoCursor.next();
Set<Map.Entry<String,Object>> sets = documentVal.entrySet();
Iterator<Map.Entry<String,Object>> iterators = sets.iterator();
while(iterators.hasNext()){
Map.Entry<String,Object> map = iterators.next();
String key = map.getKey();
Object value = map.getValue();
sql.append("\"");
sql.append(key);
sql.append("\"");
sql.append(":");
sql.append("\"");
sql.append((value == null ? "" : value));
sql.append("\",");
}
sql.deleteCharAt(sql.lastIndexOf(","));
sql.append("},");
lineNum++;
}
//这里判断是防止上述没值的情况
if(sql.length() > 0){
sql.deleteCharAt(sql.lastIndexOf(","));
}
String returnList = getFinalQueryResultsSql(lineNum,sql.toString());
return returnList;
}
/**
* 拼接返回前端的JSON串
* @param lineNum 子集中JSON的条数
* @param querySql 子集中的所有结果JSON
* @return
*/
private static String getFinalQueryResultsSql(Integer lineNum,String querySql) {
StringBuilder sql = new StringBuilder();
sql.append("{");
sql.append("\"");
sql.append("jsonRoot");
sql.append("\"");
sql.append(":");
sql.append("\"");
sql.append(lineNum);
sql.append("\",");
sql.append("\"");
sql.append("jsonList");
sql.append("\"");
sql.append(":");
sql.append("[");
sql.append(querySql);
sql.append("]");
sql.append("}");
return sql.toString();
}
/**
* 以list的形式获取mongdb库中的所有表
* @return
*/
public static List<String> getALLCollectionNameOfList(){
List<String> list = new ArrayList<String>();
MongoIterable<String> mongoIterable = getMongodbDatabase().listCollectionNames();
for(String name : mongoIterable){
list.add(name);
}
return list;
}
/**
* 以map的形式获取mongdb库中的所有表
* @return
*/
public static Map<String,String> getALLCollectionNameOfMap() {
Map<String,String> map = new HashMap<String,String>();
MongoIterable<String> mongoIterable = getMongodbDatabase().listCollectionNames();
for(String name : mongoIterable){
map.put(name,name);
}
return map;
}
/**
* 获取表中的数据条数
* @param queryDocument 传null为查询表中所有数据
* @return
*/
public static Integer queryCollectionCount(Document queryDocument){
int count = (int) getMongoCollection().count(queryDocument);
return count;
}
/**
* 通过表ID获取某条数据
* @param id
* @return
*/
public static String queryCollectionModelById(String id){
ObjectId objectId = new ObjectId(id);//注意在处理主键问题上一定要用ObjectId转一下
Document document = getMongoCollection().find(Filters.eq("_id",objectId)).first();
return (document == null ? null : document.toJson());
}
/**
* 根据ID更新某一条数据
* @param id 查询条件主键ID
* @param updateMap 更新内容,如果是此ID中不存在的字段,那么会在此ID对应的数据中加入新的字段内容
* 注意这里跟updateOptions.upsert(ifInsert);没关系
*/
public static void updateCollectionById(String id,Map<String,Object> updateMap){
Document queryDocument = new Document();
ObjectId objId = new ObjectId(id);//注意在处理主键问题上一定要用ObjectId转一下
queryDocument.append("_id", objId);
Document updateDocument = handleMap(updateMap);
getMongoCollection().updateOne(queryDocument,new Document("$set",updateDocument));
}
/**
* 根据某几个字段更新多条数据,document的条件拼接可参考queryCollectionByCondition方法
* @param queryDocument 查询条件,一定不要加_id,根据ID处理的话参考updateCollectionById方法
* @param updateDocument 更新内容,当查询条件和更新内容有出入并且ifInsert为true时才插入
* @param ifInsert 数据不存在是否插入,true插入,false不插入
*/
public static void updateCollectionByCondition(Document queryDocument,Document updateDocument,Boolean
ifInsert){
UpdateOptions updateOptions = new UpdateOptions();
updateOptions.upsert(ifInsert);
getMongoCollection().updateMany(queryDocument,new Document("$set",updateDocument),updateOptions);
}
/**
* 根据ID删除某一条数据
* @param id
* @return
*/
public static Integer deleteCollectionById(String id){
ObjectId objectId = new ObjectId(id);
Bson bson = Filters.eq("_id",objectId);
DeleteResult deleteResult = getMongoCollection().deleteOne(bson);
int count = (int) deleteResult.getDeletedCount();
return count;
}
/**
* 根据MAP删除表中的某些数据
* @param map
*/
public static void deleteCollectionByMap(Map<String,Object> map){
getMongoCollection().deleteMany(handleMap(map));
}
/**
* 根据实体对象删除表中的某些数据
* @param obj
*/
public static void deleteCollectionByModel(Object obj){
getMongoCollection().deleteMany(handleModel(obj));
}
/**
* 根据预先手工拼接的document删除表中的某些数据
* @param document
*/
public static void deleteCollectionByDocument(Document document){
getMongoCollection().deleteMany(document);
}
/**
* 通过实体对象obj拼接document
* @param obj
* @return
*/
private static Document handleModel(Object obj){
Document document = null;
if(obj != null){
document = new Document();
try {
Class clz = obj.getClass();
Field fields[] = clz.getDeclaredFields();
for(Field field : fields){
String fieldName = field.getName();
PropertyDescriptor propertyDescriptor = new PropertyDescriptor(fieldName,clz);
Method method = propertyDescriptor.getReadMethod();
Object fieldValue = method.invoke(obj);
document.append(fieldName,(fieldValue == null ? "" : fieldValue));
}
} catch (IntrospectionException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (InvocationTargetException e) {
e.printStackTrace();
}
}else{
document = new Document("","");
}
return document;
}
/**
* 通过集合map拼接document
* @param map
* @return
*/
private static Document handleMap(Map<String,Object> map){
Document document = null;
if(null != map){
document = new Document();
Set<String> sets = map.keySet();
Iterator<String> iterators = sets.iterator();
while(iterators.hasNext()){
String key = iterators.next();
Object value = map.get(key);
document.append(key,(value == null ? "" : value));
}
}else{
document = new Document("","");//这种设置查询不到任何数据
}
return document;
}
/**
* 删除某个库
* @param databaseName
*/
public static void dropDatabase(String databaseName){
MONGODB_CLIENT.dropDatabase(databaseName);
}
/**
* 删除某个库中的某个表
* @param databaseName
* @param collectionName
*/
public static void dropCollection(String databaseName,String collectionName){
MONGODB_CLIENT.getDatabase(databaseName).getCollection(collectionName).drop();
}
/**
* 下述方式个人并不推荐,没有直接用document直接拼串方便
*/
public static void testquery(){
List<Integer> list = new ArrayList<Integer>();
list.add(20);
list.add(21);
list.add(22);
FindIterable<Document> findIterable =
//getMongoCollection().find(Filters.and(Filters.lt("num",22),Filters.gt("num",17)));
//getMongoCollection().find(Filters.in("num",17,18));
getMongoCollection().find(Filters.nin("num",list));
MongoCursor<Document> mongoCursor = findIterable.iterator();
while(mongoCursor.hasNext()){
Document document = mongoCursor.next();
System.out.println(document.toJson());
}
}
}本文出自 “你可以选择不平凡” 博客，请务必保留此出处http://ylcodes01.blog.51cto.com/5607366/1933342
以上是关于java封装Mongodb3.2.1工具类的主要内容，如果未能解决你的问题，请参考以下文章
elasticsearch代码片段，及工具类SearchEsUtil.java