514a65e453
* perf(backend): avoid N+1 selects from `user` table when packing many entities * perf(backend): use `packMany` instead of mapping to `pack`
74 lines
2.6 KiB
TypeScript
74 lines
2.6 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import { Inject, Injectable } from '@nestjs/common';
|
|
import { DI } from '@/di-symbols.js';
|
|
import type { GalleryLikesRepository, GalleryPostsRepository } from '@/models/_.js';
|
|
import { awaitAll } from '@/misc/prelude/await-all.js';
|
|
import type { Packed } from '@/misc/json-schema.js';
|
|
import type { } from '@/models/Blocking.js';
|
|
import type { MiUser } from '@/models/User.js';
|
|
import type { MiGalleryPost } from '@/models/GalleryPost.js';
|
|
import { bindThis } from '@/decorators.js';
|
|
import { IdService } from '@/core/IdService.js';
|
|
import { UserEntityService } from './UserEntityService.js';
|
|
import { DriveFileEntityService } from './DriveFileEntityService.js';
|
|
|
|
@Injectable()
|
|
export class GalleryPostEntityService {
|
|
constructor(
|
|
@Inject(DI.galleryPostsRepository)
|
|
private galleryPostsRepository: GalleryPostsRepository,
|
|
|
|
@Inject(DI.galleryLikesRepository)
|
|
private galleryLikesRepository: GalleryLikesRepository,
|
|
|
|
private userEntityService: UserEntityService,
|
|
private driveFileEntityService: DriveFileEntityService,
|
|
private idService: IdService,
|
|
) {
|
|
}
|
|
|
|
@bindThis
|
|
public async pack(
|
|
src: MiGalleryPost['id'] | MiGalleryPost,
|
|
me?: { id: MiUser['id'] } | null | undefined,
|
|
hint?: {
|
|
packedUser?: Packed<'UserLite'>
|
|
},
|
|
): Promise<Packed<'GalleryPost'>> {
|
|
const meId = me ? me.id : null;
|
|
const post = typeof src === 'object' ? src : await this.galleryPostsRepository.findOneByOrFail({ id: src });
|
|
|
|
return await awaitAll({
|
|
id: post.id,
|
|
createdAt: this.idService.parse(post.id).date.toISOString(),
|
|
updatedAt: post.updatedAt.toISOString(),
|
|
userId: post.userId,
|
|
user: hint?.packedUser ?? this.userEntityService.pack(post.user ?? post.userId, me),
|
|
title: post.title,
|
|
description: post.description,
|
|
fileIds: post.fileIds,
|
|
// TODO: packMany causes N+1 queries
|
|
files: this.driveFileEntityService.packManyByIds(post.fileIds),
|
|
tags: post.tags.length > 0 ? post.tags : undefined,
|
|
isSensitive: post.isSensitive,
|
|
likedCount: post.likedCount,
|
|
isLiked: meId ? await this.galleryLikesRepository.exists({ where: { postId: post.id, userId: meId } }) : undefined,
|
|
});
|
|
}
|
|
|
|
@bindThis
|
|
public async packMany(
|
|
posts: MiGalleryPost[],
|
|
me?: { id: MiUser['id'] } | null | undefined,
|
|
) {
|
|
const _users = posts.map(({ user, userId }) => user ?? userId);
|
|
const _userMap = await this.userEntityService.packMany(_users, me)
|
|
.then(users => new Map(users.map(u => [u.id, u])));
|
|
return Promise.all(posts.map(post => this.pack(post, me, { packedUser: _userMap.get(post.userId) })));
|
|
}
|
|
}
|
|
|