tailchat/client/shared/cache/useCache.ts

import { useQuery } from 'react-query';
import {
  fetchUserInfo,
  getUserOnlineStatus,
  UserBaseInfo,
} from '../model/user';
import { isValidStr } from '../utils/string-helper';

/**
 * 用户缓存
 */
export function useCachedUserInfo(
  userId: string | null,
  refetch = false
): UserBaseInfo | Record<string, never> {
  const { data } = useQuery(
    ['user', userId],
    () => {
      if (!isValidStr(userId)) {
        return {};
      }

      return fetchUserInfo(userId);
    },
    {
      staleTime: 2 * 60 * 60 * 1000, // 缓存2小时
      refetchOnMount: refetch ? 'always' : true,
    }
  );

  return data ?? {};
}

/**
 * 用户登录状态
 */
export function useCachedOnlineStatus(
  ids: string[],
  onOnlineStatusUpdate?: (onlineStatus: boolean[]) => void
): boolean[] {
  const { data, isSuccess } = useQuery(
    ['onlineStatus', ids.join(',')],
    () => getUserOnlineStatus(ids),
    {
      staleTime: 10 * 1000, // 缓存10s
    }
  );

  if (isSuccess && Array.isArray(data)) {
    if (typeof onOnlineStatusUpdate === 'function' && data) {
      onOnlineStatusUpdate(data);
    }
  }

  return data ?? ids.map(() => false);
}