Sharkey/src/queue/index.ts

124 lines
2.4 KiB
TypeScript
Raw Normal View History

2019-02-04 06:35:58 +02:00
import * as Queue from 'bee-queue';
2019-02-06 08:01:43 +02:00
import * as httpSignature from 'http-signature';
2019-02-06 08:01:43 +02:00
import config from '../config';
2018-05-07 12:20:15 +03:00
import { ILocalUser } from '../models/user';
2019-02-04 06:37:50 +02:00
import { program } from '../argv';
import handler from './processors';
import { queueLogger } from './logger';
2018-04-04 17:12:35 +03:00
const enableQueue = !program.disableQueue;
const queueAvailable = config.redis.isJust();
2019-02-04 06:35:58 +02:00
const queue = initializeQueue();
function initializeQueue() {
return config.redis.map(({ port, host, pass }) => {
return new Queue('misskey', {
redis: {
port: port,
host: host,
password: pass.getOrElse(null)
},
removeOnSuccess: true,
removeOnFailure: true,
getEvents: false,
sendEvents: false,
storeJobs: false
});
}).getOrElse(null);
}
2019-02-04 06:35:58 +02:00
2019-02-06 08:01:43 +02:00
export function deliver(user: ILocalUser, content: any, to: any) {
if (content == null) return;
const data = {
type: 'deliver',
user,
content,
to
};
if (queueAvailable && !program.disableApQueue) {
2019-02-04 06:35:58 +02:00
return queue.createJob(data)
2019-02-06 08:01:43 +02:00
.retries(8)
2019-02-06 07:53:02 +02:00
.backoff('exponential', 1000)
2019-02-04 06:35:58 +02:00
.save();
} else {
return handler({ data }, () => {});
2019-02-04 06:35:58 +02:00
}
2018-04-04 17:12:35 +03:00
}
2019-02-06 08:01:43 +02:00
export function processInbox(activity: any, signature: httpSignature.IParsedSignature) {
const data = {
type: 'processInbox',
activity: activity,
signature
};
2019-02-06 08:01:43 +02:00
if (queueAvailable && !program.disableApQueue) {
return queue.createJob(data)
.retries(3)
.backoff('exponential', 500)
.save();
} else {
return handler({ data }, () => {});
}
2018-04-05 17:24:51 +03:00
}
export function createExportNotesJob(user: ILocalUser) {
if (!queueAvailable) throw 'queue unavailable';
return queue.createJob({
type: 'exportNotes',
user: user
})
.save();
}
2019-02-04 06:35:58 +02:00
2019-02-06 14:21:49 +02:00
export function createExportFollowingJob(user: ILocalUser) {
if (!queueAvailable) throw 'queue unavailable';
return queue.createJob({
type: 'exportFollowing',
user: user
})
.save();
}
export function createExportMuteJob(user: ILocalUser) {
if (!queueAvailable) throw 'queue unavailable';
return queue.createJob({
type: 'exportMute',
user: user
})
.save();
}
export function createExportBlockingJob(user: ILocalUser) {
if (!queueAvailable) throw 'queue unavailable';
return queue.createJob({
type: 'exportBlocking',
user: user
})
.save();
}
2019-02-04 06:35:58 +02:00
export default function() {
if (queueAvailable && enableQueue) {
queue.process(128, handler);
queueLogger.succ('Processing started');
2019-02-04 06:35:58 +02:00
}
return queue;
2019-02-04 06:35:58 +02:00
}
2019-02-06 08:24:59 +02:00
export function destroy() {
queue.destroy().then(n => {
queueLogger.succ(`All job removed (${n} jobs)`);
});
}