ai.vespa.onnxruntime.OnnxBundleActivator Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of container-onnxruntime Show documentation
There is a newer version: 8.441.21
Show newest version
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

package ai.vespa.onnxruntime;

import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;

import java.util.logging.Logger;

/**
 * @author arnej
 * Loads native libraries when the bundle is activated.
 * Use system properties to ensure onnxruntime won't try
 * to load them itself.
 **/
public class OnnxBundleActivator implements BundleActivator {

    private static final String ONNX_PREFIX = "onnxruntime.native.";
    private static final String PATH_SUFFIX = "path";
    private static final String SKIP_SUFFIX = ".skip";
    private static final String SKIP_VALUE = "true";
    private static final String[] LIBRARY_NAMES = { "onnxruntime", "onnxruntime4j_jni" };
    private static final Logger log = Logger.getLogger(OnnxBundleActivator.class.getName());

    @Override
    public void start(BundleContext ctx) {
        String skipAll = OnnxBundleActivator.class.getSimpleName() + SKIP_SUFFIX;
        if (SKIP_VALUE.equals(System.getProperty(skipAll))) {
            log.info("skip loading of native libraries");
            return;
        }
        System.setProperty(ONNX_PREFIX + PATH_SUFFIX, "/opt/vespa-deps/lib64");
        for (String libName : LIBRARY_NAMES) {
            String skipProp = ONNX_PREFIX + libName + SKIP_SUFFIX;
            if (SKIP_VALUE.equals(System.getProperty(skipProp))) {
                log.fine("already loaded native library "+libName+", skipping");
            } else {
                log.fine("loading native library: "+libName);
                try {
                    System.loadLibrary(libName);
                    // this property also signals onnxruntime to skip loading:
                    System.setProperty(skipProp, SKIP_VALUE);
                    log.fine("loaded native library OK: "+libName);
                } catch (Exception|UnsatisfiedLinkError e) {
                    log.info("Could not load native library '"+libName+"' because: "+e.getMessage());
                }
            }
        }
    }

    @Override
    public void stop(BundleContext ctx) {
        // not sure how to test that loading and unloading multiple times actually works,
        // but this should in theory do the necessary thing.
        for (String libName : LIBRARY_NAMES) {
            String skipProp = ONNX_PREFIX + libName + SKIP_SUFFIX;
            if (SKIP_VALUE.equals(System.getProperty(skipProp))) {
                log.fine("will unload native library: "+libName);
            }
            System.clearProperty(skipProp);
        }
    }
}