paricafe/src/queue/index.ts

53 lines
1 KiB
TypeScript
Raw Normal View History

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