1
0
mirror of https://github.com/misskey-dev/misskey synced 2024-12-18 08:38:52 +09:00
misskey/src/queue/index.ts

90 lines
1.8 KiB
TypeScript
Raw Normal View History

2019-02-04 13:35:58 +09:00
import * as Queue from 'bee-queue';
2019-02-06 15:01:43 +09:00
import * as httpSignature from 'http-signature';
2019-02-06 15:01:43 +09:00
import config from '../config';
2018-05-07 18:20:15 +09:00
import { ILocalUser } from '../models/user';
2019-02-04 13:37:50 +09:00
import { program } from '../argv';
import handler from './processors';
import { queueLogger } from './logger';
2018-04-04 23:12:35 +09:00
const enableQueue = !program.disableQueue;
const queueAvailable = config.redis != null;
2019-02-04 13:35:58 +09:00
const queue = initializeQueue();
function initializeQueue() {
if (queueAvailable) {
return new Queue('misskey', {
redis: {
port: config.redis.port,
host: config.redis.host,
password: config.redis.pass
},
removeOnSuccess: true,
removeOnFailure: true,
getEvents: false,
sendEvents: false,
storeJobs: false
});
} else {
return null;
}
}
2019-02-04 13:35:58 +09:00
2019-02-06 15:01:43 +09: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 13:35:58 +09:00
return queue.createJob(data)
2019-02-06 15:01:43 +09:00
.retries(8)
2019-02-06 14:53:02 +09:00
.backoff('exponential', 1000)
2019-02-04 13:35:58 +09:00
.save();
} else {
return handler({ data }, () => {});
2019-02-04 13:35:58 +09:00
}
2018-04-04 23:12:35 +09:00
}
2019-02-06 15:01:43 +09:00
export function processInbox(activity: any, signature: httpSignature.IParsedSignature) {
const data = {
type: 'processInbox',
activity: activity,
signature
};
2019-02-06 15:01:43 +09:00
if (queueAvailable && !program.disableApQueue) {
return queue.createJob(data)
.retries(3)
.backoff('exponential', 500)
.save();
} else {
return handler({ data }, () => {});
}
2018-04-05 23:24:51 +09:00
}
export function createExportNotesJob(user: ILocalUser) {
if (!queueAvailable) throw 'queue unavailable';
return queue.createJob({
type: 'exportNotes',
user: user
})
.save();
}
2019-02-04 13:35:58 +09:00
export default function() {
if (queueAvailable && enableQueue) {
queue.process(128, handler);
queueLogger.succ('Processing started');
2019-02-04 13:35:58 +09:00
}
return queue;
2019-02-04 13:35:58 +09:00
}