com.epam.deltix.qsrv.hf.codec.cg.ObjectManager Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021 EPAM Systems, Inc
 *
 * See the NOTICE file distributed with this work for additional information
 * regarding copyright ownership. 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 com.epam.deltix.qsrv.hf.codec.cg;

import com.epam.deltix.util.collections.ElementsEnumeration;
import com.epam.deltix.util.collections.generated.ObjectArrayList;
import com.epam.deltix.util.collections.generated.ObjectToObjectHashMap;
import com.epam.deltix.util.memory.MemoryDataInput;

/**
 * Objects Collection Pool, not Thread-Safe
 */
public final class ObjectManager {

    private final CharSequencePool charSequencePool;
    private final CharSequenceListPool charSequenceListPool;
    private final ObjectToObjectHashMap collections;
    private final ElementsEnumeration ce;

    private final ObjectToObjectHashMap objects;
    private final ElementsEnumeration oe;

    private boolean isClean;

    public ObjectManager() {
        this(5);
    }

    public ObjectManager(int capacity) {
        collections = new ObjectToObjectHashMap<>(capacity);
        ce = collections.elements();

        objects = new ObjectToObjectHashMap<>(capacity);
        oe = objects.elements();

        charSequencePool = new CharSequencePool();
        charSequenceListPool = new CharSequenceListPool();
        isClean = true;
    }

    public Object               useObject(final Class clazz) {
        isClean = false;

        String name = clazz.getName();

        ObjectPool pool = objects.get(name, null);
        if (pool == null)
            objects.put(name, pool = new ObjectPool() {
                @Override
                public Object newItem() {
                    try {
                        return clazz.newInstance();
                    } catch (IllegalAccessException | InstantiationException  e) {
                        throw new RuntimeException(e);
                    }
                }
            });

        return pool.borrow();
    }

    public Object               use(Class clazz) {
        return  this.use(clazz,5);
    }

    public Object               use(Class clazz, int capacity) {
        isClean = false;
        ManagerObjectPool pool = getManagerObjectPool(clazz.getName());
        return  pool.use(clazz, capacity);
    }

    private ManagerObjectPool   getManagerObjectPool(String className) {
        ManagerObjectPool pool = collections.get(className, null);
        if (pool == null)
            collections.put(className, pool = new ManagerObjectPool());

        return pool;
    }

    public void                 clean() {
        charSequencePool.reset();
        charSequenceListPool.reset();

        if (!isClean) {
            ce.reset();
            while (ce.hasMoreElements())
                ce.nextElement().clean();

            oe.reset();
            while (oe.hasMoreElements())
                oe.nextElement().reset();

            isClean = true;
        }
    }

    public ObjectArrayList useCharSequenceList() {
        return charSequenceListPool.borrow();
    }

    public CharSequence readCharSequence(MemoryDataInput input) {
        return charSequencePool.readCharSequence(input);
    }
}