com.kurento.kmf.thrift.pool.AbstractPool.java Source code

Introduction

Here is the source code for com.kurento.kmf.thrift.pool.AbstractPool.java

Source

/*
 * (C) Copyright 2013 Kurento (http://kurento.org/)
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public License
 * (LGPL) version 2.1 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 */
package com.kurento.kmf.thrift.pool;

import java.util.NoSuchElementException;

import org.apache.commons.pool2.BasePooledObjectFactory;
import org.apache.commons.pool2.ObjectPool;
import org.apache.commons.pool2.PoolUtils;
import org.apache.commons.pool2.impl.GenericObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.springframework.beans.factory.annotation.Autowired;

import com.kurento.kmf.common.exception.KurentoException;
import com.kurento.kmf.thrift.ThriftInterfaceConfiguration;

public abstract class AbstractPool<T> implements Pool<T> {

    @Autowired
    private ThriftInterfaceConfiguration apiConfig;

    private ObjectPool<T> pool;

    /**
     * Default constructor, to be used in spring environments
     */
    public AbstractPool() {
    }

    /**
     * Constructor for non-spring environments.
     * 
     * @param apiConfig
     *            configuration object
     */
    public AbstractPool(ThriftInterfaceConfiguration apiConfig) {
        this.apiConfig = apiConfig;
    }

    protected void init(BasePooledObjectFactory<T> factory) {
        GenericObjectPoolConfig config = new GenericObjectPoolConfig();
        config.setMinIdle(apiConfig.getClientPoolSize());
        config.setTestOnBorrow(true);
        config.setTestOnReturn(true);
        this.pool = PoolUtils.erodingPool(new GenericObjectPool<>(factory, config));
    }

    @Override
    public T acquire() throws PoolLimitException, KurentoException, ClientPoolException {
        try {
            return this.pool.borrowObject();
        } catch (NoSuchElementException e) {
            throw new PoolLimitException("Max number of pooled client instances reached");
        } catch (IllegalStateException e) {
            throw new ClientPoolException("Trying to acquire an object from a closed pool", e);
        } catch (Exception e) {
            if (e instanceof KurentoException) {
                throw (KurentoException) e;
            }

            throw new ClientPoolException("Object creation failed", e);
        }
    }

    @Override
    public void release(T obj) throws ClientPoolException {
        try {
            this.pool.returnObject(obj);
        } catch (Exception e) {
            throw new ClientPoolException("Object could not be realeased", e);
        }
    }
}