Improve queue option

src/argv.ts

@@ -5,6 +5,7 @@ program
 	.version(pkg.version)
 	.option('--no-daemons', 'Disable daemon processes (for debbuging)')
 	.option('--disable-clustering', 'Disable clustering')
+	.option('--disable-ap-queue', 'Disable creating job queue related to ap')
 	.option('--disable-queue', 'Disable job queue processing')
 	.option('--only-queue', 'Pocessing job queue only')
 	.option('--quiet', 'Suppress all logs')
@@ -13,6 +14,7 @@ program
 	.option('--color', 'This option is a dummy for some external program\'s (e.g. forever) issue.')
 	.parse(process.argv);
 
+if (process.env.MK_DISABLE_AP_QUEUE) program.disableApQueue = true;
 if (process.env.MK_DISABLE_QUEUE) program.disableQueue = true;
 if (process.env.MK_ONLY_QUEUE) program.onlyQueue = true;
 

src/queue/index.ts

@@ -1,6 +1,7 @@
 import * as Queue from 'bee-queue';
-import config from '../config';
+import * as httpSignature from 'http-signature';
 
+import config from '../config';
 import { ILocalUser } from '../models/user';
 import { program } from '../argv';
 import handler from './processors';
@@ -31,10 +32,19 @@ function initializeQueue() {
 	}
 }
 
-export function createHttpJob(data: any) {
+export function deliver(user: ILocalUser, content: any, to: any) {
-	if (queueAvailable) {
+	if (content == null) return;
+
+	const data = {
+		type: 'deliver',
+		user,
+		content,
+		to
+	};
+
+	if (queueAvailable && !program.disableApQueue) {
 		return queue.createJob(data)
-			.retries(3)
+			.retries(8)
 			.backoff('exponential', 1000)
 			.save();
 	} else {
@@ -42,15 +52,21 @@ export function createHttpJob(data: any) {
 	}
 }
 
-export function deliver(user: ILocalUser, content: any, to: any) {
+export function processInbox(activity: any, signature: httpSignature.IParsedSignature) {
-	if (content == null) return;
+	const data = {
+		type: 'processInbox',
+		activity: activity,
+		signature
+	};
 
-	createHttpJob({
+	if (queueAvailable && !program.disableApQueue) {
-		type: 'deliver',
+		return queue.createJob(data)
-		user,
+			.retries(3)
-		content,
+			.backoff('exponential', 500)
-		to
+			.save();
-	});
+	} else {
+		return handler({ data }, () => {});
+	}
 }
 
 export function createExportNotesJob(user: ILocalUser) {

src/server/activitypub.ts

@@ -3,7 +3,6 @@ import * as Router from 'koa-router';
 import * as json from 'koa-json-body';
 import * as httpSignature from 'http-signature';
 
-import { createHttpJob } from '../queue';
 import { renderActivity } from '../remote/activitypub/renderer';
 import Note from '../models/note';
 import User, { isLocalUser, ILocalUser, IUser } from '../models/user';
@@ -17,6 +16,7 @@ import Followers from './activitypub/followers';
 import Following from './activitypub/following';
 import Featured from './activitypub/featured';
 import renderQuestion from '../remote/activitypub/renderer/question';
+import { processInbox } from '../queue';
 
 // Init router
 const router = new Router();
@@ -35,11 +35,7 @@ function inbox(ctx: Router.IRouterContext) {
 		return;
 	}
 
-	createHttpJob({
+	processInbox(ctx.request.body, signature);
-		type: 'processInbox',
-		activity: ctx.request.body,
-		signature
-	});
 
 	ctx.status = 202;
 }