PooledRepositoryConnectionFactory.java

/*******************************************************************************
 * Copyright (c) 2021 Eclipse RDF4J contributors.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Distribution License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 *******************************************************************************/

package org.eclipse.rdf4j.spring.pool;

import java.lang.invoke.MethodHandles;

import org.apache.commons.pool2.ObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.eclipse.rdf4j.repository.RepositoryConnection;
import org.eclipse.rdf4j.spring.support.connectionfactory.RepositoryConnectionFactory;
import org.eclipse.rdf4j.spring.tx.exception.RepositoryConnectionPoolException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.DisposableBean;

/**
 * Uses the delegate factory to actually obtain connections and provides these connections, managing an internal pool.
 *
 * @author Florian Kleedorfer
 * @since 4.0.0
 */
public class PooledRepositoryConnectionFactory
		implements DisposableBean, RepositoryConnectionFactory {

	private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());

	private final ObjectPool<RepositoryConnection> pool;

	public PooledRepositoryConnectionFactory(
			RepositoryConnectionFactory delegateFactory,
			GenericObjectPoolConfig<RepositoryConnection> config) {

		PooledConnectionObjectFactory factory = new PooledConnectionObjectFactory(delegateFactory);
		if (config == null) {
			this.pool = new GenericObjectPool<>(factory);
		} else {
			this.pool = new GenericObjectPool<>(factory, config);
		}
		factory.setPool(pool);
	}

	public PooledRepositoryConnectionFactory(RepositoryConnectionFactory delegateFactory) {
		this(delegateFactory, null);
	}

	@Override
	public void destroy() throws Exception {
		logger.info("shutting down RepositoryConnection pool...");
		pool.close();
		logger.info("\tdone");
	}

	@Override
	public RepositoryConnection getConnection() {
		try {
			return pool.borrowObject();
		} catch (Exception e) {
			throw new RepositoryConnectionPoolException(
					"Cannot obtain RepositoryConnection from pool", e);
		}
	}
}