CouchConnector.java

/*******************************************************************************
 * Copyright (c) 2013 aegif.
 *
 * This file is part of NemakiWare.
 *
 * NemakiWare is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * NemakiWare is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with NemakiWare.
 * If not, see <http://www.gnu.org/licenses/>.
 *
 * Contributors:
 *     linzhixing(https://github.com/linzhixing) - initial API and implementation
 ******************************************************************************/
package jp.aegif.nemaki.dao.impl.couch.connector;

import org.ektorp.CouchDbConnector;
import org.ektorp.CouchDbInstance;
import org.ektorp.http.HttpClient;
import org.ektorp.http.StdHttpClient;
import org.ektorp.http.StdHttpClient.Builder;
import org.ektorp.impl.StdCouchDbInstance;
import org.springframework.beans.factory.annotation.Value;

/**
 * Connection class for CouchDB
 */

public class CouchConnector {

	/**
	 * CouchDB connector.
	 */
	private CouchDbConnector connector;

	/**
	 * Host, for instance: 127.0.0.1
	 */
	private String host;

	/**
	 * Repository id, for instance: books
	 */
	@Value("${repositoryId}")
	private String repositoryId;

	/**
	 * Port, for instance: 5984
	 */
	private int port;

	/**
	 * Max connections, for instance: 40
	 */
	private int maxConnections;

	private int connectionTimeout;

	private int socketTimeout;
	
	private boolean authEnabled;
	private String authUserName;
	private String authPassword;

	/**
	 * Initialize this class with host, maxConnections.
	 */
	public void init() {
		Builder builder = new StdHttpClient.Builder()
		.host(host)
		.port(port)
		.maxConnections(maxConnections)
		.connectionTimeout(connectionTimeout)
		.socketTimeout(socketTimeout)
		.cleanupIdleConnections(true);
		
		if(authEnabled){
			builder.username(authUserName).password(authPassword);
		}
		
		HttpClient httpClient = builder.build();
		CouchDbInstance dbInstance = new StdCouchDbInstance(httpClient);

		String repo = "";
		try {
			repo = repositoryId;
		} catch (Exception ex) {
			ex.printStackTrace();
		}
		connector = dbInstance.createConnector(repo, true);
	}

	public CouchDbConnector getConnection() {
		return connector;
	}
	
	public void setRepositoryId(String repositoryId) {
		this.repositoryId = repositoryId;
	}

	public void setHost(String host) {
		this.host = host;
	}

	public void setPort(int port) {
		this.port = port;
	}

	public void setMaxConnections(int maxConnections) {
		this.maxConnections = maxConnections;
	}

	public void setConnectionTimeout(int connectionTimeout) {
		this.connectionTimeout = connectionTimeout;
	}

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

	public void setAuthEnabled(boolean authEnabled) {
		this.authEnabled = authEnabled;
	}

	public void setAuthUserName(String authUserName) {
		this.authUserName = authUserName;
	}

	public void setAuthPassword(String authPassword) {
		this.authPassword = authPassword;
	}
	
}