一、為什麼要使用多資料來源

在一些業務複雜的系統中，業務資料儲存可能在不同的mongodb庫中，此時，可能需要同時讀取這兩個庫裡的資料，進行一些邏輯處理，此時需要讀取不同的庫，為了解決這個問題，採用多資料來源操作。

二、多資料來源配置

spring boot自動配置了多種操作mongodb的api，這裡講解的是MongoTemplate，具體程式碼如下

1.新增依賴

<!-- mongoDB -->
<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-data-mongodb</artifactId>
</dependency>
 

2.新增配置

#第一個mongodb配置
spring.data.mongodb.primary.database=barrage
#如果有多個地址 則這樣寫localhost:27017,localhost:27017  中間以逗號隔開
spring.data.mongodb.primary.hostPort=localhost:37017
spring.data.mongodb.primary.password=
spring.data.mongodb.primary.username=

#第二個mongodb配置
spring.data.mongodb.secondary.database=watch-record
spring.data.mongodb.secondary.hostPort=localhost:37017
spring.data.mongodb.secondary.password=
spring.data.mongodb.secondary.username=
 

3.重寫java配置

package com.example.multidatasource.config.mongodb;

import com.mongodb.MongoClient;
import com.mongodb.MongoCredential;
import com.mongodb.ServerAddress;
import org.springframework.data.mongodb.MongoDbFactory;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.SimpleMongoDbFactory;
import org.springframework.util.StringUtils;

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

/**
 * 根據配置檔案建立MongoDbFactory
 * @date   2017年10月14日
 *
 */
public abstract class AbstractMongoConfig {
	// Mongo DB Properties
	private String hostPort, database, username, password;
//	private String port;
	// Setter methods go here..
	public String getHostPort() {
		return hostPort;
	}

	public void setHostPort(String hostPort) {
		this.hostPort = hostPort;
	}

	public String getDatabase() {
		return database;
	}

	public void setDatabase(String database) {
		this.database = database;
	}

	public String getUsername() {
		return username;
	}

	public void setUsername(String username) {
		this.username = username;
	}

	public String getPassword() {
		return password;
	}

	public void setPassword(String password) {
		this.password = password;
	}

	public MongoDbFactory mongoDbFactory() throws Exception {
		List<ServerAddress> seeds = new ArrayList<>();
		String[] hostPorts = hostPort.split(",");
		for(int i = 0 ; i < hostPorts.length;i++){
			String[] hps = hostPorts[i].split(":");
			ServerAddress serverAddress = new ServerAddress(hps[0], Integer.valueOf(hps[1]));
			seeds.add(serverAddress);
		}
		if(StringUtils.isEmpty(username) || StringUtils.isEmpty(password)){
			return new SimpleMongoDbFactory(new MongoClient(seeds), database);
		}

		List<MongoCredential> mongoCredentialList = new ArrayList<>();
		mongoCredentialList.add(MongoCredential.createCredential(username, database, password.toCharArray()));
		return new SimpleMongoDbFactory(new MongoClient(seeds, mongoCredentialList), database);
	}

	abstract public MongoTemplate getMongoTemplate() throws Exception;
}
 

從程式碼中可以看到，從配置檔案中讀取到相應屬性進行解析之後，建立MongoDbFactory物件即可。這裡要說的是，如果mongodb沒有設定使用者名稱密碼，則直接使用new SimpleMongoDbFactory(new MongoClient(seeds), database);連線即可。這裡使用的是構造方法是：

public MongoClient(List<ServerAddress> seeds) {
         this(seeds, (new Builder()).build());
}

如果mongodb設定了使用者名稱密碼，則需要使用密碼驗證，此時使用的MongoClient構造方法是：

public MongoClient(List<ServerAddress> seeds, List<MongoCredential> credentialsList) {
	this(seeds, credentialsList, (new Builder()).build());
}

MongoClient還提供了一些其他構造方法，感興趣的可以檢視原始碼

MongoTemplate相關配置如下：

• 第一個mongodb配置

package com.example.multidatasource.config.mongodb;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.mongodb.core.MongoTemplate;

@Configuration
@ConfigurationProperties(prefix = "spring.data.mongodb.primary")
public class PrimaryMongoConfig extends AbstractMongoConfig {

     @Primary    
     @Override    
     public @Bean(name = "primaryMongoTemplate") MongoTemplate getMongoTemplate() throws Exception {        
         return new MongoTemplate(mongoDbFactory());    
     }
}

這裡使用了註解@Primary，此註解表示如果沒有指明注入哪個MongoTemplate，優先注入此MongoTemplate

• 第二個MongoTemplate配置

package com.example.multidatasource.config.mongodb;

import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.mongodb.core.MongoTemplate;

@Configuration
@ConfigurationProperties(prefix = "spring.data.mongodb.secondary")
public class SecondaryMongoConfig extends  AbstractMongoConfig{     

    @Override
    public @Bean(name = "secondaryMongoTemplate") MongoTemplate getMongoTemplate() throws Exception {
        return new MongoTemplate(mongoDbFactory());    
    }
}

4.去除spring boot mongodb自動配置

#去除mongodb自動配置
spring.autoconfigure.exclude=org.springframework.boot.autoconfigure.mongo.MongoAutoConfiguration,org.springframework.boot.autoconfigure.data.mongo.MongoDataAutoConfiguration

到此為止mongodb的多資料來源配置就算完成了

三、使用示例