@antv/scale/src/utils/utc-interval.ts

Toolkit for mapping abstract data into visual representation.

import {
  Interval,
  createInterval,
  IntervalMap,
  DURATION_SECOND,
  DURATION_MINUTE,
  DURATION_HOUR,
  DURATION_DAY,
  DURATION_WEEK,
  DURATION_MONTH,
  DURATION_YEAR,
} from './time-interval';

export const utcMillisecond: Interval = createInterval(
  1,
  (date) => date,
  (date, step = 1) => {
    date.setTime(+date + step);
  },
  (date) => date.getTime()
);

export const utcSecond: Interval = createInterval(
  DURATION_SECOND,
  (date) => {
    date.setUTCMilliseconds(0);
  },
  (date, step = 1) => {
    date.setTime(+date + DURATION_SECOND * step);
  },
  (date) => date.getUTCSeconds()
);

export const utcMinute: Interval = createInterval(
  DURATION_MINUTE,
  (date) => {
    date.setUTCSeconds(0, 0);
  },
  (date, step = 1) => {
    date.setTime(+date + DURATION_MINUTE * step);
  },
  (date) => date.getUTCMinutes()
);

export const utcHour: Interval = createInterval(
  DURATION_HOUR,
  (date) => {
    date.setUTCMinutes(0, 0, 0);
  },
  (date, step = 1) => {
    date.setTime(+date + DURATION_HOUR * step);
  },
  (date) => date.getUTCHours()
);

export const utcDay: Interval = createInterval(
  DURATION_DAY,
  (date) => {
    date.setUTCHours(0, 0, 0, 0);
  },
  (date, step = 1) => {
    date.setTime(+date + DURATION_DAY * step);
  },
  (date) => date.getUTCDate() - 1
);

export const utcMonth: Interval = createInterval(
  DURATION_MONTH,
  (date) => {
    date.setUTCDate(1);
    date.setUTCHours(0, 0, 0, 0);
  },
  (date, step = 1) => {
    const month = date.getUTCMonth();
    date.setUTCMonth(month + step);
  },
  (date) => date.getUTCMonth()
);

export const utcWeek: Interval = createInterval(
  DURATION_WEEK,
  (date) => {
    date.setUTCDate(date.getUTCDate() - ((date.getUTCDay() + 7) % 7));
    date.setUTCHours(0, 0, 0, 0);
  },
  (date, step = 1) => {
    date.setTime(+date + DURATION_WEEK * step);
  },
  (date) => {
    const start = utcMonth.floor(date);
    const end = new Date(+date);
    return Math.floor((+end - +start) / DURATION_WEEK);
  }
);

export const utcYear: Interval = createInterval(
  DURATION_YEAR,
  (date) => {
    date.setUTCMonth(0, 1);
    date.setUTCHours(0, 0, 0, 0);
  },
  (date, step = 1) => {
    const year = date.getUTCFullYear();
    date.setUTCFullYear(year + step);
  },
  (date) => date.getUTCFullYear()
);

export const utcIntervalMap: IntervalMap = {
  millisecond: utcMillisecond,
  second: utcSecond,
  minute: utcMinute,
  hour: utcHour,
  day: utcDay,
  week: utcWeek,
  month: utcMonth,
  year: utcYear,
};