-
Notifications
You must be signed in to change notification settings - Fork 417
/
utils.ts
73 lines (66 loc) · 1.61 KB
/
utils.ts
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
import IORedis from 'ioredis';
export const errorObject: { [index: string]: any } = { value: null };
export function tryCatch(fn: (...args: any) => any, ctx: any, args: any[]) {
try {
return fn.apply(ctx, args);
} catch (e) {
errorObject.value = e;
return errorObject;
}
}
export function isEmpty(obj: object) {
for (const key in obj) {
if (obj.hasOwnProperty(key)) {
return false;
}
}
return true;
}
export function array2obj(arr: string[]) {
const obj: { [index: string]: string } = {};
for (let i = 0; i < arr.length; i += 2) {
obj[arr[i]] = arr[i + 1];
}
return obj;
}
export function delay(ms: number): Promise<void> {
return new Promise(resolve => {
setTimeout(() => resolve(), ms);
});
}
export function isRedisInstance(obj: any): boolean {
if (!obj) {
return false;
}
const redisApi = ['connect', 'disconnect', 'duplicate'];
return redisApi.every(name => typeof obj[name] === 'function');
}
export async function removeAllQueueData(
client: IORedis.Redis,
queueName: string,
prefix = 'bull',
) {
const pattern = `${prefix}:${queueName}:*`;
return new Promise((resolve, reject) => {
const stream = client.scanStream({
match: pattern,
});
stream.on('data', (keys: string[]) => {
if (keys.length) {
const pipeline = client.pipeline();
keys.forEach(key => {
pipeline.del(key);
});
pipeline.exec().catch(error => {
reject(error);
});
}
});
stream.on('end', () => {
resolve();
});
stream.on('error', error => {
reject(error);
});
});
}