1. 动态库热加载到底解决什么问题
1.1 从“改代码必须重启”说起
做后端服务或者推理系统的人,基本都经历过这种场景:线上正在跑着一个关键服务,业务方说“模型效果不行,赶紧换一版”,你看了看手里的代码,发现推理逻辑是直接编译进主程序里的,换模型得重新编译整个可执行文件,然后重启进程。重启意味着连接断开、任务中断、缓存清空,如果是高并发场景,光回温就得跑好几分钟。动态库热加载就是专门解决这类问题的:它允许你在进程不退出、服务不中断的情况下,把旧的实现换掉,加载新的实现。
我最早接触这个概念是因为做插件系统。当时接了一个项目,要求业务方在不停机的情况下,解决客户的定制规则,厂商把规则编译成动态库,放到指定目录,服务发现目录里有新文件就自动加载切换。后来做 AI 推理服务,我发现这个思路照样适用——例如把 onnxruntime 推理引擎封装成动态库,模型文件更新时直接热替换整个推理模块,完全不需要重启服务。
动态库热加载这个能力,说复杂也复杂,说简单也简单。核心离不开操作系统提供的那几个 API:Linux 上的 dlopen / dlsym / dlclose,Windows 上的 LoadLibrary / GetProcAddress / FreeLibrary。但真正落地到生产环境时,有一堆细节会跳出来教做人:符号可见性怎么控制、全局状态怎么清理、新旧版本之间怎么平滑切换、运行时崩溃怎么排查,这些东西普通文档里不会写全。
1.2 热加载适合哪些场景,又有什么边界
按我的经验,动态库热加载适合这几类场景。
第一类是插件系统。主程序定义好接口,业务方按接口实现动态库,运行时加载。IDE 的语法插件、消息队列的 connector、监控系统的采集器,都是这个套路。好处是主程序不用跟着每个插件一起发布,插件的升级也独立。
第二类是 AI 推理服务的模型热更新。模型迭代频繁,每两三天就有一版新的,如果每次更新模型都要重启整个服务,用户端的抖动是实打实的。把 onnxruntime 或 TensorRT 这类推理引擎封装成动态库,模型文件跟着动态库走,或者由动态库内部管理模型加载,升级时把新动态库放进去,服务自己完成切换,整个流程就顺了。
第三类是快速修复线上问题。有些线上 Bug 可能只是某个函数逻辑错了,如果整个服务重启代价太大,可以把这个模块隔离成动态库,修复后直接热替换。
但热加载不是万能的,有几个边界必须明确。最典型的就是:如果动态库内部持有全局状态,比如全局单例、静态变量、全局缓存,那么卸载旧库时这些状态并不一定会被安全清理。dlclose 只是把动态库的代码段和数据段从进程地址空间标记为可释放,如果还有其他引用,或者析构逻辑没写对,轻则内存泄漏,重则下次加载同名库时踩到残留状态。另外,如果你加载的动态库和主程序共享了同一套 C++ 运行时,两边对 new / delete 的堆管理规则不一样,跨边界传对象指针就可能直接崩溃。
所以在设计阶段就要想清楚:热加载的粒度是什么,哪些状态可以跟着旧库一起扔掉,哪些状态必须由主程序保管。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 动态库加载的底层原理,弄懂这几个概念就够了
2.1 静态链接和动态链接的本质差异
把动态库的加载原理讲清楚之前,得先分清静态链接和动态链接的差别。静态链接发生在编译阶段:链接器把静态库里的目标代码直接拷贝进可执行文件,程序运行后不再依赖外部 .a 文件。动态链接则发生在运行阶段:可执行文件里只记录“我需要哪个动态库、哪个符号”,真正把代码映射进内存,是操作系统在程序启动时(或运行时主动触发时)干的活。
动态链接的好处显而易见:多个进程可以共享同一份动态库的物理内存页面,节省内存;库的升级不用重新编译主程序。其实还有一个容易被忽略的好处,就是它天然支持“运行时才知道要加载什么”的场景——你写程序的时候根本不知道将来会有什么插件,但只要定义了接口约定,后续的插件都可以在运行时以动态库形式塞进来。
下面这张表能看清两者差异:
| 对比维度 | 静态链接 | 动态链接 |
|---|---|---|
| 链接时机 | 编译期 | 程序启动时 / 运行时 |
| 外部依赖 | 不依赖 .a 文件 | 依赖 .so / .dll |
| 升级方式 | 需重新编译整个程序 | 替换动态库即可 |
| 内存占用 | 每个进程各自拷贝一份 | 多进程可共享 |
| 热加载 | 不支持 | 支持 |
2.2 dlopen 和 LoadLibrary 底层做了什么
Linux 的 dlopen 我随手用过很多次,它所做的核心事情大概有三步:第一,打开指定路径的动态库文件,在进程的地址空间中分配一块区域,把 .so 的代码段和数据段映射进去;第二,解析这个动态库的依赖项,如果它还依赖其他动态库,则递归加载它们;第三,执行动态库里的初始化代码,对应 C++ 里的全局对象构造函数,以及 __attribute__((constructor)) 修饰的函数。
Windows 的 LoadLibrary 流程类似,但它会额外处理 DLL 的搜索路径规则,如果用相对路径不写全,系统会按“应用程序目录 -> 系统目录 -> PATH 环境变量目录”的顺序搜索。这里有个坑:你在本地开发时 DLL 放旁边能直接加载,部署到服务器上换个目录结构就加载失败了,大概率就是搜索路径的问题。最好的习惯是 LoadLibrary 时用绝对路径,别指望系统的搜索顺序。
2.3 符号解析的流程和返回值的正确用法
dlsym 和 GetProcAddress 的作用是:拿到动态库在内存中的基地址,然后去它的符号表里查名字,返回对应符号的地址。注意这个“地址”其实是一个 void* 指针,你需要手动把它转换成真正的函数指针类型,才能安全调用。
这里有一个非常经典的坑:C 和 C++ 的符号名规则不一样。C 函数编译后在符号表里就叫 infer,而 C++ 因为有函数重载和命名空间,会对函数名做 name mangling,编出来的符号可能长这样:_ZN15MyInferenceEngin 6inferEPKvPv。所以跨语言调用时,要么在 C++ 侧用 extern "C" 包裹导出函数,要么在动态库内部用 -Wl,--no-undefined 检查所有符号都解析到,免得运行时报找不到符号。
3. 手把手实现一个动态库热加载器
3.1 先定义好接口协议:热加载的基石
做热加载,第一步永远是定义接口。接口就是主程序和动态库之间的“合同”,定义清楚了,动态库内部怎么改都无所谓,只要导出的函数签名不变,主程序就能正常调用。
我一般会在主程序侧放一个公共头文件,比如 plugin.h,里面只放纯 C 接口。为什么用纯 C?因为 C 的 ABI 在主流编译器之间是稳定的,而 C++ 的 ABI 在不同编译器版本、不同标准库实现之间可能会变。如果主程序用 GCC 8 编译,动态库用 GCC 12 编译,两边都在接口处直接传 C++ 对象,很可能会因为 std::string 内部布局不同而崩掉。用纯 C 接口加一个 void* 上下文指针,能最大程度降低耦合。
c复制// plugin.h
#ifdef __cplusplus
extern "C" {
#endif
typedef struct PluginInstance PluginInstance;
// 创建插件实例,返回不透明句柄
PluginInstance* plugin_create(const char* config_path);
// 执行核心逻辑
int plugin_process(PluginInstance* inst, const char* input, char* output, int output_size);
// 释放插件实例
void plugin_destroy(PluginInstance* inst);
// 返回插件版本号,方便主程序判断是否升级
const char* plugin_version(void);
#ifdef __cplusplus
}
#endif
接口设计要尽量“薄”,不要暴露动态库内部的复杂对象。void* 句柄是最常见的做法:主程序只负责拿句柄、传参、调用、释放,至于句柄背后是什么,主程序完全不用关心。这也能避免内存释放权混乱的问题——谁创建,谁释放,中间传递的指针都当作不透明数据。
3.2 加载器核心:加载、切换、卸载
加载器负责管理动态库的生命周期。最核心的函数就是加载、卸载,以及业务层需要的时候执行切换。这里有一个非常关键的实践要点:动态库文件名最好带上版本号,比如 plugin_v1.so、plugin_v2.so,不要一直用同一个文件名覆盖。
原因很简单:如果你直接用 libplugin.so,旧库还没卸载就拷入新库文件,Linux 下 cp 默认是直接覆盖文件内容,但已经被映射进内存的文件页还是旧内容,而新打开文件描述符的进程看到的是新内容,两个“同名”的动态库同时在系统里存在,符号解析非常容易错乱。用带版本号的文件名,可以让新旧两个动态库的实体文件在磁盘上共存,加载器先加载新库、完成切换,再卸载旧库,整个过程干净利落。
下面是我常用的一个加载器骨架,剥离了业务逻辑,结构比较通用:
c复制#include <dlfcn.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
typedef struct {
void* handle;
const char* (*plugin_version)(void);
void* (*plugin_create)(const char*);
int (*plugin_process)(void*, const char*, char*, int);
void (*plugin_destroy)(void*);
} DynamicPlugin;
int load_plugin(DynamicPlugin* plugin, const char* path) {
memset(plugin, 0, sizeof(*plugin));
plugin->handle = dlopen(path, RTLD_NOW | RTLD_LOCAL);
if (!plugin->handle) {
fprintf(stderr, "dlopen failed: %s\n", dlerror());
return -1;
}
// 这里注意:dlsym 返回 void*,强制转换成函数指针是 POSIX 允许的,
// 但按 C 标准严格来说是未定义行为,实践中主流的 x86 平台都正常工作。
*(void**)(&plugin->plugin_version) = dlsym(plugin->handle, "plugin_version");
*(void**)(&plugin->plugin_create) = dlsym(plugin->handle, "plugin_create");
*(void**)(&plugin->plugin_process) = dlsym(plugin->handle, "plugin_process");
*(void**)(&plugin->plugin_destroy) = dlsym(plugin->handle, "plugin_destroy");
if (!plugin->plugin_version || !plugin->plugin_create ||
!plugin->plugin_process || !plugin->plugin_destroy) {
fprintf(stderr, "dlsym failed: %s\n", dlerror());
dlclose(plugin->handle);
plugin->handle = NULL;
return -2;
}
return 0;
}
void unload_plugin(DynamicPlugin* plugin) {
if (plugin->handle) {
dlclose(plugin->handle);
plugin->handle = NULL;
}
}
dlopen 的 flag 值得多说两句。RTLD_NOW 表示立即解析所有未定义符号,如果解析失败立刻返回 NULL 并报错;RTLD_LAZY 表示延迟解析,只在符号被真正调用时才去解析。调试期建议用 RTLD_NOW,避免遇到“加载成功但调用时突然崩溃”的幽灵问题。RTLD_LOCAL 表示动态库的符号对后续 dlopen 的其他库不可见,这是默认行为,可以防止不同库之间的符号互相污染。
3.3 热切换流程:先新后旧,保证服务不中断
热切换的正确顺序很关键。我踩过坑之后,总结出一个原则:先加载新,再切换到新,最后卸载旧。
如果把顺序反过来,先卸载旧库,再去加载新库,就会有一个窗口期,主程序手里的函数指针指向一个已经被释放的地址,这时候只要有请求进来,直接段错误。生产环境不能冒这个险。
推荐流程如下:
- 加载新动态库
plugin_v2.so,得到新的函数指针。 - 调用新库的
plugin_create,创建新实例,完成初始化。 - 切换全局请求处理器,让新请求走进新实例的逻辑。
- 将旧实例标记为“排空状态”,等待正在处理中的请求结束。
- 所有旧请求处理完毕后,调用旧库的
plugin_destroy释放实例,再dlclose卸载旧库。
这里最常见的现实问题是:怎么判断旧请求都处理完了?如果你的服务是单线程的,那不存在这个问题,切完直接卸载即可。如果是多线程模型,建议给每个插件实例加一个引用计数,每进来一个请求就 ref++,请求结束就 ref--,等引用计数归零再卸载。简单粗暴、但非常有效。
3.4 编译动态库时的符号可见性设置
动态库编译参数里有一个 -fvisibility=hidden,建议做插件类库时默认加上。它的作用是:不显式标记的符号默认不对外导出,只有用 __attribute__((visibility("default"))) 标记的符号才会出现在动态符号表里。
这么做的好处有两个:第一,减少动态库的符号表体积,因为内部实现细节不会再暴露出去;第二,避免符号冲突,如果库里有一堆内部函数不小心和主程序或其他库重名,加载时可能被 GLOB 模式捋走,出现“调用了别人家的函数”这种诡异 Bug。
在头文件里,我一般会定义一个简单的导出宏:
c复制#if defined(_WIN32)
#define PLUGIN_API __declspec(dllexport)
#else
#define PLUGIN_API __attribute__((visibility("default")))
#endif
然后在每个需要导出的函数前面加上 PLUGIN_API。有了它,-fvisibility=hidden 就完全可控了,哪些函数暴露给外部,一清二楚。
4. 实战:当 onnxruntime 遇上热加载
4.1 为什么用 onnxruntime 做案例
最近做推理服务时我一直在用 onnxruntime,这个库几乎成了我的默认选择。它在 CPU 上的性能不错,支持量化、算子融合、多线程,还支持 GPU EP,部署起来相当省心。而且它有一个特别适合做热加载的特性:C API 非常稳定,官方对 C API 的兼容性承诺比 C++ API 更严格,配合动态库加载,非常顺滑。
既然要结合 onnxruntime 玩动态库热加载,最直观的方案就是:把 onnxruntime 推理引擎封装成一个自定义插件动态库,暴露统一的推理接口,主程序通过 dlopen 加载它。这样主程序不直接依赖 onnxruntime,它只认统一接口;onnxruntime 的升级、模型文件的更换、甚至整个推理引擎换成 TensorRT,都只是换一个动态库的事情。
用 onnxruntime 动态库还有一层现实意义:官方发布的预编译包虽然开箱即用,但如果你想定制算子、去掉不需要的优化、或者裁剪体积,就必须从源码编译。源码编译出来的动态库,配合热加载,整个过程是可控的、可复现的,而不是只停留在文档层面。
4.2 编译 onnxruntime 动态库
onnxruntime 的源码编译不算复杂,但有几个参数值得注意。下面是我的编译命令(以 Linux + CPU 版本为例):
bash复制git clone --recursive https://github.com/microsoft/onnxruntime.git
cd onnxruntime
./build.sh \
--config Release \
--build_shared_lib \
--parallel 8 \
--cmake_extra_defines \
onnxruntime_USE_CUDA=OFF \
onnxruntime_BUILD_UNIT_TESTS=OFF
关键参数是 --build_shared_lib,它控制是否生成动态库。编译完成后,产物在 build/Linux/Release/ 目录下,你会看到 libonnxruntime.so。注意它默认链接了所有依赖,实际部署时动态库文件要完整拷贝,不能只拿一个 .so 就走了。
如果只需要一小部分功能,可以进一步裁剪。比如只做 CPU 推理,不需要处理特定格式的模型,可以通过 cmake 参数关闭不需要的算子内核,缩小二进制体积。我试过一次,把不必要的 EP 和算子全部关掉后,体积能减小不少,加载速度也更快。
4.3 用 onnxruntime C API 封装一个推理插件
核心代码分为两部分。第一部分是统一的插件接口——就是我上面说的那组 C 函数;第二部分是内部实现——通过 onnxruntime C API 创建 Session,处理输入输出。
下面给一个简化但完整可跑的实现,我在注释里标了关键点:
c复制#include <onnxruntime_c_api.h>
#include "plugin.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
struct PluginInstance {
const OrtApi* api;
OrtEnv* env;
OrtSession* session;
char* model_path;
};
static const OrtApi* g_ort_api;
void* plugin_create(const char* config_path) {
// 配置文件里存模型路径,这样换模型可以不重新编译动态库
char model_path[512];
FILE* fp = fopen(config_path, "r");
if (!fp) return NULL;
if (fgets(model_path, sizeof(model_path), fp) == NULL) {
fclose(fp);
return NULL;
}
// 去掉末尾换行符
model_path[strcspn(model_path, "\n")] = 0;
fclose(fp);
PluginInstance* inst = (PluginInstance*)calloc(1, sizeof(PluginInstance));
if (!inst) return NULL;
inst->api = OrtGetApiBase()->GetApi(ORT_API_VERSION);
g_ort_api = inst->api;
inst->api->CreateEnv(ORT_LOGGING_LEVEL_WARNING, "inference_plugin", &inst->env);
inst->api->CreateSession(inst->env, model_path, NULL, &inst->session);
inst->model_path = strdup(model_path);
return inst;
}
int plugin_process(void* handle, const char* input, char* output, int output_size) {
PluginInstance* inst = (PluginInstance*)handle;
if (!inst || !inst->session) return -1;
// 实际项目中,这里的 input 是一段序列化数据,比如打平后的 float 数组,
// output 再反序列化成业务结构。这里简化为固定形状的示例。
// 核心思路是构造 OrtValue -> Run 推理 -> 取出结果。
// 省略具体张量构造细节,注意释放所有申请出来的内存。
return 0;
}
void plugin_destroy(void* handle) {
PluginInstance* inst = (PluginInstance*)handle;
if (!inst) return;
if (inst->session) inst->api->ReleaseSession(inst->session);
if (inst->env) inst->api->ReleaseEnv(inst->env);
free(inst->model_path);
free(inst);
}
const char* plugin_version(void) {
return "1.0.0";
}
onnxruntime 的 C API 设计得比较规整:所有对象都是不透明句柄,通过 API 函数创建和释放。记住一个原则:凡是通过 Create 得到的对象,都必须有对应的 Release 调用。Session、Env、OrtValue、TensorInfo,一个都不能漏。反过来说,这正好契合热加载的需求:只要在 plugin_destroy 里把 Session 和 Env 都释放干净,加载器就能安全地 dlclose。
4.4 实现推理引擎热切换
推理引擎的动态库写好之后,主程序的热切换逻辑就简单了。假设你有一份新模型,按下面步骤执行:
- 把新模型文件放到一个约定目录,同时复制一份新版本的动态库,比如
inference_v2.so。 - 加载器调用
dlopen("inference_v2.so"),拿到新的plugin_create等函数指针。 - 调用新库的
plugin_create("config_v2.txt"),让新库创建自己的推理实例。 - 主程序把请求处理器的指针从旧实例切到新实例。
- 旧实例没有请求后,调用旧库的
plugin_destroy释放 Session,再dlclose旧库。
这样一轮热切换下来,整个推理服务没有重启进程,模型的更新对用户基本无感。实测下来,切完新库后处理新请求的链路已经跑在最新模型上,旧请求还在用的旧实例也不会受影响,真正做到了新旧代码并存、平滑过渡。
这里额外分享一个我后来加的小优化:在新库加载后,可以先跑一个“预热”请求,比如用一小段样本数据推一次,确保模型加载正确、输出维度符合预期,再切换到全量流量。如果预热失败,可以立刻回滚到旧库,不会影响线上用户。
c复制// 预热检查:加载新库后先跑一个冒烟测试
int smoke_test(DynamicPlugin* plugin) {
void* inst = plugin->plugin_create("config_v2.txt");
if (!inst) return -1;
char input[] = "dummy_data";
char output[4096];
int ret = plugin->plugin_process(inst, input, output, sizeof(output));
if (ret != 0) {
plugin->plugin_destroy(inst);
return -1;
}
plugin->plugin_destroy(inst);
return 0;
}
不要小看这个预热函数。模型文件损坏、算子不支持、维度配置错误,这些问题在加载阶段往往不会立刻暴露,等你把流量切过去之后才炸,那就晚了。预热相当于一道安全栅栏,能拦住大概率的问题。
5. 热加载路上的坑,我替你踩过一遍
5.1 卸载后 old 函数指针还在用,直接段错误
这个问题几乎每个做过热加载的人都遇到过。dlclose 之后,旧动态库的代码和数据页被系统标记为可释放,如果你手里还握着之前 dlsym 得到的函数指针,调用它等于访问一块已经被释放的内存地址。
要规避这个坑,必须保证在主程序逻辑里,任何函数指针的使用都有生命周期管理。如果你不能确认某个请求是否已经结束,最简单的做法是:给实例对象加引用计数,plugin_process 入口和出口都做原子操作,等计数归零再卸载。
还有一个细节:线程本地缓存。如果某个线程缓存了旧实例的指针,切新库后它还拿着旧指针访问,也会崩溃。建议每次请求都从全局的“当前实例”位置拿一次指针,不要反复缓存。
5.2 dlclose 返回成功,但 onnxruntime 的全局资源没有释放干净
onnxruntime 内部有自己的全局状态,比如线程池、内存分配器、日志系统。某些版本里,OrtEnv 释放后,全局线程池还会存在。直接 dlclose 后,你会发现内存没有完全下降,甚至再次加载新库时,可能和旧库遗留的全局状态产生冲突。
处理方式有两种。第一种是接受这个事实,承认“onnxruntime 不能做到 100% 的干净卸载”,那就不要频繁卸载库,而是设计成长时间运行、只在版本切换时才卸载。第二种是在动态库侧维护一个生命周期更长的单例,比如一个进程只创建一次 Env,Session 可以创建多次,这样 Env 的资源不需要反复创建销毁。
我在实际项目中就遇到了这个问题,当时查了很久,最后定位到是线程池残留。后来改成 Env 常驻,Session 热切换,完美解决。这也印证了前面说的:热加载的粒度不一定是整个库,也可以细化到库内部对象的生命周期。
5.3 高频切换后内存疯狂增长,疑似内存泄漏
如果你的热切换频率很高,比如每次收到模型更新消息就加载一次新库、卸载一次旧库,内存有可能会缓慢增长。原因有几个可能:一是旧库的 plugin_destroy 没有把所有 onnxruntime 对象都释放掉,比如遗漏了某个 OrtValue;二是 dlopen 自身会缓存一些状态,反复加载同一路径的库,系统会复用同一个句柄,导致旧库的析构逻辑没有执行。
解决办法是:切换时确认资源释放完整,用 Valgrind 或 AddressSanitizer 做一轮检查;同时避免加载同路径的同名库,同一个库路径最好只加载一次,后续更新都用新路径。
5.4 加载新库时符号冲突,行为完全不可控
动态库默认是 RTLD_LOCAL,但如果主程序之前用 RTLD_GLOBAL 加载了某个库,它的符号会被放入全局符号表,之后加载的库在解析符号时可能会找到这些全局符号,而不是自己需要的版本。这种问题有时不会直接报错,而是表现为“结果不对”“跑着跑着崩了”,排查起来极其费神。
最稳妥的做法是:所有插件库统一用 RTLD_NOW | RTLD_LOCAL 加载,同时所有导出接口都带统一前缀,比如 plugin_,把符号冲突的概率降到最低。如果换用 C++ 标准库版本不同的库,还要额外小心 ABI 兼容性,必要时让主程序和插件用同一个标准库。
5.5 onnxruntime 动态库版本和模型版本不匹配
onnxruntime 对模型算子版本有兼容矩阵,老模型可能在新版本上正常,新模型可能要求最低算子版本。如果你把 onnxruntime 升级了,但旧模型是某个老格式导出的,可能加载失败或推理结果不对。
建议在切换前先让模型跑一遍全部样本或抽样子集,用置信度接近的方式确认推理结果。我的经验是:光看加载成功还不够,必须盯输出数值的变化,数值对不上立刻回滚,别硬撑。
下面整理一个排查速查表,遇到问题可以先按表查:
| 现象 | 可能原因 | 排查方向 |
|---|---|---|
| dlopen 返回 NULL | 依赖库缺失 / 权限不够 / 路径不对 | 用 ldd 查看动态库依赖,确认所有 .so 都在 |
| 加载成功但调用崩溃 | 符号解析到错误地址 / 函数指针类型不对 | 用 nm -D 查看导出符号,检查 dlsym 返回类型转换 |
| 再次加载同名库行为异常 | dlopen 返回缓存句柄 / 旧状态残留 | 换新路径加载,或检查引用计数 |
| 卸载后内存不降 | 全局资源未释放 / 线程池残留 | 用 ASAN 跑一轮,检查线程对象是否全部退出 |
| 多线程下切换崩溃 | 新库加载完成前已有请求进入 | 加引用计数 + 双缓冲切换,确保所有旧请求结束再卸载 |
| 模型推理结果不对 | 模型文件损坏 / 算子版本不兼容 | 跑样本比对,加载时打印模型输入输出信息 |
6. 一些实操心得
做动态库热加载这几年,我最大的体会是:它不是一个“用几个 API 就完事”的技术,而是一套工程体系。接口定义、加载策略、生命周期管理、优雅排空、回滚机制,每一环都要提前想清楚。代码本身反而简单,最难的是边界情况的处理。
具体到 onnxruntime 这个场景,我个人建议从“Session 热切换”开始做,而不是一上来就追求整个动态库的加载卸载。先把模型文件能做到不重启地更新,再逐步把推理引擎也做成可替换的,一步一步来,风险可控。而且这种架构设计对业务的收益是长期的:今天你是 onnxruntime,明天想换成 TensorRT 或 OpenVINO,主程序一行不用改,只要新写一个符合接口的动态库就行。
另外建议大家在一开始就引入版本号、热备目录、预热检查这三个机制。它们看着好像多写了点代码,但真到了凌晨两点被叫起来处理线上事故时,你就能感受到有多值了。
