在 Zustand 中实现 computed 的方式

注：本文结合本人真实项目实践经验，经过 AI 润色。

引言

在状态管理领域，计算属性（computed properties）是一个极其重要的概念。MobX 和 Pinia 等库都内置了计算属性功能，允许开发者声明式地定义派生状态。虽然 Zustand 本身没有直接提供 computed API，但这并不意味着我们无法实现类似的功能。
本文将介绍三种在 Zustand 中实现计算属性的优雅方式，包含官方推荐等方案。
方案一：derive-zustand

https://github.com/zustandjs/derive-zustand 是 Zustand 官方维护的派生状态库，它提供了一种响应式的方式来处理计算逻辑。
核心优势

• 响应式更新：自动追踪依赖，当依赖状态变化时自动更新
  
• 类型安全：完美支持 TypeScript 类型推断
  
• 性能优化：避免不必要的重新计算
  

基本用法

1. import { create, useStore } from 'zustand';
2. import { derive } from 'derive-zustand';
4. // 基础 store
5. const useCountStore = create<{ count: number; inc: () => void }>((set) => ({
6.   count: 0,
7.   inc: () => set((state) => ({ count: state.count + 1 })),
8. }));
10. // 派生 store
11. const doubleCountStore = derive<number>((get) => get(useCountStore).count * 2);
13. // 自定义 hook
14. const useDoubleCountStore = () => useStore(doubleCountStore);
16. // 组件中使用
17. const Counter = () => {
18.   const { count, inc } = useCountStore();
19.   const doubleCount = useDoubleCountStore();
20.   return (
21.    
22.       count: {count}
23.       doubleCount: {doubleCount}
24.       <button type="button" onClick={inc}>
25.         +1
26.       </button>
27.    
28.   );
29. };

复制代码

适用场景

• 需要多个组件共享的派生状态
  
• 复杂的计算逻辑需要复用
  
• 希望保持响应式更新的特性
  

方案二：手动维护计算属性

对于简单的计算需求，可以直接在 store 中声明派生状态，并在相关操作后手动更新。
实现模式

1. import { create } from 'zustand';
3. type CartItem = { id: string; price: number; quantity: number };
5. type CartState = {
6.   items: CartItem[];
7.   total: number; // ← 计算属性
8.   addItem: (item: Omit<CartItem, 'id'>) => void;
9.   updateTotal: () => void;
10. };
12. const useCartStore = create<CartState>((set, get) => ({
13.   items: [],
14.   total: 0,
16.   addItem: (item) => {
17.     set((state) => ({
18.       items: [...state.items, { ...item, id: crypto.randomUUID() }],
19.     }));
20.     get().updateTotal(); // 添加商品后更新总价
21.   },
23.   updateTotal: () => {
24.     const newTotal = get().items.reduce(
25.       (sum, item) => sum + item.price * item.quantity,
26.       0
27.     );
28.     set({ total: newTotal });
29.   },
30. }));

复制代码

优缺点分析

优点：

• 不需要额外依赖
  
• 逻辑集中，便于维护
  
• 更新时机明确可控
  

缺点：

• 需要手动触发更新
  
• 可能遗漏更新点
  
• 不适合复杂依赖关系
  

最佳实践

• 为计算属性添加专门的更新方法
  
• 在文档中明确标注哪些操作会影响计算属性
  
• 考虑使用 immer 简化不可变更新逻辑
  

方案三：在组件内派生状态

对于简单的、仅限单个组件使用的派生状态，可以直接在组件内部计算。
实现示例

1. const UserProfile = () => {
2.   const firstName = useUserStore(（s） => s.firstName);
3.   const lastName = useUserStore(（s） => s.lastName);
5.   const fullName = `${firstName} ${lastName}`;
7.   return (
8.     {fullName}
9.   );
10. };

复制代码

适用条件

• 派生状态只在一个组件中使用
  
• 计算逻辑非常简单
  
• 不需要响应式更新（或可以接受组件重新渲染）
  

性能考虑

当派生计算较复杂时，可以使用 useMemo 优化：

1. const expensiveValue = useMemo(() => {
2.   return computeExpensiveValue(a, b);
3. }, [a, b]);

复制代码

方案对比与选择指南

方案适用场景复杂度性能维护性derive-zustand多组件共享的复杂派生状态中优优Store 内维护简单的全局计算属性低良中组件内计算单一组件使用的简单派生最低视情况视情况选择建议：

• 优先考虑 derive-zustand，特别是需要响应式更新时
  
• 对于简单场景，Store 内维护更轻量
  
• 组件内计算适合临时性、局部性的简单逻辑
  

高级技巧：组合使用多种方案

在实际项目中，你可以灵活组合这些方案。例如：

1. // 使用 derive-zustand 创建基础派生状态
2. const filteredTodosStore = derive<Todo[]>(get => {
3.   const { todos, filter } = get(useTodoStore);
4.   return todos.filter(todo =>
5.     filter === 'all' ||
6.     (filter === 'completed' && todo.completed) ||
7.     (filter === 'active' && !todo.completed)
8.   );
9. });
11. // 在组件内进一步派生
12. const TodoStats = () => {
13.   const filteredTodos = useStore(filteredTodosStore);
14.   
15.   // 组件特有的派生状态
16.   const completionPercentage = useMemo(() => {
17.     if (filteredTodos.length === 0) return 0;
18.     const completed = filteredTodos.filter(t => t.completed).length;
19.     return Math.round((completed / filteredTodos.length) * 100);
20.   }, [filteredTodos]);
22.   return 完成度: {completionPercentage}%;
23. };

复制代码

总结与最佳实践

在 Zustand 中实现计算属性有多种方式，没有绝对的"最佳"方案，关键是根据具体场景选择最合适的：

• 保持简单：不要过度设计，简单的组件内计算可能就足够了
  
• 关注性能：对于昂贵的计算，使用 memoization 技术
  
• 类型安全：充分利用 TypeScript 确保类型正确
  
• 文档说明：明确标注哪些是计算属性及其依赖关系
  
• 测试覆盖：为重要的计算逻辑添加单元测试
  

Zustand 的灵活性允许你根据项目需求选择最适合的计算属性实现方式，这种设计哲学正是它受到开发者喜爱的原因之一。

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！