jint JNI_OnLoad()

in source/neuropod/bindings/java/src/main/native/jclass_register.cc [69:128]
49 lines of code
3 McCabe index (conditional complexity)

jint JNI_OnLoad(JavaVM *vm, void *reserved)
{
    // Obtain the JNIEnv from the VM and confirm JNI_VERSION
    JNIEnv *env;

    if (vm->GetEnv(reinterpret_cast<void **>(&env), JNI_VERSION) != JNI_OK)
    {
        return JNI_ERR;
    }
    // Move this exception class out of try catch block to avoid unexpected error when throw a java exception and the
    // exception type is wrong
    com_uber_neuropod_NeuropodJNIException =
        static_cast<jclass>(env->NewGlobalRef(find_class(env, "com/uber/neuropod/NeuropodJNIException")));
    try
    {
        java_util_ArrayList      = static_cast<jclass>(env->NewGlobalRef(find_class(env, "java/util/ArrayList")));
        java_util_ArrayList_     = get_method_id(env, java_util_ArrayList, "<init>", "(I)V");
        java_util_ArrayList_add  = get_method_id(env, java_util_ArrayList, "add", "(Ljava/lang/Object;)Z");
        java_util_ArrayList_get  = get_method_id(env, java_util_ArrayList, "get", "(I)Ljava/lang/Object;");
        java_util_ArrayList_size = get_method_id(env, java_util_ArrayList, "size", "()I");

        java_util_HashMap  = static_cast<jclass>(env->NewGlobalRef(find_class(env, "java/util/HashMap")));
        java_util_HashMap_ = get_method_id(env, java_util_HashMap, "<init>", "()V");
        java_util_HashMap_put =
            get_method_id(env, java_util_HashMap, "put", "(Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;");

        java_util_Map_Entry          = static_cast<jclass>(env->NewGlobalRef(find_class(env, "java/util/Map$Entry")));
        java_util_Map_Entry_getKey   = get_method_id(env, java_util_Map_Entry, "getKey", "()Ljava/lang/Object;");
        java_util_Map_Entry_getValue = get_method_id(env, java_util_Map_Entry, "getValue", "()Ljava/lang/Object;");

        com_uber_neuropod_TensorSpec =
            static_cast<jclass>(env->NewGlobalRef(find_class(env, "com/uber/neuropod/TensorSpec")));
        com_uber_neuropod_TensorSpec_ =
            get_method_id(env,
                          com_uber_neuropod_TensorSpec,
                          "<init>",
                          "(Ljava/lang/String;Lcom/uber/neuropod/TensorType;Ljava/util/List;)V");

        com_uber_neuropod_Dimension =
            static_cast<jclass>(env->NewGlobalRef(find_class(env, "com/uber/neuropod/Dimension")));
        com_uber_neuropod_Dimension_value_ = get_method_id(env, com_uber_neuropod_Dimension, "<init>", "(J)V");
        com_uber_neuropod_Dimension_symbol_ =
            get_method_id(env, com_uber_neuropod_Dimension, "<init>", "(Ljava/lang/String;)V");

        com_uber_neuropod_TensorType =
            static_cast<jclass>(env->NewGlobalRef(find_class(env, "com/uber/neuropod/TensorType")));

        com_uber_neuropod_NeuropodTensor =
            static_cast<jclass>(env->NewGlobalRef(find_class(env, "com/uber/neuropod/NeuropodTensor")));
        com_uber_neuropod_NeuropodTensor_ = get_method_id(env, com_uber_neuropod_NeuropodTensor, "<init>", "(J)V");
        com_uber_neuropod_NeuropodTensor_getHandle =
            get_method_id(env, com_uber_neuropod_NeuropodTensor, "getHandle", "()J");
    }
    catch (const std::exception &e)
    {
        throw_java_exception(env, e.what());
    }
    // Return the JNI Version as required by method
    return JNI_VERSION;
}