57 lines
1.9 KiB
JavaScript
57 lines
1.9 KiB
JavaScript
/**
|
||
* A股交易日历：基于 chinese-days 节假日数据，严格判断某日期是否为交易日
|
||
* 交易日 = 周一至周五 且 不在法定节假日
|
||
* 调休补班日（周末变工作日）A股仍休市，故不视为交易日
|
||
*/
|
||
|
||
const CDN_BASE = 'https://cdn.jsdelivr.net/npm/chinese-days@1/dist/years';
|
||
const yearCache = new Map(); // year -> Set<dateStr> (holidays)
|
||
|
||
/**
|
||
* 加载某年的节假日数据
|
||
* @param {number} year
|
||
* @returns {Promise<Set<string>>} 节假日日期集合，格式 YYYY-MM-DD
|
||
*/
|
||
export async function loadHolidaysForYear(year) {
|
||
if (yearCache.has(year)) {
|
||
return yearCache.get(year);
|
||
}
|
||
try {
|
||
const res = await fetch(`${CDN_BASE}/${year}.json`);
|
||
if (!res.ok) throw new Error(`HTTP ${res.status}`);
|
||
const data = await res.json();
|
||
const holidays = new Set(Object.keys(data?.holidays ?? {}));
|
||
yearCache.set(year, holidays);
|
||
return holidays;
|
||
} catch (e) {
|
||
console.warn(`[tradingCalendar] 加载 ${year} 年节假日失败:`, e);
|
||
yearCache.set(year, new Set());
|
||
return yearCache.get(year);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 加载多个年份的节假日数据
|
||
* @param {number[]} years
|
||
*/
|
||
export async function loadHolidaysForYears(years) {
|
||
await Promise.all([...new Set(years)].map(loadHolidaysForYear));
|
||
}
|
||
|
||
/**
|
||
* 判断某日期是否为 A股交易日
|
||
* @param {dayjs.Dayjs} date - dayjs 对象
|
||
* @param {Map<number, Set<string>>} [cache] - 可选，已加载的年份缓存，默认使用内部 yearCache
|
||
* @returns {boolean}
|
||
*/
|
||
export function isTradingDay(date, cache = yearCache) {
|
||
const dayOfWeek = date.day(); // 0=周日, 6=周六
|
||
if (dayOfWeek === 0 || dayOfWeek === 6) return false;
|
||
|
||
const dateStr = date.format('YYYY-MM-DD');
|
||
const year = date.year();
|
||
const holidays = cache.get(year);
|
||
if (!holidays) return true; // 未加载该年数据时，仅排除周末
|
||
return !holidays.has(dateStr);
|
||
}
|