Function get_module

datafusion/ffi/src/tests/utils.rs:74–107

()

}

pub fn get_module() -> Result<ForeignLibraryModule> {
    let expected_version = crate::version();

    let crate_root = Path::new(env!("CARGO_MANIFEST_DIR"));
    let target_dir = crate_root
        .parent()
        .expect("Failed to find crate parent")
        .parent()
        .expect("Failed to find workspace root")
        .join("target");

    let library_dir = compute_library_dir(target_dir.as_path());
    let lib_path = find_cdylib(&library_dir.join("deps"))?;

    // Load the library using libloading
    let lib = unsafe {
        libloading::Library::new(&lib_path)
            .map_err(|e| DataFusionError::External(Box::new(e)))?
    };

    let get_module: libloading::Symbol<extern "C" fn() -> ForeignLibraryModule> = unsafe {
        lib.get(b"datafusion_ffi_get_module")
            .map_err(|e| DataFusionError::External(Box::new(e)))?
    };

    let module = get_module();

    assert_eq!((module.version)(), expected_version);

    // Leak the library to keep it loaded for the duration of the test
    std::mem::forget(lib);

    Ok(module)
}