源码分析React中useCallback和useMemo

 // callback 表示回调函数，deps是依赖数组，下同
function mountCallback(callback, deps) {
 // 调用 mountWorkInProgress方法创建新的hook，并将其放到fiber的memoizedState链表上，最后返回新的hook
  const hook = mountWorkInProgressHook();
  // 判断依赖，若不存在，则为null
  const nextDeps = deps === undefined ? null : deps;
  // 将callback和deps 存放到fiber.memoizedState中，以便后续复用
  hook.memoizedState = [callback, nextDeps];
  // 最后返回callback
  return callback;
}

function updateCallback(callback, deps) {
  // 调用updateWorkInProgressHook读取旧Fiber的对应旧hook，以及复用旧hook，创建新hook并返回
  const hook = updateWorkInProgressHook();
  // 新的依赖取值
  const nextDeps = deps === undefined ? null : deps;
  // 从新hook上取hook之前数据
  const prevState = hook.memoizedState;
  // 若新依赖不等于null
  if (nextDeps !== null) {
    const prevDeps = prevState[1];
    // 则判断新旧依赖是否发生了改变
    if (areHookInputsEqual(nextDeps, prevDeps)) {
    // 若没变化，则直接返回之前的callback
      return prevState[0];
    }
  }
  // 若新依赖不存在，或者新旧依赖不等，则再次将callback和deps依赖存入到fiber中
  hook.memoizedState = [callback, nextDeps];
  // 返回callback
  return callback;
}

function mountMemo(
  nextCreate,
  deps,
): T {
  const hook = mountWorkInProgressHook();
  const nextDeps = deps === undefined ? null : deps;
  const nextValue = nextCreate();
  hook.memoizedState = [nextValue, nextDeps];
  return nextValue;
}
function updateMemo(
  nextCreate,
  deps,
) {
  const hook = updateWorkInProgressHook();
  const nextDeps = deps === undefined ? null : deps;
  const prevState = hook.memoizedState;
  if (nextDeps !== null) {
    const prevDeps = prevState[1];
    if (areHookInputsEqual(nextDeps, prevDeps)) {
      return prevState[0];
    }
  }
  const nextValue = nextCreate();
  hook.memoizedState = [nextValue, nextDeps];
  return nextValue;
}