Java与MongoDB的配合使用备忘1-CURD操作模板创建

为了提高代码的复用率和可维护性，将每个操作中重复的数据库链接、集合获取等方法代码提取到操作模板类中，并在操作模板类中封装了CURD操作方法。

package com.dataudata.mongodb;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.lang.StringUtils;
import org.bson.Document;

import com.dataudata.mongodb.MongoDBConfig;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientOptions;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.model.Filters;
import com.mongodb.client.result.DeleteResult;

/**
 * mongodb基础操作模板类
 */
public class MongoDBTemplate{
	private MongoClient mongoClient = null;
	private static final MongoDBTemplate mongoDBDao = new MongoDBTemplate();// 饿汉式单例模式

	private MongoDBTemplate() {
		if (mongoClient == null) {
			MongoClientOptions.Builder buide = new MongoClientOptions.Builder();
			buide.connectionsPerHost(100);// 与目标数据库可以建立的最大链接数
			buide.connectTimeout(1000 * 60);// 与数据库建立链接的超时时间
			buide.maxWaitTime(100 * 60);// 一个线程成功获取到一个可用数据库之前的最大等待时间
			buide.threadsAllowedToBlockForConnectionMultiplier(100);// 线程阻塞时的最大等待数量
			buide.maxConnectionIdleTime(1000 * 60 * 20);// 连接的最大空闲时间，超过会自动释放
			buide.maxConnectionLifeTime(1000 * 60 * 60);// 连接的最大存活时间
			buide.socketTimeout(0);
			buide.socketKeepAlive(true);
			MongoClientOptions myOptions = buide.build();

			String user = MongoDBConfig.getUser();
			if (StringUtils.isNotBlank(user)) {
				// MongoCredential.createScramSha1Credential()三个参数分别为 用户名 数据库名称
				// 密码
				MongoCredential credential = MongoCredential
						.createScramSha1Credential(user, MongoDBConfig
								.getDefaultDatabase(), MongoDBConfig.getPassword()
								.toCharArray());
				List<MongoCredential> credentials = new ArrayList<MongoCredential>();
				credentials.add(credential);
				mongoClient = new MongoClient(new ServerAddress(
						MongoDBConfig.getHost(), MongoDBConfig.getPort()),
						credentials, myOptions);
			} else {
				mongoClient = new MongoClient(new ServerAddress(
						MongoDBConfig.getHost(), MongoDBConfig.getPort()),
						myOptions);
			}
		}
	}

	public static MongoDBTemplate getMongoDBTemplate() {
		return mongoDBDao;
	}

	/**
	 * 获取指定的数据库
	 * 
	 * @param dbName 数据库名称
	 * @return
	 */
	public MongoDatabase getDatabase(String dbName) {
		return mongoClient.getDatabase(dbName);
	}

	/**
	 * 获取默认的数据
	 * 
	 * @return
	 */
	public MongoDatabase getDatabase() {
		return getDatabase(MongoDBConfig.getDatabase());
	}

	/**
	 * 获取表
	 * 
	 * @param dbName 数据库名 
	 * @param collectionName 表名
	 * @return
	 */
	public MongoCollection<Document> getCollection(String dbName,
			String collectionName) {
		MongoDatabase db = getDatabase(dbName);
		return db.getCollection(collectionName);
	}
	
	/**
	 * 从默认的数据库连接表
	 * 
	 * @param collectionName 表名 
	 * @return
	 */
	public MongoCollection<Document> getCollection(String collectionName) {
		MongoDatabase db = getDatabase();
		return db.getCollection(collectionName);
	}

	/**
	 * 插入单个数据
	 * 
	 * @param dbName 数据库
	 * @param collectionName 表名
	 * @param document 保存的数据
	 * @return
	 */
	public boolean insert(String dbName, String collectionName,
			Document document) {
		List<Document> documents = new ArrayList<Document>();
		documents.add(document);
		return insert(dbName, collectionName, documents);
	}
	
	/**
	 * 插入数据
	 * 
	 * @param dbName
	 * @param collectionName
	 * @param documents
	 * @return
	 */
	public boolean insert(String dbName, String collectionName,
			List<Document> documents) {
		MongoDatabase db = getDatabase(dbName);
		MongoCollection<Document> dbCollection = db
				.getCollection(collectionName);
		long num = dbCollection.count();
		/**
		 * 1. 创建文档 org.bson.Document 参数为key-value的格式 
		 * 2. 创建文档集合List<Document> 
		 * 3. 将文档集合插入数据库集合中 mongoCollection.insertMany(List<Document>) 插入单个文档可以用
		 * mongoCollection.insertOne(Document)
		 * */
		dbCollection.insertMany(documents);
		if (dbCollection.count() - num > 0) {
			return true;
		}
		return false;
	}
	
	/**
	 * 插入数据
	 * 
	 * @param collectionName
	 * @param document
	 * @return
	 */
	public boolean insert(String collectionName, Document document) {
		return insert(MongoDBConfig.getDatabase(), collectionName, document);
	}
	
	/**
	 * 插入数据
	 * 
	 * @param collectionName
	 * @param documents
	 * @return
	 */
	public boolean insert(String collectionName, List<Document> documents) {
		return insert(MongoDBConfig.getDatabase(), collectionName, documents);
	}

	/**
	 * 数据指定数据库下的数据
	 * 
	 * @param dbName
	 * @param collectionName
	 * @param keys
	 * @param values
	 * @return
	 */
	public boolean delete(String dbName, String collectionName, String keys,
			Object values) {
		MongoDatabase db = getDatabase(dbName);
		MongoCollection<Document> dbCollection = db
				.getCollection(collectionName);
		// 删除所有符合条件的文档
		DeleteResult result = dbCollection.deleteMany(Filters.eq(keys, values));
		if (result.getDeletedCount() > 0) {
			return true;
		}
		return false;
	}
	
	/**
	 * 删除默认数据库下的数据
	 * 
	 * @param collectionName 
	 * @param keys
	 * @param values
	 * @return
	 */
	public boolean delete(String collectionName, String keys, Object values) {
		return delete(MongoDBConfig.getDatabase(), collectionName, keys, values);
	}
	
	public static void main(String args[]) {
		MongoDBTemplate mongoDBDaoImpl = MongoDBTemplate.getMongoDBTemplate();
		ArrayList<DBObject> list = new ArrayList<DBObject>();
	}
}

其中MongoDBConfig是一个自定义工具类，用于读取自定义的MongoDB配置文件，可在自定义配置文件中配置数据库地址、数据库端口、默认集合、用户名、密码等信息。