org.seedstack.mqtt.internal.MqttPoolDefinition.java Source code

Introduction

Here is the source code for org.seedstack.mqtt.internal.MqttPoolDefinition.java

Source

/**
 * Copyright (c) 2013-2016, The SeedStack authors <http://seedstack.org>
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 */
/**
 * 
 */
package org.seedstack.mqtt.internal;

import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

import org.apache.commons.configuration.Configuration;
import org.eclipse.paho.client.mqttv3.MqttMessage;
import org.seedstack.mqtt.MqttRejectedExecutionHandler;

/**
 * {@link ThreadPoolExecutor} definition. This {@link ThreadPoolExecutor} is
 * used when a {@link MqttMessage} is arrived.
 * 
 * @author thierry.bouvet@mpsa.com
 *
 */
class MqttPoolDefinition {
    private static final int DEFAULT_KEEP_ALIVE = 60;
    private static final int DEFAULT_QUEUE_SIZE = 500;
    private static final int DEFAULT_MAX_SIZE = 1;
    private static final int DEFAULT_CORE_SIZE = 1;
    private static final String POOL_ENABLED = "enabled";
    private static final String POOL_KEEP_ALIVE = "keep-alive";
    private static final String POOL_QUEUE_SIZE = "queue";
    private static final String POOL_MAX_SIZE = "max";
    private static final String POOL_CORE_SIZE = "core";
    private String rejectHandlerName;
    private Class<? extends MqttRejectedExecutionHandler> rejectHandlerClass;
    private Boolean available = Boolean.TRUE;
    private ThreadPoolExecutor threadPoolExecutor;

    public MqttPoolDefinition(Configuration configuration) {
        this.available = configuration.getBoolean(POOL_ENABLED, Boolean.TRUE);
        if (this.available) {
            int coreSize = configuration.getInt(POOL_CORE_SIZE, DEFAULT_CORE_SIZE);
            int maxSize = configuration.getInt(POOL_MAX_SIZE, DEFAULT_MAX_SIZE);
            int queueSize = configuration.getInt(POOL_QUEUE_SIZE, DEFAULT_QUEUE_SIZE);
            int keepAlive = configuration.getInt(POOL_KEEP_ALIVE, DEFAULT_KEEP_ALIVE);
            this.threadPoolExecutor = new ThreadPoolExecutor(coreSize, maxSize, keepAlive, TimeUnit.SECONDS,
                    new ArrayBlockingQueue<Runnable>(queueSize));
        }
    }

    public String getRejectHandlerName() {
        return rejectHandlerName;
    }

    public Class<? extends MqttRejectedExecutionHandler> getRejectHandlerClass() {
        return rejectHandlerClass;
    }

    public void setRejectHandler(Class<? extends MqttRejectedExecutionHandler> rejectHandlerClass, String name) {
        this.rejectHandlerClass = rejectHandlerClass;
        this.rejectHandlerName = name;
    }

    public ThreadPoolExecutor getThreadPoolExecutor() {
        return threadPoolExecutor;
    }

}