ci4sManagement-cloud/react-ui/src/hooks/useServerTime.ts

/*
 * @Author: 赵伟
 * @Date: 2024-10-10 08:51:41
 * @Description: 服务器时间 hook
 */

import { getSeverTimeReq } from '@/services/experiment';
import { to } from '@/utils/promise';
import { useCallback, useEffect, useState } from 'react';

let globalTimeOffset: number | undefined = undefined;

export const globalGetSeverTime = async () => {
  const requestStartTime = Date.now()
  const [res] = await to(getSeverTimeReq());
  const requestEndTime = Date.now()
  const requestDuration = (requestEndTime - requestStartTime) / 2;
  if (res && res.data) {
    const serverDate = new Date(res.data);
    const timeOffset = serverDate.getTime() + requestDuration - requestEndTime;
    globalTimeOffset = timeOffset;
    return timeOffset
  }
};

export const now = () => {
  return new Date(Date.now() + (globalTimeOffset ?? 0))
}

/** 获取服务器时间 */
export function useServerTime() {
  const [timeOffset, setTimeOffset] = useState<number>(globalTimeOffset ?? 0);

  useEffect(() => {
    // 获取服务器时间
    const getSeverTime = async () => {
      const [res] = await to(globalGetSeverTime());
      if (res) {
        setTimeOffset(res)
      }
    };

    if (!globalTimeOffset) {
      getSeverTime();
    }
  }, []);

  const now = useCallback(() => {
    return new Date(Date.now() + timeOffset)
  }, [timeOffset])


  return [now, timeOffset] as const;
}