Zustand持久化存储中的onRehydrateStorage执行时机问题解析

问题背景

在使用Zustand状态管理库时，开发者经常会配合persist中间件来实现状态的持久化存储。然而，在实际开发中，很多开发者会遇到一个常见问题：onRehydrateStorage回调函数的执行时机早于存储数据的实际加载，这会导致状态被意外重置。

现象分析

通过观察控制台日志，我们可以清晰地看到执行顺序：

1. 首先触发onRehydrateStorage回调
2. 然后才执行storage.getItem获取存储数据
3. 最后将新状态写入存储

这种执行顺序会导致一个严重问题：在存储数据加载完成前，回调函数就已经执行，此时状态会被重置为默认值，覆盖掉之前存储的状态。

技术原理

Zustand的持久化机制工作流程如下：

1. 初始化时创建带有默认值的store
2. 触发onRehydrateStorage回调
3. 从存储中加载已保存的状态
4. 合并加载的状态到当前store
5. 标记hydration完成

问题出在第2步和第3步的顺序上。回调过早执行意味着开发者无法在回调中访问到已加载的持久化数据。

解决方案

正确的做法是在onRehydrateStorage返回的函数中执行状态更新操作，而不是直接在回调中执行。这是因为：

• 直接回调执行时数据尚未加载
• 返回的函数会在hydration完成后执行

修改后的代码示例如下：

onRehydrateStorage: (state) => {
  console.log("hydration starts");
  return (state, error) => {
    if (error) {
      console.error("hydration error", error);
    } else {
      console.log("hydration finished", state);
      state.setHasHydrated(true);
    }
  };
}

最佳实践

1. 将状态更新逻辑放在返回的函数中
2. 始终处理可能的hydration错误
3. 使用_hasHydrated标志来跟踪hydration状态
4. 在UI中可以根据hydration状态显示加载指示器

总结

理解Zustand持久化机制的执行顺序对于正确使用onRehydrateStorage回调至关重要。通过将状态更新逻辑放在返回的函数中，可以确保在数据加载完成后再执行相关操作，避免状态被意外重置的问题。这种模式也体现了React生态中常见的"渲染后执行"的设计思想。