RapidOCR多平台支持中的动态库加载问题解析-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_01397/article/details/147517002

RapidOCR多平台支持中的动态库加载问题解析

RapidOCR A cross platform OCR Library based on PaddleOCR & OnnxRuntime & OpenVINO. 项目地址: https://gitcode.com/gh_mirrors/ra/RapidOCR

背景介绍

RapidOCR是一个基于ONNX和NCNN的高效OCR识别库，它通过Java Native Interface(JNI)调用本地库来实现高性能的文字识别功能。在实际应用中，开发者常常需要让应用程序支持多个操作系统平台，这就涉及到跨平台动态库的加载问题。

问题现象

当开发者尝试在单个"fat jar"包中集成所有平台的动态库支持时，会遇到动态库加载错误。具体表现为：

在Maven依赖中同时引入了多个平台的库文件
运行时系统会尝试加载错误的平台动态库
使用Thread.currentThread().getContextClassLoader().getResourceAsStream方法加载资源时，可能会获取到不匹配当前运行环境的库文件

技术原理分析

这个问题源于Java类加载机制和JNI库加载的特殊性：

类加载器资源查找：默认情况下，Java会从类路径中查找资源，当多个平台的库文件都存在时，无法保证加载到正确的平台特定库。
JNI库加载机制：JNI要求加载与当前运行环境完全匹配的本地库，包括操作系统类型、CPU架构等。加载不匹配的库会导致运行时错误。
Fat Jar打包问题：将所有平台的库打包到一个jar中时，虽然资源都存在，但缺乏有效的平台检测和选择机制。

解决方案探讨

针对这个问题，可以采取以下几种解决方案：

1. 改进的库加载器实现

修改OnnxMacArm64LibraryLoader等平台特定的加载器，使其能够精确识别并加载正确的库文件：

public class OnnxMacArm64LibraryLoader implements LibraryLoader {
    @Override
    public void loadLibrary() throws IOException {
        String root = this.getClass().getProtectionDomain().getCodeSource().getLocation().getPath();
        JarFileUtil.copyFileFromJar("lib/libRapidOcr.dylib", "/onnx", true, false, root);
    }
}

2. 增强的Jar文件工具

改进JarFileUtil工具类，使其能够根据提供的jar路径精确提取特定文件：

try {
    InputStream is = null;
    if (root == null) {
        is = Thread.currentThread().getContextClassLoader()
                .getResourceAsStream(filePath);
    } else {
        JarFile jarFile = new JarFile(root);
        JarEntry entry = jarFile.getJarEntry(filePath);
        if (entry != null) {
            is = jarFile.getInputStream(entry);
        }
    }
    // ...其余处理逻辑
}

3. 运行时平台检测

在加载库之前，先检测当前运行环境，然后动态选择正确的库文件路径：

String osName = System.getProperty("os.name").toLowerCase();
String osArch = System.getProperty("os.arch").toLowerCase();

if (osName.contains("mac")) {
    if (osArch.contains("aarch64") || osArch.contains("arm64")) {
        // 加载Mac ARM64库
    } else {
        // 加载Mac x86_64库
    }
} else if (osName.contains("win")) {
    // Windows平台处理
} else if (osName.contains("linux")) {
    // Linux平台处理
}