1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
// @ts-ignore internal api
const core = Deno.core;
export function formatToCronSchedule(
value?: number | { exact: number | number[] } | {
start?: number;
end?: number;
every?: number;
},
): string {
if (value === undefined) {
return "*";
} else if (typeof value === "number") {
return value.toString();
} else {
const { exact } = value as { exact: number | number[] };
if (exact === undefined) {
const { start, end, every } = value as {
start?: number;
end?: number;
every?: number;
};
if (start !== undefined && end !== undefined && every !== undefined) {
return start + "-" + end + "/" + every;
} else if (start !== undefined && end !== undefined) {
return start + "-" + end;
} else if (start !== undefined && every !== undefined) {
return start + "/" + every;
} else if (start !== undefined) {
return start + "/1";
} else if (end === undefined && every !== undefined) {
return "*/" + every;
} else {
throw new TypeError("Invalid cron schedule");
}
} else {
if (typeof exact === "number") {
return exact.toString();
} else {
return exact.join(",");
}
}
}
}
export function parseScheduleToString(
schedule: string | Deno.CronSchedule,
): string {
if (typeof schedule === "string") {
return schedule;
} else {
const {
minute,
hour,
dayOfMonth,
month,
dayOfWeek,
} = schedule;
return formatToCronSchedule(minute) +
" " + formatToCronSchedule(hour) +
" " + formatToCronSchedule(dayOfMonth) +
" " + formatToCronSchedule(month) +
" " + formatToCronSchedule(dayOfWeek);
}
}
function cron(
name: string,
schedule: string | Deno.CronSchedule,
handlerOrOptions1:
| (() => Promise<void> | void)
| ({ backoffSchedule?: number[]; signal?: AbortSignal }),
handlerOrOptions2?:
| (() => Promise<void> | void)
| ({ backoffSchedule?: number[]; signal?: AbortSignal }),
) {
if (name === undefined) {
throw new TypeError("Deno.cron requires a unique name");
}
if (schedule === undefined) {
throw new TypeError("Deno.cron requires a valid schedule");
}
schedule = parseScheduleToString(schedule);
let handler: () => Promise<void> | void;
let options: { backoffSchedule?: number[]; signal?: AbortSignal } | undefined;
if (typeof handlerOrOptions1 === "function") {
handler = handlerOrOptions1;
if (typeof handlerOrOptions2 === "function") {
throw new TypeError("options must be an object");
}
options = handlerOrOptions2;
} else if (typeof handlerOrOptions2 === "function") {
handler = handlerOrOptions2;
options = handlerOrOptions1;
} else {
throw new TypeError("Deno.cron requires a handler");
}
const rid = core.ops.op_cron_create(
name,
schedule,
options?.backoffSchedule,
);
if (options?.signal) {
const signal = options?.signal;
signal.addEventListener(
"abort",
() => {
core.close(rid);
},
{ once: true },
);
}
return (async () => {
let success = true;
while (true) {
const r = await core.opAsync("op_cron_next", rid, success);
if (r === false) {
break;
}
try {
const result = handler();
const _res = result instanceof Promise ? (await result) : result;
success = true;
} catch (error) {
console.error(`Exception in cron handler ${name}`, error);
success = false;
}
}
})();
}
export { cron };
|
