Retry HTTP requests

This commit is contained in:
Akihiko Odaki 2018-04-04 22:45:55 +09:00
parent 86b1345c17
commit d7c13b975f
24 changed files with 145 additions and 93 deletions

View File

@ -0,0 +1,42 @@
import User, { pack as packUser } from '../models/user';
import FollowingLog from '../models/following-log';
import FollowedLog from '../models/followed-log';
import event from '../publishers/stream';
import notify from '../publishers/notify';
export default async (follower, followee) => Promise.all([
// Increment following count
User.update(follower._id, {
$inc: {
followingCount: 1
}
}),
FollowingLog.insert({
createdAt: new Date(),
userId: followee._id,
count: follower.followingCount + 1
}),
// Increment followers count
User.update({ _id: followee._id }, {
$inc: {
followersCount: 1
}
}),
FollowedLog.insert({
createdAt: new Date(),
userId: follower._id,
count: followee.followersCount + 1
}),
followee.host === null && Promise.all([
// Notify
notify(followee.id, follower.id, 'follow'),
// Publish follow event
packUser(follower, followee)
.then(packed => event(followee._id, 'followed', packed))
])
]);

View File

@ -99,7 +99,7 @@ async function workerMain(opt) {
if (!opt['only-server']) { if (!opt['only-server']) {
// start processor // start processor
require('./processor').default(); require('./queue').process();
} }
// Send a 'ready' message to parent process // Send a 'ready' message to parent process

View File

@ -8,7 +8,7 @@ import User, { isLocalUser } from '../models/user';
import stream, { publishChannelStream } from '../publishers/stream'; import stream, { publishChannelStream } from '../publishers/stream';
import notify from '../publishers/notify'; import notify from '../publishers/notify';
import pushSw from '../publishers/push-sw'; import pushSw from '../publishers/push-sw';
import queue from '../queue'; import { createHttp } from '../queue';
import watch from './watch'; import watch from './watch';
export default async (user, mentions, post) => { export default async (user, mentions, post) => {
@ -84,7 +84,7 @@ export default async (user, mentions, post) => {
} }
return new Promise((resolve, reject) => { return new Promise((resolve, reject) => {
queue.create('http', { createHttp({
type: 'deliverPost', type: 'deliverPost',
fromId: user._id, fromId: user._id,
toId: following.followerId, toId: following.followerId,

View File

@ -1,7 +0,0 @@
import User from '../../models/user';
import act from '../../remote/activitypub/act';
import Resolver from '../../remote/activitypub/resolver';
export default ({ data }) => User.findOne({ _id: data.actor })
.then(actor => act(new Resolver(), actor, data.outbox))
.then(Promise.all);

View File

@ -1,18 +0,0 @@
import queue from '../queue';
import db from './db';
import http from './http';
export default () => {
queue.process('db', db);
/*
256 is the default concurrency limit of Mozilla Firefox and Google
Chromium.
a8af215e691f3a2205a3758d2d96e9d328e100ff - chromium/src.git - Git at Google
https://chromium.googlesource.com/chromium/src.git/+/a8af215e691f3a2205a3758d2d96e9d328e100ff
Network.http.max-connections - MozillaZine Knowledge Base
http://kb.mozillazine.org/Network.http.max-connections
*/
queue.process('http', 256, http);
};

View File

@ -1,10 +0,0 @@
import { createQueue } from 'kue';
import config from './config';
export default createQueue({
redis: {
port: config.redis.port,
host: config.redis.host,
auth: config.redis.pass
}
});

38
src/queue/index.ts Normal file
View File

@ -0,0 +1,38 @@
import { createQueue } from 'kue';
import config from '../config';
import db from './processors/db';
import http from './processors/http';
const queue = createQueue({
redis: {
port: config.redis.port,
host: config.redis.host,
auth: config.redis.pass
}
});
export function createHttp(data) {
return queue
.create('http', data)
.attempts(16)
.backoff({ delay: 16384, type: 'exponential' });
}
export function createDb(data) {
return queue.create('db', data);
}
export function process() {
queue.process('db', db);
/*
256 is the default concurrency limit of Mozilla Firefox and Google
Chromium.
a8af215e691f3a2205a3758d2d96e9d328e100ff - chromium/src.git - Git at Google
https://chromium.googlesource.com/chromium/src.git/+/a8af215e691f3a2205a3758d2d96e9d328e100ff
Network.http.max-connections - MozillaZine Knowledge Base
http://kb.mozillazine.org/Network.http.max-connections
*/
queue.process('http', 256, http);
}

View File

@ -1,9 +1,9 @@
import Favorite from '../../models/favorite'; import Favorite from '../../../models/favorite';
import Notification from '../../models/notification'; import Notification from '../../../models/notification';
import PollVote from '../../models/poll-vote'; import PollVote from '../../../models/poll-vote';
import PostReaction from '../../models/post-reaction'; import PostReaction from '../../../models/post-reaction';
import PostWatching from '../../models/post-watching'; import PostWatching from '../../../models/post-watching';
import Post from '../../models/post'; import Post from '../../../models/post';
export default async ({ data }) => Promise.all([ export default async ({ data }) => Promise.all([
Favorite.remove({ postId: data._id }), Favorite.remove({ postId: data._id }),

View File

@ -1,9 +1,9 @@
import Post from '../../models/post'; import Post from '../../../models/post';
import User, { IRemoteUser } from '../../models/user'; import User, { IRemoteUser } from '../../../models/user';
import context from '../../remote/activitypub/renderer/context'; import context from '../../../remote/activitypub/renderer/context';
import renderCreate from '../../remote/activitypub/renderer/create'; import renderCreate from '../../../remote/activitypub/renderer/create';
import renderNote from '../../remote/activitypub/renderer/note'; import renderNote from '../../../remote/activitypub/renderer/note';
import request from '../../remote/request'; import request from '../../../remote/request';
export default async ({ data }) => { export default async ({ data }) => {
const promisedTo = User.findOne({ _id: data.toId }) as Promise<IRemoteUser>; const promisedTo = User.findOne({ _id: data.toId }) as Promise<IRemoteUser>;

View File

@ -1,13 +1,13 @@
import User, { isLocalUser, isRemoteUser, pack as packUser } from '../../models/user'; import User, { isLocalUser, isRemoteUser, pack as packUser } from '../../../models/user';
import Following from '../../models/following'; import Following from '../../../models/following';
import FollowingLog from '../../models/following-log'; import FollowingLog from '../../../models/following-log';
import FollowedLog from '../../models/followed-log'; import FollowedLog from '../../../models/followed-log';
import event from '../../publishers/stream'; import event from '../../../publishers/stream';
import notify from '../../publishers/notify'; import notify from '../../../publishers/notify';
import context from '../../remote/activitypub/renderer/context'; import context from '../../../remote/activitypub/renderer/context';
import render from '../../remote/activitypub/renderer/follow'; import render from '../../../remote/activitypub/renderer/follow';
import request from '../../remote/request'; import request from '../../../remote/request';
import Logger from '../../utils/logger'; import Logger from '../../../utils/logger';
export default async ({ data }) => { export default async ({ data }) => {
const { followerId, followeeId } = await Following.findOne({ _id: data.following }); const { followerId, followeeId } = await Following.findOne({ _id: data.following });

View File

@ -0,0 +1,7 @@
import User from '../../../models/user';
import act from '../../../remote/activitypub/act';
import Resolver from '../../../remote/activitypub/resolver';
export default ({ data }) => User.findOne({ _id: data.actor })
.then(actor => act(new Resolver(), actor, data.outbox))
.then(Promise.all);

View File

@ -1,9 +1,9 @@
import { verifySignature } from 'http-signature'; import { verifySignature } from 'http-signature';
import parseAcct from '../../acct/parse'; import parseAcct from '../../../acct/parse';
import User, { IRemoteUser } from '../../models/user'; import User, { IRemoteUser } from '../../../models/user';
import act from '../../remote/activitypub/act'; import act from '../../../remote/activitypub/act';
import resolvePerson from '../../remote/activitypub/resolve-person'; import resolvePerson from '../../../remote/activitypub/resolve-person';
import Resolver from '../../remote/activitypub/resolver'; import Resolver from '../../../remote/activitypub/resolver';
export default async ({ data }): Promise<void> => { export default async ({ data }): Promise<void> => {
const keyIdLower = data.signature.keyId.toLowerCase(); const keyIdLower = data.signature.keyId.toLowerCase();

View File

@ -1,6 +1,6 @@
import * as request from 'request-promise-native'; import * as request from 'request-promise-native';
import User from '../../models/user'; import User from '../../../models/user';
const createPost = require('../../server/api/endpoints/posts/create'); const createPost = require('../../../server/api/endpoints/posts/create');
export default async ({ data }) => { export default async ({ data }) => {
const asyncBot = User.findOne({ _id: data.userId }); const asyncBot = User.findOne({ _id: data.userId });

View File

@ -1,13 +1,13 @@
import FollowedLog from '../../models/followed-log'; import FollowedLog from '../../../models/followed-log';
import Following from '../../models/following'; import Following from '../../../models/following';
import FollowingLog from '../../models/following-log'; import FollowingLog from '../../../models/following-log';
import User, { isLocalUser, isRemoteUser, pack as packUser } from '../../models/user'; import User, { isLocalUser, isRemoteUser, pack as packUser } from '../../../models/user';
import stream from '../../publishers/stream'; import stream from '../../../publishers/stream';
import renderFollow from '../../remote/activitypub/renderer/follow'; import renderFollow from '../../../remote/activitypub/renderer/follow';
import renderUndo from '../../remote/activitypub/renderer/undo'; import renderUndo from '../../../remote/activitypub/renderer/undo';
import context from '../../remote/activitypub/renderer/context'; import context from '../../../remote/activitypub/renderer/context';
import request from '../../remote/request'; import request from '../../../remote/request';
import Logger from '../../utils/logger'; import Logger from '../../../utils/logger';
export default async ({ data }) => { export default async ({ data }) => {
const following = await Following.findOne({ _id: data.id }); const following = await Following.findOne({ _id: data.id });

View File

@ -3,7 +3,7 @@ import parseAcct from '../../../acct/parse';
import Following, { IFollowing } from '../../../models/following'; import Following, { IFollowing } from '../../../models/following';
import User from '../../../models/user'; import User from '../../../models/user';
import config from '../../../config'; import config from '../../../config';
import queue from '../../../queue'; import { createHttp } from '../../../queue';
import context from '../renderer/context'; import context from '../renderer/context';
import renderAccept from '../renderer/accept'; import renderAccept from '../renderer/accept';
import request from '../../request'; import request from '../../request';
@ -44,7 +44,7 @@ export default async (resolver: Resolver, actor, activity, distribute) => {
followerId: actor._id, followerId: actor._id,
followeeId: followee._id followeeId: followee._id
}).then(following => new Promise((resolve, reject) => { }).then(following => new Promise((resolve, reject) => {
queue.create('http', { createHttp({
type: 'follow', type: 'follow',
following: following._id following: following._id
}).save(error => { }).save(error => {

View File

@ -1,7 +1,7 @@
import queue from '../../../../queue'; import { createHttp } from '../../../../queue';
export default ({ $id }) => new Promise((resolve, reject) => { export default ({ $id }) => new Promise((resolve, reject) => {
queue.create('http', { type: 'unfollow', id: $id }).save(error => { createHttp({ type: 'unfollow', id: $id }).save(error => {
if (error) { if (error) {
reject(error); reject(error);
} else { } else {

View File

@ -1,10 +1,10 @@
import Post from '../../../models/post'; import Post from '../../../models/post';
import queue from '../../../queue'; import { createDb } from '../../../queue';
export default async ({ $id }) => { export default async ({ $id }) => {
const promisedDeletion = Post.findOneAndDelete({ _id: $id }); const promisedDeletion = Post.findOneAndDelete({ _id: $id });
await new Promise((resolve, reject) => queue.create('db', { await new Promise((resolve, reject) => createDb({
type: 'deletePostDependents', type: 'deletePostDependents',
id: $id id: $id
}).delay(65536).save(error => error ? reject(error) : resolve())); }).delay(65536).save(error => error ? reject(error) : resolve()));

View File

@ -1,7 +1,7 @@
import { JSDOM } from 'jsdom'; import { JSDOM } from 'jsdom';
import { toUnicode } from 'punycode'; import { toUnicode } from 'punycode';
import User, { validateUsername, isValidName, isValidDescription } from '../../models/user'; import User, { validateUsername, isValidName, isValidDescription } from '../../models/user';
import queue from '../../queue'; import { createHttp } from '../../queue';
import webFinger from '../webfinger'; import webFinger from '../webfinger';
import create from './create'; import create from './create';
import Resolver from './resolver'; import Resolver from './resolver';
@ -69,7 +69,7 @@ export default async (value, verifier?: string) => {
}, },
}); });
queue.create('http', { createHttp({
type: 'performActivityPub', type: 'performActivityPub',
actor: user._id, actor: user._id,
outbox outbox

View File

@ -1,7 +1,7 @@
import * as bodyParser from 'body-parser'; import * as bodyParser from 'body-parser';
import * as express from 'express'; import * as express from 'express';
import { parseRequest } from 'http-signature'; import { parseRequest } from 'http-signature';
import queue from '../../queue'; import { createHttp } from '../../queue';
const app = express(); const app = express();
@ -22,7 +22,7 @@ app.post('/@:user/inbox', bodyParser.json({
return res.sendStatus(401); return res.sendStatus(401);
} }
queue.create('http', { createHttp({
type: 'processInbox', type: 'processInbox',
inbox: req.body, inbox: req.body,
signature, signature,

View File

@ -4,7 +4,7 @@
import $ from 'cafy'; import $ from 'cafy';
import User from '../../../../models/user'; import User from '../../../../models/user';
import Following from '../../../../models/following'; import Following from '../../../../models/following';
import queue from '../../../../queue'; import { createHttp } from '../../../../queue';
/** /**
* Follow a user * Follow a user
@ -56,7 +56,7 @@ module.exports = (params, user) => new Promise(async (res, rej) => {
followeeId: followee._id followeeId: followee._id
}); });
queue.create('http', { type: 'follow', following: _id }).save(); createHttp({ type: 'follow', following: _id }).save();
// Send response // Send response
res(); res();

View File

@ -4,7 +4,7 @@
import $ from 'cafy'; import $ from 'cafy';
import User from '../../../../models/user'; import User from '../../../../models/user';
import Following from '../../../../models/following'; import Following from '../../../../models/following';
import queue from '../../../../queue'; import { createHttp } from '../../../../queue';
/** /**
* Unfollow a user * Unfollow a user
@ -49,7 +49,7 @@ module.exports = (params, user) => new Promise(async (res, rej) => {
return rej('already not following'); return rej('already not following');
} }
queue.create('http', { createHttp({
type: 'unfollow', type: 'unfollow',
id: exist._id id: exist._id
}).save(error => { }).save(error => {

View File

@ -3,7 +3,7 @@ import * as express from 'express';
//const crypto = require('crypto'); //const crypto = require('crypto');
import User from '../../../models/user'; import User from '../../../models/user';
import config from '../../../config'; import config from '../../../config';
import queue from '../../../queue'; import { createHttp } from '../../../queue';
module.exports = async (app: express.Application) => { module.exports = async (app: express.Application) => {
if (config.github_bot == null) return; if (config.github_bot == null) return;
@ -42,7 +42,7 @@ module.exports = async (app: express.Application) => {
const commit = event.commit; const commit = event.commit;
const parent = commit.parents[0]; const parent = commit.parents[0];
queue.create('http', { createHttp({
type: 'gitHubFailureReport', type: 'gitHubFailureReport',
userId: bot._id, userId: bot._id,
parentUrl: parent.url, parentUrl: parent.url,