com.dropchop.snakejar.embed.python.MemoryManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of snakejar Show documentation

A Java library with embedded Python (CPython) which supports native image compile and usage in Java Application Servers, targeting native Python Machine Learning frameworks and libraries.

There is a newer version: 1.35.7

Show newest version

/**
 * Copyright (c) 2017-2021 JEP AUTHORS.
 *
 * This file is licensed under the the zlib/libpng License.
 *
 * This software is provided 'as-is', without any express or implied
 * warranty. In no event will the authors be held liable for any
 * damages arising from the use of this software.
 * 
 * Permission is granted to anyone to use this software for any
 * purpose, including commercial applications, and to alter it and
 * redistribute it freely, subject to the following restrictions:
 * 
 *     1. The origin of this software must not be misrepresented; you
 *     must not claim that you wrote the original software. If you use
 *     this software in a product, an acknowledgment in the product
 *     documentation would be appreciated but is not required.
 * 
 *     2. Altered source versions must be plainly marked as such, and
 *     must not be misrepresented as being the original software.
 * 
 *     3. This notice may not be removed or altered from any source
 *     distribution.
 */
package com.dropchop.snakejar.embed.python;

import java.lang.ref.ReferenceQueue;
import java.util.Collections;
import java.util.IdentityHashMap;
import java.util.Iterator;
import java.util.Set;

import com.dropchop.snakejar.embed.JepException;

/**
 * Manages the native memory associated with PyObjects in an Interpreter.
 *
 * @see How
 *      to Handle Java Finalization's Memory-Retention Issues
 * @author Nate Jensen
 * @since 3.8
 */
public final class MemoryManager {

    private ReferenceQueue refQueue = new ReferenceQueue<>();

    private Set pointers = Collections
            .newSetFromMap(new IdentityHashMap());

    protected ReferenceQueue getReferenceQueue() throws JepException {
        cleanupWeakReferences();
        return refQueue;
    }

    protected void addReference(PyPointer pointer) {
        pointers.add(pointer);
    }

    protected void removeReference(PyPointer pyPtr) {
        pointers.remove(pyPtr);
    }

    /**
     * Cleans out all the known references to PyPointers associated with this
     * Interpreter.
     * 
     * @throws JepException
     *             if an error occurs
     */
    public void cleanupReferences() throws JepException {
        Iterator itr = pointers.iterator();
        while (itr.hasNext()) {
            PyPointer ptr = itr.next();
            /*
             * ptr.dispose() will remove from the set, so we remove it here
             * first to avoid ConcurrentModificationException
             */
            itr.remove();
            ptr.dispose();
        }
    }

    /**
     * Cleans out weak references to PyPointers associated with this
     * Interpreter. Attempts to free memory earlier than a Jep.close() if the
     * developer did not explicitly free the memory with PyObject.close().
     * 
     * @throws JepException
     *             if an error occurs
     */
    public void cleanupWeakReferences() throws JepException {
        PyPointer p = (PyPointer) refQueue.poll();
        while (p != null) {
            p.dispose();
            p = (PyPointer) refQueue.poll();
        }
    }

}