2023-07-27 00:31:52 -05:00
|
|
|
/*
|
2024-02-11 20:37:45 -06:00
|
|
|
* SPDX-FileCopyrightText: syuilo and misskey-project
|
2023-07-27 00:31:52 -05:00
|
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
*/
|
|
|
|
|
2022-09-17 13:27:08 -05:00
|
|
|
import { Inject, Injectable } from '@nestjs/common';
|
|
|
|
import { DataSource } from 'typeorm';
|
2023-04-13 23:50:05 -05:00
|
|
|
import * as Redis from 'ioredis';
|
2022-09-17 13:27:08 -05:00
|
|
|
import { DI } from '@/di-symbols.js';
|
2023-09-19 21:33:36 -05:00
|
|
|
import { MiMeta } from '@/models/Meta.js';
|
2022-09-20 12:52:19 -05:00
|
|
|
import { GlobalEventService } from '@/core/GlobalEventService.js';
|
2022-12-04 00:03:09 -06:00
|
|
|
import { bindThis } from '@/decorators.js';
|
2023-09-28 21:29:54 -05:00
|
|
|
import type { GlobalEvents } from '@/core/GlobalEventService.js';
|
2023-12-20 20:34:19 -06:00
|
|
|
import { FeaturedService } from '@/core/FeaturedService.js';
|
2023-01-13 17:27:23 -06:00
|
|
|
import type { OnApplicationShutdown } from '@nestjs/common';
|
2022-09-17 13:27:08 -05:00
|
|
|
|
|
|
|
@Injectable()
|
|
|
|
export class MetaService implements OnApplicationShutdown {
|
2023-08-16 03:51:28 -05:00
|
|
|
private cache: MiMeta | undefined;
|
2023-09-03 23:33:38 -05:00
|
|
|
private intervalId: NodeJS.Timeout;
|
2022-09-17 13:27:08 -05:00
|
|
|
|
|
|
|
constructor(
|
2023-04-09 03:09:27 -05:00
|
|
|
@Inject(DI.redisForSub)
|
|
|
|
private redisForSub: Redis.Redis,
|
2022-09-20 12:35:49 -05:00
|
|
|
|
2022-09-17 13:27:08 -05:00
|
|
|
@Inject(DI.db)
|
|
|
|
private db: DataSource,
|
2022-09-20 12:52:19 -05:00
|
|
|
|
2023-12-20 20:34:19 -06:00
|
|
|
private featuredService: FeaturedService,
|
2022-09-20 12:52:19 -05:00
|
|
|
private globalEventService: GlobalEventService,
|
2022-09-17 13:27:08 -05:00
|
|
|
) {
|
2022-12-04 00:03:09 -06:00
|
|
|
//this.onMessage = this.onMessage.bind(this);
|
2022-09-20 12:35:49 -05:00
|
|
|
|
2022-09-17 13:27:08 -05:00
|
|
|
if (process.env.NODE_ENV !== 'test') {
|
2022-09-18 13:11:50 -05:00
|
|
|
this.intervalId = setInterval(() => {
|
2022-09-17 13:27:08 -05:00
|
|
|
this.fetch(true).then(meta => {
|
2022-09-20 12:35:49 -05:00
|
|
|
// fetch内でもセットしてるけど仕様変更の可能性もあるため一応
|
2022-09-18 13:11:50 -05:00
|
|
|
this.cache = meta;
|
2022-09-17 13:27:08 -05:00
|
|
|
});
|
2022-09-20 12:35:49 -05:00
|
|
|
}, 1000 * 60 * 5);
|
2022-09-17 13:27:08 -05:00
|
|
|
}
|
2022-09-20 12:35:49 -05:00
|
|
|
|
2023-04-09 03:09:27 -05:00
|
|
|
this.redisForSub.on('message', this.onMessage);
|
2022-09-17 13:27:08 -05:00
|
|
|
}
|
|
|
|
|
2022-12-04 00:03:09 -06:00
|
|
|
@bindThis
|
2022-09-23 17:12:11 -05:00
|
|
|
private async onMessage(_: string, data: string): Promise<void> {
|
2022-09-20 12:52:19 -05:00
|
|
|
const obj = JSON.parse(data);
|
|
|
|
|
|
|
|
if (obj.channel === 'internal') {
|
2023-09-28 21:29:54 -05:00
|
|
|
const { type, body } = obj.message as GlobalEvents['internal']['payload'];
|
2022-09-20 12:52:19 -05:00
|
|
|
switch (type) {
|
|
|
|
case 'metaUpdated': {
|
2024-01-22 02:44:03 -06:00
|
|
|
this.cache = { // TODO: このあたりのデシリアライズ処理は各modelファイル内に関数としてexportしたい
|
|
|
|
...body,
|
|
|
|
proxyAccount: null, // joinなカラムは通常取ってこないので
|
|
|
|
};
|
2022-09-20 12:52:19 -05:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-12-04 00:03:09 -06:00
|
|
|
@bindThis
|
2023-08-16 03:51:28 -05:00
|
|
|
public async fetch(noCache = false): Promise<MiMeta> {
|
2022-09-18 13:11:50 -05:00
|
|
|
if (!noCache && this.cache) return this.cache;
|
2023-07-07 17:08:16 -05:00
|
|
|
|
2022-09-17 13:27:08 -05:00
|
|
|
return await this.db.transaction(async transactionalEntityManager => {
|
|
|
|
// 過去のバグでレコードが複数出来てしまっている可能性があるので新しいIDを優先する
|
2023-08-16 03:51:28 -05:00
|
|
|
const metas = await transactionalEntityManager.find(MiMeta, {
|
2022-09-17 13:27:08 -05:00
|
|
|
order: {
|
|
|
|
id: 'DESC',
|
|
|
|
},
|
|
|
|
});
|
2023-07-07 17:08:16 -05:00
|
|
|
|
2022-09-17 13:27:08 -05:00
|
|
|
const meta = metas[0];
|
2023-07-07 17:08:16 -05:00
|
|
|
|
2022-09-17 13:27:08 -05:00
|
|
|
if (meta) {
|
2022-09-18 13:11:50 -05:00
|
|
|
this.cache = meta;
|
2022-09-17 13:27:08 -05:00
|
|
|
return meta;
|
|
|
|
} else {
|
|
|
|
// metaが空のときfetchMetaが同時に呼ばれるとここが同時に呼ばれてしまうことがあるのでフェイルセーフなupsertを使う
|
|
|
|
const saved = await transactionalEntityManager
|
|
|
|
.upsert(
|
2023-08-16 03:51:28 -05:00
|
|
|
MiMeta,
|
2022-09-17 13:27:08 -05:00
|
|
|
{
|
|
|
|
id: 'x',
|
|
|
|
},
|
|
|
|
['id'],
|
|
|
|
)
|
2023-08-16 03:51:28 -05:00
|
|
|
.then((x) => transactionalEntityManager.findOneByOrFail(MiMeta, x.identifiers[0]));
|
2023-07-07 17:08:16 -05:00
|
|
|
|
2022-09-18 13:11:50 -05:00
|
|
|
this.cache = saved;
|
2022-09-17 13:27:08 -05:00
|
|
|
return saved;
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
2022-09-20 12:35:49 -05:00
|
|
|
|
2022-12-04 00:03:09 -06:00
|
|
|
@bindThis
|
2023-08-16 03:51:28 -05:00
|
|
|
public async update(data: Partial<MiMeta>): Promise<MiMeta> {
|
2023-12-20 20:34:19 -06:00
|
|
|
let before: MiMeta | undefined;
|
|
|
|
|
2022-09-20 12:52:19 -05:00
|
|
|
const updated = await this.db.transaction(async transactionalEntityManager => {
|
2023-08-16 03:51:28 -05:00
|
|
|
const metas = await transactionalEntityManager.find(MiMeta, {
|
2022-09-20 12:52:19 -05:00
|
|
|
order: {
|
|
|
|
id: 'DESC',
|
|
|
|
},
|
|
|
|
});
|
2022-09-20 12:35:49 -05:00
|
|
|
|
2023-12-20 20:34:19 -06:00
|
|
|
before = metas[0];
|
2022-09-20 12:52:19 -05:00
|
|
|
|
2023-12-20 20:34:19 -06:00
|
|
|
if (before) {
|
|
|
|
await transactionalEntityManager.update(MiMeta, before.id, data);
|
2022-09-20 12:52:19 -05:00
|
|
|
|
2023-08-16 03:51:28 -05:00
|
|
|
const metas = await transactionalEntityManager.find(MiMeta, {
|
2022-09-20 12:52:19 -05:00
|
|
|
order: {
|
|
|
|
id: 'DESC',
|
|
|
|
},
|
|
|
|
});
|
|
|
|
|
|
|
|
return metas[0];
|
|
|
|
} else {
|
2023-08-16 03:51:28 -05:00
|
|
|
return await transactionalEntityManager.save(MiMeta, data);
|
2022-09-20 12:35:49 -05:00
|
|
|
}
|
2022-09-20 12:52:19 -05:00
|
|
|
});
|
|
|
|
|
2023-12-20 20:34:19 -06:00
|
|
|
if (data.hiddenTags) {
|
|
|
|
process.nextTick(() => {
|
|
|
|
const hiddenTags = new Set<string>(data.hiddenTags);
|
|
|
|
if (before) {
|
|
|
|
for (const previousHiddenTag of before.hiddenTags) {
|
|
|
|
hiddenTags.delete(previousHiddenTag);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (const hiddenTag of hiddenTags) {
|
|
|
|
this.featuredService.removeHashtagsFromRanking(hiddenTag);
|
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2022-09-20 12:52:19 -05:00
|
|
|
this.globalEventService.publishInternalEvent('metaUpdated', updated);
|
|
|
|
|
|
|
|
return updated;
|
2022-09-20 12:35:49 -05:00
|
|
|
}
|
|
|
|
|
2022-12-04 00:03:09 -06:00
|
|
|
@bindThis
|
2023-05-28 23:21:26 -05:00
|
|
|
public dispose(): void {
|
2022-09-18 13:11:50 -05:00
|
|
|
clearInterval(this.intervalId);
|
2023-04-09 03:09:27 -05:00
|
|
|
this.redisForSub.off('message', this.onMessage);
|
2022-09-17 13:27:08 -05:00
|
|
|
}
|
2023-05-28 23:21:26 -05:00
|
|
|
|
|
|
|
@bindThis
|
|
|
|
public onApplicationShutdown(signal?: string | undefined): void {
|
|
|
|
this.dispose();
|
|
|
|
}
|
2022-09-17 13:27:08 -05:00
|
|
|
}
|