edu.brown.pools.TypedPoolableObjectFactory.java Source code

Introduction

Here is the source code for edu.brown.pools.TypedPoolableObjectFactory.java

Source

/***************************************************************************
 *  Copyright (C) 2012 by H-Store Project                                  *
 *  Brown University                                                       *
 *  Massachusetts Institute of Technology                                  *
 *  Yale University                                                        *
 *                                                                         *
 *  http://hstore.cs.brown.edu/                                            *
 *                                                                         *
 *  Permission is hereby granted, free of charge, to any person obtaining  *
 *  a copy of this software and associated documentation files (the        *
 *  "Software"), to deal in the Software without restriction, including    *
 *  without limitation the rights to use, copy, modify, merge, publish,    *
 *  distribute, sublicense, and/or sell copies of the Software, and to     *
 *  permit persons to whom the Software is furnished to do so, subject to  *
 *  the following conditions:                                              *
 *                                                                         *
 *  The above copyright notice and this permission notice shall be         *
 *  included in all copies or substantial portions of the Software.        *
 *                                                                         *
 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,        *
 *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF     *
 *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. *
 *  IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR      *
 *  OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,  *
 *  ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR  *
 *  OTHER DEALINGS IN THE SOFTWARE.                                        *
 ***************************************************************************/
package edu.brown.pools;

import java.lang.reflect.Constructor;
import java.util.concurrent.atomic.AtomicInteger;

import org.apache.commons.pool.BasePoolableObjectFactory;

import edu.brown.utils.ClassUtil;

/**
 * @author pavlo
 * @param <T>
 */
public abstract class TypedPoolableObjectFactory<T extends Poolable> extends BasePoolableObjectFactory {

    private boolean enable_counting;
    private final AtomicInteger created = new AtomicInteger(0);
    private final AtomicInteger passivated = new AtomicInteger(0);
    private final AtomicInteger destroyed = new AtomicInteger(0);

    public TypedPoolableObjectFactory(boolean enable_counting) {
        this.enable_counting = enable_counting;
    }

    public abstract T makeObjectImpl() throws Exception;

    public void setEnableCounting(boolean enable) {
        this.enable_counting = enable;
    }

    public boolean isCountingEnabled() {
        return (this.enable_counting);
    }

    @Override
    public final T makeObject() throws Exception {
        T obj = this.makeObjectImpl();
        if (this.enable_counting)
            this.created.getAndIncrement();
        return obj;
    }

    @Override
    public final void passivateObject(Object obj) throws Exception {
        Poolable poolable = (Poolable) obj;
        // There might be a race condition here... but maybe not...
        // if (poolable.isInitialized()) {
        poolable.finish();
        // }
        if (this.enable_counting)
            this.passivated.getAndIncrement();
    }

    @Override
    public final void destroyObject(Object obj) throws Exception {
        if (this.enable_counting)
            this.destroyed.getAndIncrement();
    }

    public int getCreatedCount() {
        return (this.created.get());
    }

    public int getPassivatedCount() {
        return (this.passivated.get());
    }

    public int getDestroyedCount() {
        return (this.destroyed.get());
    }

    @Override
    public String toString() {
        String ret = super.toString();
        if (this.enable_counting) {
            ret += String.format("[created:%d / passivated:%s / destroyed: %d]", this.created.get(),
                    this.passivated.get(), this.destroyed.get());
        }
        return (ret);
    }

    /**
     * @param <X>
     * @param clazz
     * @param enable_tracking
     * @param args
     * @return
     */
    public static <X extends Poolable> TypedPoolableObjectFactory<X> makeFactory(final Class<X> clazz,
            final boolean enable_tracking, final Object... args) {
        Class<?> argsClazz[] = new Class[args.length];
        for (int i = 0; i < args.length; i++) {
            assert (args[i] != null) : "[" + i + "]";
            argsClazz[i] = args[i].getClass();
        } // FOR
        final Constructor<X> constructor = ClassUtil.getConstructor(clazz, argsClazz);
        return new TypedPoolableObjectFactory<X>(enable_tracking) {
            @Override
            public X makeObjectImpl() throws Exception {
                return (constructor.newInstance(args));
            }
        };
    }
}