org.frameworkset.tran.BaseMongoDBConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bboss-datatran-mongodb Show documentation
bboss etl,datastream,elasticsearch client with restful and java api without elasticsearch jar dependended.
There is a newer version: 7.3.0
Show newest version
package org.frameworkset.tran;
/**
 * Copyright 2008 biaoping.yin
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * http://www.apache.org/licenses/LICENSE-2.0
 * 

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import com.fasterxml.jackson.annotation.JsonIgnore;
import com.frameworkset.util.SimpleStringUtil;
import org.frameworkset.nosql.mongodb.ClientMongoCredential;
import org.frameworkset.nosql.mongodb.CustomSettingBuilder;
import org.frameworkset.tran.config.ImportBuilder;
import org.frameworkset.tran.context.ImportContext;
import org.frameworkset.tran.plugin.BaseConfig;

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

/**
 * 
Description: 
 * 
 * Copyright (c) 2018
 * @Date 2019/9/20 11:46
 * @author biaoping.yin
 * @version 1.0
 */
public class BaseMongoDBConfig extends BaseConfig  {
	private String name;
	private String serverAddresses;
	private String option;
	private String writeConcern;
	private String readPreference;


	private List credentials;
	private int connectionsPerHost = 50;

	private int maxWaitTime = 120000;
	private int socketTimeout = 0;
	private int connectTimeout = 15000;
	private String connectString;



	private Boolean socketKeepAlive = false;

	private String mode;
	private String dbCollection;
	private String db;

    @JsonIgnore
    private CustomSettingBuilder customSettingBuilder;


	public String getName() {
		return name;
	}

	public BaseMongoDBConfig setName(String name) {
		this.name = name;
		return this;
	}

	public String getServerAddresses() {
		return serverAddresses;
	}

	public BaseMongoDBConfig setServerAddresses(String serverAddresses) {
		this.serverAddresses = serverAddresses;
		return this;
	}

	public String getOption() {
		return option;
	}

	public BaseMongoDBConfig setOption(String option) {
		this.option = option;
		return this;
	}

	public String getWriteConcern() {
		return writeConcern;
	}

	public BaseMongoDBConfig setWriteConcern(String writeConcern) {
		this.writeConcern = writeConcern;
		return this;
	}

	public String getReadPreference() {
		return readPreference;
	}

	public BaseMongoDBConfig setReadPreference(String readPreference) {
		this.readPreference = readPreference;
		return this;
	}



	public int getConnectionsPerHost() {
		return connectionsPerHost;
	}

	public BaseMongoDBConfig setConnectionsPerHost(int connectionsPerHost) {
		this.connectionsPerHost = connectionsPerHost;
		return this;
	}

	public int getMaxWaitTime() {
		return maxWaitTime;
	}

	public BaseMongoDBConfig setMaxWaitTime(int maxWaitTime) {
		this.maxWaitTime = maxWaitTime;
		return this;
	}

	public int getSocketTimeout() {
		return socketTimeout;
	}

	public BaseMongoDBConfig setSocketTimeout(int socketTimeout) {
		this.socketTimeout = socketTimeout;
		return this;
	}

	public int getConnectTimeout() {
		return connectTimeout;
	}

	public BaseMongoDBConfig setConnectTimeout(int connectTimeout) {
		this.connectTimeout = connectTimeout;
		return this;
	}


	public Boolean getSocketKeepAlive() {
		return socketKeepAlive;
	}

	public BaseMongoDBConfig setSocketKeepAlive(Boolean socketKeepAlive) {
		this.socketKeepAlive = socketKeepAlive;
		return this;
	}

	public String getMode() {
		return mode;
	}

	public BaseMongoDBConfig setMode(String mode) {
		this.mode = mode;
		return this;
	}



	public String getDBCollection() {
		return dbCollection;
	}

	public String getDB() {
		return db;
	}

	public BaseMongoDBConfig setDbCollection(String dbCollection) {
		this.dbCollection = dbCollection;
		return this;
	}

	public BaseMongoDBConfig setDb(String db) {
		this.db = db;
		return this;
	}
	public List getCredentials() {
		return credentials;
	}

	public BaseMongoDBConfig setCredentials(List credentials) {
		this.credentials = credentials;
		return this;
	}

	private String mechanism;
	private String userName;
	@JsonIgnore
	private String password;


	private String authDb;

	public String getAuthDb() {
		return authDb;
	}

	public BaseMongoDBConfig setAuthDb(String authDb) {
		this.authDb = authDb;
		return this;
	}
	public void build(ImportContext importContext, ImportBuilder importBuilder) {

		if(SimpleStringUtil.isNotEmpty(userName) && SimpleStringUtil.isNotEmpty(password)) {
			if (credentials == null) {
				credentials = new ArrayList();
				ClientMongoCredential clientMongoCredential = new ClientMongoCredential();
				clientMongoCredential.setDatabase(authDb == null?db:authDb);
				clientMongoCredential.setMechanism(mechanism);
				clientMongoCredential.setUserName(userName);
				clientMongoCredential.setPassword(password);
				credentials.add(clientMongoCredential);
			}
		}
	}


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

	public BaseMongoDBConfig setUserName(String userName) {
		this.userName = userName;
		return this;
	}

	/**
	 * MONGODB-AWS
	 * GSSAPI
	 * PLAIN
	 * MONGODB_X509
	 * SCRAM-SHA-1
	 * SCRAM-SHA-256
	 */
	public BaseMongoDBConfig setMechanism(String mechanism) {
		this.mechanism = mechanism;
		return this;
	}

	/**
	 * see https://www.mongodb.com/docs/manual/reference/connection-string/#std-label-mongodb-uri
	 * @return
	 */
	public String getConnectString() {
		return connectString;
	}

	public BaseMongoDBConfig setConnectString(String connectString) {
		this.connectString = connectString;
		return this;
	}

    public CustomSettingBuilder getCustomSettingBuilder() {
        return customSettingBuilder;
    }

    public BaseMongoDBConfig setCustomSettingBuilder(CustomSettingBuilder customSettingBuilder) {
        this.customSettingBuilder = customSettingBuilder;
        return this;
    }
}