Sharkey/src/queue/index.ts

150 lines
2.7 KiB
TypeScript
Raw Normal View History

2019-03-07 16:07:21 +02:00
import * as Queue from 'bull';
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';
2019-03-07 16:07:21 +02:00
import processDeliver from './processors/deliver';
import processInbox from './processors/process-inbox';
import processDb from './processors/db';
function initializeQueue(name: string) {
return new Queue(name, config.redis != null ? {
redis: {
port: config.redis.port,
host: config.redis.host,
password: config.redis.pass,
db: 1
}
} : null);
}
2019-02-04 06:35:58 +02:00
2019-03-07 16:07:21 +02:00
const deliverQueue = initializeQueue('deliver');
const inboxQueue = initializeQueue('inbox');
const dbQueue = initializeQueue('db');
2019-02-06 08:01:43 +02:00
export function deliver(user: ILocalUser, content: any, to: any) {
2019-03-07 16:07:21 +02:00
if (content == null) return null;
2019-02-06 08:01:43 +02:00
const data = {
user,
content,
to
};
2019-03-07 16:07:21 +02:00
return deliverQueue.add(data, {
attempts: 4,
backoff: {
type: 'exponential',
delay: 1000
},
removeOnComplete: true,
removeOnFail: true
});
2018-04-04 17:12:35 +03:00
}
2019-03-07 16:07:21 +02:00
export function inbox(activity: any, signature: httpSignature.IParsedSignature) {
2019-02-06 08:01:43 +02:00
const data = {
activity: activity,
signature
};
2019-03-07 16:07:21 +02:00
return inboxQueue.add(data, {
attempts: 4,
backoff: {
type: 'exponential',
delay: 1000
},
removeOnComplete: true,
removeOnFail: true
});
2018-04-05 17:24:51 +03:00
}
2019-02-20 18:30:21 +02:00
export function createDeleteNotesJob(user: ILocalUser) {
const data = {
type: 'deleteNotes',
user: user
};
2019-03-07 16:07:21 +02:00
return dbQueue.add(data, {
removeOnComplete: true,
removeOnFail: true
});
2019-02-20 18:30:21 +02:00
}
export function createDeleteDriveFilesJob(user: ILocalUser) {
const data = {
type: 'deleteDriveFiles',
user: user
};
2019-03-07 16:07:21 +02:00
return dbQueue.add(data, {
removeOnComplete: true,
removeOnFail: true
});
2019-02-20 18:30:21 +02:00
}
export function createExportNotesJob(user: ILocalUser) {
const data = {
type: 'exportNotes',
user: user
};
2019-03-07 16:07:21 +02:00
return dbQueue.add(data, {
removeOnComplete: true,
removeOnFail: true
});
}
2019-02-04 06:35:58 +02:00
2019-02-06 14:21:49 +02:00
export function createExportFollowingJob(user: ILocalUser) {
const data = {
2019-02-06 14:21:49 +02:00
type: 'exportFollowing',
user: user
};
2019-03-07 16:07:21 +02:00
return dbQueue.add(data, {
removeOnComplete: true,
removeOnFail: true
});
2019-02-06 14:21:49 +02:00
}
export function createExportMuteJob(user: ILocalUser) {
const data = {
2019-02-06 14:21:49 +02:00
type: 'exportMute',
user: user
};
2019-03-07 16:07:21 +02:00
return dbQueue.add(data, {
removeOnComplete: true,
removeOnFail: true
});
2019-02-06 14:21:49 +02:00
}
export function createExportBlockingJob(user: ILocalUser) {
const data = {
2019-02-06 14:21:49 +02:00
type: 'exportBlocking',
user: user
};
2019-03-07 16:07:21 +02:00
return dbQueue.add(data, {
removeOnComplete: true,
removeOnFail: true
});
2019-02-06 14:21:49 +02:00
}
2019-02-04 06:35:58 +02:00
export default function() {
2019-03-07 16:07:21 +02:00
if (!program.onlyServer) {
deliverQueue.process(processDeliver);
inboxQueue.process(processInbox);
dbQueue.process(processDb);
2019-02-04 06:35:58 +02:00
}
}
2019-02-06 08:24:59 +02:00
export function destroy() {
2019-03-07 16:07:21 +02:00
/*
2019-02-06 08:24:59 +02:00
queue.destroy().then(n => {
queueLogger.succ(`All job removed (${n} jobs)`);
2019-03-07 16:07:21 +02:00
});*/
2019-02-06 08:24:59 +02:00
}