dorkbox.objectPool.NonBlockingSoftPool Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2016 dorkbox, llc
 *
 * 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
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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 dorkbox.objectPool;

import java.lang.ref.SoftReference;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;

/**
 * @author dorkbox, llc
 */
class NonBlockingSoftPool extends ObjectPool {
    private final Queue> queue;
    private final PoolableObject poolableObject;

    NonBlockingSoftPool(final PoolableObject poolableObject) {
        this(poolableObject, new ConcurrentLinkedQueue>());
    }

    NonBlockingSoftPool(final PoolableObject poolableObject, final Queue> queue) {
        this.poolableObject = poolableObject;
        this.queue = queue;
    }

    /**
     * Takes an object from the pool.
     */
    public
    T take() {
        T obj;
        SoftReference ref;
        while((ref = queue.poll()) != null) {
            if((obj = ref.get()) != null) {
                poolableObject.onTake(obj);
                return obj;
            }
        }

        final T take = poolableObject.create();
        poolableObject.onTake(take);
        return take;
    }

    /**
     * Takes an object from the pool.
     */
    public
    T takeInterruptibly() throws InterruptedException {
        return take();
    }

    /**
     * Return object to the pool.
     */
    public
    void put(T object) {
        poolableObject.onReturn(object);
        this.queue.offer(new SoftReference(object));
    }

    /**
     * @return a new object instance created by the pool.
     */
    public
    T newInstance() {
        return poolableObject.create();
    }
}