forked from mirror/misskey
58 lines
1.7 KiB
TypeScript
58 lines
1.7 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and other misskey contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import type * as Bull from 'bullmq';
|
|
import type { RedisOptions } from "ioredis";
|
|
import type { RedisOptionsSource } from '@/config.js';
|
|
|
|
export const QUEUE = {
|
|
DELIVER: 'deliver',
|
|
INBOX: 'inbox',
|
|
SYSTEM: 'system',
|
|
ENDED_POLL_NOTIFICATION: 'endedPollNotification',
|
|
DB: 'db',
|
|
RELATIONSHIP: 'relationship',
|
|
OBJECT_STORAGE: 'objectStorage',
|
|
WEBHOOK_DELIVER: 'webhookDeliver',
|
|
};
|
|
|
|
export function baseQueueOptions(config: RedisOptions & RedisOptionsSource, queueOptions: Partial<Bull.QueueOptions>, queueName: typeof QUEUE[keyof typeof QUEUE]): Bull.QueueOptions {
|
|
return {
|
|
...queueOptions,
|
|
connection: {
|
|
...config,
|
|
maxRetriesPerRequest: null,
|
|
keyPrefix: undefined,
|
|
reconnectOnError: (err: Error) => {
|
|
if ( err.message.includes('READONLY')
|
|
|| err.message.includes('ETIMEDOUT')
|
|
|| err.message.includes('Command timed out')
|
|
) return 2;
|
|
return 1;
|
|
},
|
|
},
|
|
prefix: config.prefix ? `${config.prefix}:queue:${queueName}` : `queue:${queueName}`,
|
|
};
|
|
}
|
|
|
|
export function baseWorkerOptions(config: RedisOptions & RedisOptionsSource, workerOptions: Partial<Bull.WorkerOptions>, queueName: typeof QUEUE[keyof typeof QUEUE]): Bull.WorkerOptions {
|
|
return {
|
|
...workerOptions,
|
|
connection: {
|
|
...config,
|
|
maxRetriesPerRequest: null,
|
|
keyPrefix: undefined,
|
|
reconnectOnError: (err: Error) => {
|
|
if ( err.message.includes('READONLY')
|
|
|| err.message.includes('ETIMEDOUT')
|
|
|| err.message.includes('Command timed out')
|
|
) return 2;
|
|
return 1;
|
|
},
|
|
},
|
|
prefix: config.prefix ? `${config.prefix}:queue:${queueName}` : `queue:${queueName}`,
|
|
};
|
|
}
|