rudeshark.net/src/queue/index.ts

53 lines
1.0 KiB
TypeScript
Raw Normal View History

2019-02-04 05:35:58 +01:00
import * as Queue from 'bee-queue';
import config from '../config';
2018-04-04 16:12:35 +02:00
import http from './processors/http';
2018-05-07 11:20:15 +02:00
import { ILocalUser } from '../models/user';
import Logger from '../misc/logger';
2019-02-04 05:37:50 +01:00
import { program } from '../argv';
2018-04-04 16:12:35 +02:00
2019-02-04 05:37:50 +01:00
const enableQueue = config.redis != null && !program.disableQueue;
2019-02-04 05:35:58 +01:00
const queue = 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
});
2018-07-26 01:11:47 +02:00
export function createHttpJob(data: any) {
2019-02-04 05:35:58 +01:00
if (enableQueue) {
return queue.createJob(data)
.retries(4)
.backoff('exponential', 16384) // 16s
.save();
} else {
return http({ data }, () => {});
}
2018-04-04 16:12:35 +02:00
}
2018-06-18 07:28:43 +02:00
export function deliver(user: ILocalUser, content: any, to: any) {
if (content == null) return;
2018-07-26 01:11:47 +02:00
createHttpJob({
2018-04-05 16:24:51 +02:00
type: 'deliver',
user,
content,
to
2018-07-26 01:11:47 +02:00
});
2018-04-05 16:24:51 +02:00
}
export const queueLogger = new Logger('queue');
2019-02-04 05:35:58 +01:00
export default function() {
if (enableQueue) {
queue.process(128, http);
}
}