perf(server): reduce db query

This commit is contained in:
syuilo 2022-03-21 05:21:37 +09:00
parent 78736c70f7
commit 6f2e93c6a1
6 changed files with 30 additions and 17 deletions

View file

@ -18,6 +18,7 @@ import { ILocalUser, User } from '@/models/entities/user.js';
import { In } from 'typeorm'; import { In } from 'typeorm';
import { renderLike } from '@/remote/activitypub/renderer/like.js'; import { renderLike } from '@/remote/activitypub/renderer/like.js';
import { getUserKeypair } from '@/misc/keypair-store.js'; import { getUserKeypair } from '@/misc/keypair-store.js';
import { noteCache, userCache } from './activitypub/cache.js';
// Init router // Init router
const router = new Router(); const router = new Router();
@ -65,11 +66,12 @@ router.post('/users/:user/inbox', json(), inbox);
router.get('/notes/:note', async (ctx, next) => { router.get('/notes/:note', async (ctx, next) => {
if (!isActivityPubReq(ctx)) return await next(); if (!isActivityPubReq(ctx)) return await next();
const note = await Notes.findOne({ // TODO: typeorm 3.0にしたら .then(x => x || null) は消せる
const note = await noteCache.fetch(ctx.params.note, () => Notes.findOne({
id: ctx.params.note, id: ctx.params.note,
visibility: In(['public' as const, 'home' as const]), visibility: In(['public' as const, 'home' as const]),
localOnly: false, localOnly: false,
}); }).then(x => x || null));
if (note == null) { if (note == null) {
ctx.status = 404; ctx.status = 404;
@ -148,7 +150,7 @@ router.get('/users/:user/publickey', async ctx => {
}); });
// user // user
async function userInfo(ctx: Router.RouterContext, user: User | undefined) { async function userInfo(ctx: Router.RouterContext, user: User | undefined | null) {
if (user == null) { if (user == null) {
ctx.status = 404; ctx.status = 404;
return; return;
@ -164,11 +166,12 @@ router.get('/users/:user', async (ctx, next) => {
const userId = ctx.params.user; const userId = ctx.params.user;
const user = await Users.findOne({ // TODO: typeorm 3.0にしたら .then(x => x || null) は消せる
const user = await userCache.fetch(userId, () => Users.findOne({
id: userId, id: userId,
host: null, host: null,
isSuspended: false, isSuspended: false,
}); }).then(x => x || null));
await userInfo(ctx, user); await userInfo(ctx, user);
}); });

View file

@ -0,0 +1,6 @@
import { Cache } from "@/misc/cache";
import { Note } from "@/models/entities/note";
import { User } from "@/models/entities/user";
export const userCache = new Cache<User | null>(1000 * 60 * 30);
export const noteCache = new Cache<Note | null>(1000 * 60 * 30);

View file

@ -5,15 +5,16 @@ import renderOrderedCollection from '@/remote/activitypub/renderer/ordered-colle
import { setResponseType } from '../activitypub.js'; import { setResponseType } from '../activitypub.js';
import renderNote from '@/remote/activitypub/renderer/note.js'; import renderNote from '@/remote/activitypub/renderer/note.js';
import { Users, Notes, UserNotePinings } from '@/models/index.js'; import { Users, Notes, UserNotePinings } from '@/models/index.js';
import { userCache } from './cache.js';
export default async (ctx: Router.RouterContext) => { export default async (ctx: Router.RouterContext) => {
const userId = ctx.params.user; const userId = ctx.params.user;
// Verify user // TODO: typeorm 3.0にしたら .then(x => x || null) は消せる
const user = await Users.findOne({ const user = await userCache.fetch(userId, () => Users.findOne({
id: userId, id: userId,
host: null, host: null,
}); }).then(x => x || null));
if (user == null) { if (user == null) {
ctx.status = 404; ctx.status = 404;

View file

@ -10,6 +10,7 @@ import renderFollowUser from '@/remote/activitypub/renderer/follow-user.js';
import { setResponseType } from '../activitypub.js'; import { setResponseType } from '../activitypub.js';
import { Users, Followings, UserProfiles } from '@/models/index.js'; import { Users, Followings, UserProfiles } from '@/models/index.js';
import { LessThan } from 'typeorm'; import { LessThan } from 'typeorm';
import { userCache } from './cache.js';
export default async (ctx: Router.RouterContext) => { export default async (ctx: Router.RouterContext) => {
const userId = ctx.params.user; const userId = ctx.params.user;
@ -27,11 +28,11 @@ export default async (ctx: Router.RouterContext) => {
return; return;
} }
// Verify user // TODO: typeorm 3.0にしたら .then(x => x || null) は消せる
const user = await Users.findOne({ const user = await userCache.fetch(userId, () => Users.findOne({
id: userId, id: userId,
host: null, host: null,
}); }).then(x => x || null));
if (user == null) { if (user == null) {
ctx.status = 404; ctx.status = 404;

View file

@ -11,6 +11,7 @@ import { setResponseType } from '../activitypub.js';
import { Users, Followings, UserProfiles } from '@/models/index.js'; import { Users, Followings, UserProfiles } from '@/models/index.js';
import { LessThan, FindConditions } from 'typeorm'; import { LessThan, FindConditions } from 'typeorm';
import { Following } from '@/models/entities/following.js'; import { Following } from '@/models/entities/following.js';
import { userCache } from './cache.js';
export default async (ctx: Router.RouterContext) => { export default async (ctx: Router.RouterContext) => {
const userId = ctx.params.user; const userId = ctx.params.user;
@ -28,11 +29,11 @@ export default async (ctx: Router.RouterContext) => {
return; return;
} }
// Verify user // TODO: typeorm 3.0にしたら .then(x => x || null) は消せる
const user = await Users.findOne({ const user = await userCache.fetch(userId, () => Users.findOne({
id: userId, id: userId,
host: null, host: null,
}); }).then(x => x || null));
if (user == null) { if (user == null) {
ctx.status = 404; ctx.status = 404;

View file

@ -15,6 +15,7 @@ import { Users, Notes } from '@/models/index.js';
import { makePaginationQuery } from '../api/common/make-pagination-query.js'; import { makePaginationQuery } from '../api/common/make-pagination-query.js';
import { Brackets } from 'typeorm'; import { Brackets } from 'typeorm';
import { Note } from '@/models/entities/note.js'; import { Note } from '@/models/entities/note.js';
import { userCache } from './cache.js';
export default async (ctx: Router.RouterContext) => { export default async (ctx: Router.RouterContext) => {
const userId = ctx.params.user; const userId = ctx.params.user;
@ -35,11 +36,11 @@ export default async (ctx: Router.RouterContext) => {
return; return;
} }
// Verify user // TODO: typeorm 3.0にしたら .then(x => x || null) は消せる
const user = await Users.findOne({ const user = await userCache.fetch(userId, () => Users.findOne({
id: userId, id: userId,
host: null, host: null,
}); }).then(x => x || null));
if (user == null) { if (user == null) {
ctx.status = 404; ctx.status = 404;