me.wuwenbin.modules.mongodb.factory.MongoFactory Maven / Gradle / Ivy
Show all versions of template-modules-mongodb Show documentation
package me.wuwenbin.modules.mongodb.factory;
import com.mongodb.MongoClient;
import com.mongodb.MongoClientOptions;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import me.wuwenbin.modules.mongodb.dao.MongoAccessor;
import me.wuwenbin.modules.mongodb.dao.MongoDbTemplate;
import me.wuwenbin.modules.mongodb.support.MongoContextHolder;
import me.wuwenbin.modules.mongodb.support.core.MongoDataSource;
import me.wuwenbin.modules.mongodb.support.exception.DataSourceKeyNotExistException;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.stereotype.Component;
import java.util.Hashtable;
import java.util.List;
import java.util.Map;
/**
* 和spring一起初始化做一些操作,并实例化MongoClient集群
* Created by wuwenbin on 2017/4/22.
*/
@Component
public class MongoFactory implements InitializingBean {
/**
* 多个mongo连接（一个集群是在一个mongo中）此处的多个仅仅是指配置上的多台mongo数据源，并不是集群
*/
private Map mongoDataSources;
/**
* 存放mongoDao的map类
*/
private Map mongoDaoMap = new Hashtable<>();
private MongoDataSource defaultMongoDataSource;
private MongoDataSource dynamicMongoDataSource;
/**
* 此Mongo连接的默认操作的数据库的dao
*/
public MongoAccessor defaultMongoDao;
/**
* 此Mongo连接的动态操作数据库的dao，默认值和{@link #defaultMongoDao}一样
*/
public MongoAccessor dynamicMongoDao;
public void setMongoDataSources(Map mongoDataSources) {
this.mongoDataSources = mongoDataSources;
}
/**
* 根据xml中配置设置默认的MongoDao
*
* @param dataSource
*/
public synchronized void setDefaultMongoDao(MongoDataSource dataSource) {
this.defaultMongoDataSource = dataSource;
this.dynamicMongoDataSource = this.defaultMongoDataSource;
List seeds = dataSource.getSeeds();
MongoClientOptions options = dataSource.getMongoClientOptions();
List credentials = dataSource.getMongoCredentials();
MongoClient mongoClient = new MongoClient(seeds, credentials, options);
MongoDbFactory dbFactory = new SimpleMongoDbFactory(mongoClient, dataSource.getDefaultDatabase());
this.defaultMongoDao = new MongoDbTemplate(new MongoTemplate(dbFactory, dataSource.getMongoConverter()));
this.dynamicMongoDao = this.defaultMongoDao;
}
/**
* 根据当前线程中的key变量值来动态数据源。
*
* 有3中情况切换.
* 1、仅切换数据源，即根据xml中配置的key来切换不同数据源的mongo。变动数据源，默认数据库按照初始化的不变
* 2、仅切换默认操作文档（数据库），即变更此同一数据源下的文档（数据库）。数据源不变，变动操作文档（数据库）
* 3、以上两种情况同时发生
*/
public synchronized void determineDynamicMongoDao() {
String currentThreadKey = MongoContextHolder.getHolder().getKey();
String currentThreadDatabase = MongoContextHolder.getHolder().getDatabase();
if (!mongoDataSources.containsKey(currentThreadKey)) {
throw new DataSourceKeyNotExistException();
}
this.dynamicMongoDataSource = mongoDataSources.get(currentThreadKey);
this.dynamicMongoDao = getMongoDbDaoByKeyAndDataBase(currentThreadKey, currentThreadDatabase);
}
public synchronized void determineDynamicMongoDaoByKey() {
String currentThreadKey = MongoContextHolder.getHolder().getKey();
if (!mongoDataSources.containsKey(currentThreadKey)) {
throw new DataSourceKeyNotExistException();
}
this.dynamicMongoDataSource = mongoDataSources.get(currentThreadKey);
this.dynamicMongoDao = getMongoDbDaoByKey(currentThreadKey);
}
public synchronized void determineDynamicMongoDaoByDatabase() {
String currentThreadDatabase = MongoContextHolder.getHolder().getDatabase();
this.dynamicMongoDao = getMongoDbDaoByDataBase(currentThreadDatabase);
}
@Override
public void afterPropertiesSet() {
if (mongoDataSources != null && mongoDataSources.size() > 0) {
for (String key : mongoDataSources.keySet()) {
if (mongoDaoMap == null || !mongoDaoMap.containsKey(key)) {
mongoDaoMap.put(key, getMongoDbDaoByKey(key));
}
}
}
}
/**
* 根据key和默认数据库（文档）来获取MongoDbDao实例
*
* @param key
* @param defaultDatabase
* @return
*/
private MongoAccessor getMongoDbDaoByKeyAndDataBase(String key, String defaultDatabase) {
if (mongoDataSources.containsKey(key)) {
MongoDataSource dataSource = mongoDataSources.get(key);
List seeds = dataSource.getSeeds();
MongoClientOptions options = dataSource.getMongoClientOptions();
List credentials = dataSource.getMongoCredentials();
MongoClient mongoClient = new MongoClient(seeds, credentials, options);
MongoDbFactory dbFactory = new SimpleMongoDbFactory(mongoClient, defaultDatabase);
return new MongoDbTemplate(new MongoTemplate(dbFactory, dataSource.getMongoConverter()));
} else {
throw new DataSourceKeyNotExistException();
}
}
/**
* 同{@link #getMongoDbDaoByKeyAndDataBase(String, String)}，默认数据库为初始化设置的
*
* @param key
* @return
*/
private MongoAccessor getMongoDbDaoByKey(String key) {
if (mongoDataSources.containsKey(key)) {
MongoDataSource dataSource = mongoDataSources.get(key);
List seeds = dataSource.getSeeds();
MongoClientOptions options = dataSource.getMongoClientOptions();
List credentials = dataSource.getMongoCredentials();
MongoClient mongoClient = new MongoClient(seeds, credentials, options);
MongoDbFactory dbFactory = new SimpleMongoDbFactory(mongoClient, dataSource.getDefaultDatabase());
return new MongoDbTemplate(new MongoTemplate(dbFactory, dataSource.getMongoConverter()));
} else {
throw new DataSourceKeyNotExistException();
}
}
/**
* 同{@link #getMongoDbDaoByKeyAndDataBase(String, String)}，默认数据源不变，切换数据库（文档）
*
* @param database
* @return
*/
private MongoAccessor getMongoDbDaoByDataBase(String database) {
MongoDataSource dataSource = this.defaultMongoDataSource;
List seeds = dataSource.getSeeds();
MongoClientOptions options = dataSource.getMongoClientOptions();
List credentials = dataSource.getMongoCredentials();
MongoClient mongoClient = new MongoClient(seeds, credentials, options);
MongoDbFactory factory = new SimpleMongoDbFactory(mongoClient, database);
return new MongoDbTemplate(new MongoTemplate(factory, dataSource.getMongoConverter()));
}
}