com.reversemind.hypergate.client.ClientPool.java Source code

Introduction

Here is the source code for com.reversemind.hypergate.client.ClientPool.java

Source

/**
 * Copyright (c) 2013-2015 Eugene Kalinin
 * <p/>
 * 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
 * <p/>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p/>
 * 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.
 *
 */

package com.reversemind.hypergate.client;

import org.apache.commons.pool2.impl.GenericObjectPoolExt;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Set;

/**
 * @author Eugene Kalinin
 */
public class ClientPool extends GenericObjectPoolExt<IHyperGateClient> {

    private final static Logger LOG = LoggerFactory.getLogger(ClientPool.class);

    private static int START_POOL_SIZE = 50;
    private ClientPoolFactory clientPoolFactory;

    private static GenericObjectPoolConfig createConfig(int poolSize) {
        GenericObjectPoolConfig genericObjectPoolConfig = new GenericObjectPoolConfig();
        genericObjectPoolConfig.setJmxEnabled(true);
        genericObjectPoolConfig.setJmxNameBase("HyperGatePool");
        genericObjectPoolConfig.setJmxNamePrefix("HyperGatePoolPrefix");
        genericObjectPoolConfig.setBlockWhenExhausted(false);
        genericObjectPoolConfig.setMinIdle(0);
        genericObjectPoolConfig.setTestOnBorrow(true);
        genericObjectPoolConfig.setMaxWaitMillis(500);
        START_POOL_SIZE = poolSize;
        genericObjectPoolConfig.setMaxTotal(START_POOL_SIZE);
        genericObjectPoolConfig.setMaxIdle(START_POOL_SIZE);
        return genericObjectPoolConfig;
    }

    /**
     * Default values 10 clients
     *
     * @param clientPoolFactory
     */
    public ClientPool(ClientPoolFactory clientPoolFactory) {
        super(clientPoolFactory, createConfig(START_POOL_SIZE));
        this.clientPoolFactory = clientPoolFactory;
    }

    public ClientPool(ClientPoolFactory clientPoolFactory, int poolSize) {
        super(clientPoolFactory, createConfig(poolSize));
        this.clientPoolFactory = clientPoolFactory;
    }

    public ClientPoolFactory getClientPoolFactory() {
        return this.clientPoolFactory;
    }

    public String printPoolMetrics() {
        StringBuffer stringBuffer = new StringBuffer();
        stringBuffer.append("\n");
        stringBuffer.append(" getMaxTotal:").append(this.getMaxTotal()).append("\n");
        stringBuffer.append(" getMaxIdle:").append(this.getMaxIdle()).append("\n");
        stringBuffer.append(" getMaxWait:").append(this.getMaxWaitMillis()).append("\n");

        stringBuffer.append(" getNumActive:").append(this.getNumActive()).append("\n");
        stringBuffer.append(" getNumIdle:").append(this.getNumIdle()).append("\n");
        stringBuffer.append(" getNumTestsPerEvictionRun:").append(this.getNumTestsPerEvictionRun()).append("\n");
        return stringBuffer.toString();
    }

    /**
     * Force to close all Netty threads
     */
    public void forceClearClose() {
        LOG.info("FORCE SHUT DOWN OBJECT POOL");
        Set<IHyperGateClient> clients = this.getAll().keySet();
        if (clients.size() > 0) {
            LOG.info("FORCE TO CLOSE POOL for size:" + clients.size());
            for (IHyperGateClient client : clients) {
                if (client != null) {
                    LOG.info("SHUTTING DOWN CLIENT - " + client.getName());
                    try {
                        this.returnObject(client);
                    } catch (Exception ex) {
                        LOG.error("COULD NOT RETURN INTO POOL OBJECT:", ex);
                    }
                    client.shutdown();
                }
            }
        }
    }

}