All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.elasticsearch.common.DefaultCacheRecycler Maven / Gradle / Ivy

There is a newer version: 8.14.1
Show newest version
/*
 * Licensed to ElasticSearch and Shay Banon under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. ElasticSearch licenses this
 * file to you 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 org.elasticsearch.common;

import gnu.trove.map.hash.TByteIntHashMap;
import gnu.trove.map.hash.TDoubleIntHashMap;
import gnu.trove.map.hash.TFloatIntHashMap;
import gnu.trove.map.hash.TIntIntHashMap;
import gnu.trove.map.hash.TIntObjectHashMap;
import gnu.trove.map.hash.TLongIntHashMap;
import gnu.trove.map.hash.TLongLongHashMap;
import gnu.trove.map.hash.TObjectFloatHashMap;
import gnu.trove.map.hash.TObjectIntHashMap;
import gnu.trove.map.hash.TShortIntHashMap;
import gnu.trove.set.hash.THashSet;

import java.lang.ref.SoftReference;
import java.util.Arrays;
import java.util.Queue;

import org.elasticsearch.common.trove.ExtTDoubleObjectHashMap;
import org.elasticsearch.common.trove.ExtTHashMap;
import org.elasticsearch.common.trove.ExtTLongObjectHashMap;
import org.elasticsearch.common.util.concurrent.ConcurrentCollections;

public class DefaultCacheRecycler implements Recycler {
 
    
    @Override
    public void clear() {
        hashMap.clear();
        hashSet.clear();
        doubleObjectHashMap.clear();
        longObjectHashMap.clear();
        longLongHashMap.clear();
        intIntHashMap.clear();
        floatIntHashMap.clear();
        doubleIntHashMap.clear();
        shortIntHashMap.clear();
        longIntHashMap.clear();
        objectIntHashMap.clear();
        intObjectHashMap.clear();
        objectFloatHashMap.clear();
        objectArray.clear();
        intArray.clear();
    }

    static class SoftWrapper {
        private SoftReference ref;

        public SoftWrapper() {
        }

        public void set(T ref) {
            this.ref = new SoftReference(ref);
        }

        public T get() {
            return ref == null ? null : ref.get();
        }

        public void clear() {
            ref = null;
        }
    }

    // ----- ExtTHashMap -----

    private final SoftWrapper> hashMap = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popHashMap()
     */
    @Override
    @SuppressWarnings("unchecked")
    public  ExtTHashMap popHashMap() {
        ExtTHashMap map = pop(hashMap);
        if (map == null) {
            return new ExtTHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushHashMap(org.elasticsearch.common.trove.ExtTHashMap)
     */
    @Override
    public void pushHashMap(ExtTHashMap map) {
        map.clear();
        push(hashMap, map);
    }

    // ----- THashSet -----

    private final SoftWrapper> hashSet = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popHashSet()
     */
    @Override
    @SuppressWarnings("unchecked")
    public  THashSet popHashSet() {
        THashSet set = pop(hashSet);
        if (set == null) {
            return new THashSet();
        }
        return set;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushHashSet(gnu.trove.set.hash.THashSet)
     */
    @Override
    public void pushHashSet(THashSet map) {
        map.clear();
        push(hashSet, map);
    }

    // ------ ExtTDoubleObjectHashMap -----

    private final SoftWrapper> doubleObjectHashMap = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popDoubleObjectMap()
     */
    @Override
    @SuppressWarnings("unchecked")
    public  ExtTDoubleObjectHashMap popDoubleObjectMap() {
        ExtTDoubleObjectHashMap map = pop(doubleObjectHashMap);
        if (map == null) {
            return new ExtTDoubleObjectHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushDoubleObjectMap(org.elasticsearch.common.trove.ExtTDoubleObjectHashMap)
     */
    @Override
    public void pushDoubleObjectMap(ExtTDoubleObjectHashMap map) {
        map.clear();
        push(doubleObjectHashMap, map);
    }

    // ----- ExtTLongObjectHashMap ----

    private final SoftWrapper> longObjectHashMap = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popLongObjectMap()
     */
    @Override
    @SuppressWarnings("unchecked")
    public  ExtTLongObjectHashMap popLongObjectMap() {
        ExtTLongObjectHashMap map = pop(longObjectHashMap);
        if (map == null ) {
            return new ExtTLongObjectHashMap();
        }
        return map;
    }
    
    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushLongObjectMap(org.elasticsearch.common.trove.ExtTLongObjectHashMap)
     */
    @Override
    public void pushLongObjectMap(ExtTLongObjectHashMap map) {
        map.clear();
        push(longObjectHashMap, map);
    }

    // ----- TLongLongHashMap ----

    private final SoftWrapper> longLongHashMap = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popLongLongMap()
     */
    @Override
    public TLongLongHashMap popLongLongMap() {
        TLongLongHashMap map = pop(longLongHashMap);
        if (map == null) {
            return new TLongLongHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushLongLongMap(gnu.trove.map.hash.TLongLongHashMap)
     */
    @Override
    public void pushLongLongMap(TLongLongHashMap map) {
        map.clear();
        push(longLongHashMap, map);    }

    // ----- TIntIntHashMap ----

    private final SoftWrapper> intIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popIntIntMap()
     */
    @Override
    public TIntIntHashMap popIntIntMap() {
        TIntIntHashMap map = pop(intIntHashMap);
        if (map == null) {
            return new TIntIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushIntIntMap(gnu.trove.map.hash.TIntIntHashMap)
     */
    @Override
    public void pushIntIntMap(TIntIntHashMap map) {
        map.clear();
        push(intIntHashMap, map);
    }


    // ----- TFloatIntHashMap ---

    private final SoftWrapper> floatIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popFloatIntMap()
     */
    @Override
    public TFloatIntHashMap popFloatIntMap() {
        TFloatIntHashMap map = pop(floatIntHashMap);
        if (map == null) {
            return new TFloatIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushFloatIntMap(gnu.trove.map.hash.TFloatIntHashMap)
     */
    @Override
    public void pushFloatIntMap(TFloatIntHashMap map) {
        map.clear();
        push(floatIntHashMap, map);
    }


    // ----- TDoubleIntHashMap ---

    private final SoftWrapper> doubleIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popDoubleIntMap()
     */
    @Override
    public TDoubleIntHashMap popDoubleIntMap() {
        TDoubleIntHashMap map = pop(doubleIntHashMap);
        if (map == null) {
            return new TDoubleIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushDoubleIntMap(gnu.trove.map.hash.TDoubleIntHashMap)
     */
    @Override
    public void pushDoubleIntMap(TDoubleIntHashMap map) {
        map.clear();
        push(doubleIntHashMap, map);
    }


    // ----- TByteIntHashMap ---

    private final SoftWrapper> byteIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popByteIntMap()
     */
    @Override
    public TByteIntHashMap popByteIntMap() {
        TByteIntHashMap map = pop(byteIntHashMap);
        if (map == null) {
            return new TByteIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushByteIntMap(gnu.trove.map.hash.TByteIntHashMap)
     */
    @Override
    public void pushByteIntMap(TByteIntHashMap map) {
        map.clear();
        push(byteIntHashMap, map);
    }
    
   
    // ----- TShortIntHashMap ---

    private final SoftWrapper> shortIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popShortIntMap()
     */
    @Override
    public TShortIntHashMap popShortIntMap() {
        TShortIntHashMap map = pop(shortIntHashMap);
        if (map == null) {
            return new TShortIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushShortIntMap(gnu.trove.map.hash.TShortIntHashMap)
     */
    @Override
    public void pushShortIntMap(TShortIntHashMap map) {
        map.clear();
        push(shortIntHashMap, map);
    }


    // ----- TLongIntHashMap ----

    private final SoftWrapper> longIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popLongIntMap()
     */
    @Override
    public TLongIntHashMap popLongIntMap() {
        Queue ref = longIntHashMap.get();
        if (ref == null) {
            return new TLongIntHashMap();
        }
        TLongIntHashMap map = pop(longIntHashMap);
        if (map == null) {
            return new TLongIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushLongIntMap(gnu.trove.map.hash.TLongIntHashMap)
     */
    @Override
    public void pushLongIntMap(TLongIntHashMap map) {
        map.clear();
        push(longIntHashMap, map);
    }

    // ------ TObjectIntHashMap -----

    private final SoftWrapper> objectIntHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popObjectIntMap()
     */
    @Override
    @SuppressWarnings({"unchecked"})
    public  TObjectIntHashMap popObjectIntMap() {
        TObjectIntHashMap map = pop(objectIntHashMap);
        if (map == null) {
            return new TObjectIntHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushObjectIntMap(gnu.trove.map.hash.TObjectIntHashMap)
     */
    @Override
    public  void pushObjectIntMap(TObjectIntHashMap map) {
        map.clear();
        push(objectIntHashMap, map);
    }

    // ------ TIntObjectHashMap -----

    private final SoftWrapper> intObjectHashMap = new SoftWrapper>();


    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popIntObjectMap()
     */
    @Override
    @SuppressWarnings({"unchecked"})
    public  TIntObjectHashMap popIntObjectMap() {
        TIntObjectHashMap map = pop(intObjectHashMap);
        if (map == null) {
            return new TIntObjectHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushIntObjectMap(gnu.trove.map.hash.TIntObjectHashMap)
     */
    @Override
    public  void pushIntObjectMap(TIntObjectHashMap map) {
        map.clear();
        push(intObjectHashMap, map);
    }

    // ------ TObjectFloatHashMap -----

    private final SoftWrapper> objectFloatHashMap = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popObjectFloatMap()
     */
    @Override
    @SuppressWarnings({"unchecked"})
    public  TObjectFloatHashMap popObjectFloatMap() {
        final TObjectFloatHashMap map = pop(objectFloatHashMap);
        if (map == null) {
            return new TObjectFloatHashMap();
        }
        return map;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushObjectFloatMap(gnu.trove.map.hash.TObjectFloatHashMap)
     */
    @Override
    public  void pushObjectFloatMap(TObjectFloatHashMap map) {
        map.clear();
        push(objectFloatHashMap, map);
    }

    // ----- int[] -----

    private final SoftWrapper> objectArray = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popObjectArray(int)
     */
    @Override
    public Object[] popObjectArray(int size) {
        size = size < 100 ? 100 : size;
        Queue ref = objectArray.get();
        if (ref == null) {
            return new Object[size];
        }
        Object[] objects = ref.poll();
        if (objects == null) {
            return new Object[size];
        }
        if (objects.length < size) {
            return new Object[size];
        }
        return objects;
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#pushObjectArray(java.lang.Object[])
     */
    @Override
    public void pushObjectArray(Object[] objects) {
        Arrays.fill(objects, null);
        push(objectArray, objects);
    }


    private final SoftWrapper> intArray = new SoftWrapper>();

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popIntArray(int)
     */
    @Override
    public int[] popIntArray(int size) {
        return popIntArray(size, 0);
    }

    /* (non-Javadoc)
     * @see org.elasticsearch.common.Recycler#popIntArray(int, int)
     */
    @Override
    public int[] popIntArray(int size, int sentinal) {
        size = size < 100 ? 100 : size;
        Queue ref = intArray.get();
        if (ref == null) {
            int[] ints = new int[size];
            if (sentinal != 0) {
                Arrays.fill(ints, sentinal);
            }
            return ints;
        }
        int[] ints = ref.poll();
        if (ints == null) {
            ints = new int[size];
            if (sentinal != 0) {
                Arrays.fill(ints, sentinal);
            }
            return ints;
        }
        if (ints.length < size) {
            ints = new int[size];
            if (sentinal != 0) {
                Arrays.fill(ints, sentinal);
            }
            return ints;
        }
        return ints;
    }
    
    @Override
    public void pushIntArray(int[] ints) {
        pushIntArray(ints, 0);
    }

    @Override
    public void pushIntArray(int[] ints, int sentinal) {
        Arrays.fill(ints, sentinal);
        push(intArray, ints);
    }
    
    private static final  void push(SoftWrapper> wrapper, T obj) {
        Queue ref = wrapper.get();
        if (ref == null) {
            ref = ConcurrentCollections.newQueue();
            wrapper.set(ref);
        }
        ref.add(obj);
    }

    private static final  T pop(SoftWrapper> wrapper) {
        Queue queue = wrapper.get();
        return queue == null ? null : queue.poll();
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy