misskey/packages/backend/src/core/MetaService.ts

138 lines
3.4 KiB
TypeScript
Raw Normal View History

/*
* SPDX-FileCopyrightText: syuilo and other misskey contributors
* SPDX-License-Identifier: AGPL-3.0-only
*/
2022-09-17 20:27:08 +02:00
import { Inject, Injectable } from '@nestjs/common';
import { DataSource } from 'typeorm';
2023-04-14 06:50:05 +02:00
import * as Redis from 'ioredis';
2022-09-17 20:27:08 +02:00
import { DI } from '@/di-symbols.js';
import { MiMeta } from '@/models/entities/Meta.js';
import { GlobalEventService } from '@/core/GlobalEventService.js';
import { bindThis } from '@/decorators.js';
import { StreamMessages } from '@/server/api/stream/types.js';
import type { OnApplicationShutdown } from '@nestjs/common';
2022-09-17 20:27:08 +02:00
@Injectable()
export class MetaService implements OnApplicationShutdown {
private cache: MiMeta | undefined;
2022-09-18 20:11:50 +02:00
private intervalId: NodeJS.Timer;
2022-09-17 20:27:08 +02:00
constructor(
@Inject(DI.redisForSub)
private redisForSub: Redis.Redis,
2022-09-17 20:27:08 +02:00
@Inject(DI.db)
private db: DataSource,
private globalEventService: GlobalEventService,
2022-09-17 20:27:08 +02:00
) {
//this.onMessage = this.onMessage.bind(this);
2022-09-17 20:27:08 +02:00
if (process.env.NODE_ENV !== 'test') {
2022-09-18 20:11:50 +02:00
this.intervalId = setInterval(() => {
2022-09-17 20:27:08 +02:00
this.fetch(true).then(meta => {
// fetch内でもセットしてるけど仕様変更の可能性もあるため一応
2022-09-18 20:11:50 +02:00
this.cache = meta;
2022-09-17 20:27:08 +02:00
});
}, 1000 * 60 * 5);
2022-09-17 20:27:08 +02:00
}
this.redisForSub.on('message', this.onMessage);
2022-09-17 20:27:08 +02:00
}
@bindThis
2022-09-24 00:12:11 +02:00
private async onMessage(_: string, data: string): Promise<void> {
const obj = JSON.parse(data);
if (obj.channel === 'internal') {
const { type, body } = obj.message as StreamMessages['internal']['payload'];
switch (type) {
case 'metaUpdated': {
this.cache = body;
break;
}
default:
break;
}
}
}
@bindThis
public async fetch(noCache = false): Promise<MiMeta> {
2022-09-18 20:11:50 +02:00
if (!noCache && this.cache) return this.cache;
2022-09-17 20:27:08 +02:00
return await this.db.transaction(async transactionalEntityManager => {
// 過去のバグでレコードが複数出来てしまっている可能性があるので新しいIDを優先する
const metas = await transactionalEntityManager.find(MiMeta, {
2022-09-17 20:27:08 +02:00
order: {
id: 'DESC',
},
});
2022-09-17 20:27:08 +02:00
const meta = metas[0];
2022-09-17 20:27:08 +02:00
if (meta) {
2022-09-18 20:11:50 +02:00
this.cache = meta;
2022-09-17 20:27:08 +02:00
return meta;
} else {
// metaが空のときfetchMetaが同時に呼ばれるとここが同時に呼ばれてしまうことがあるのでフェイルセーフなupsertを使う
const saved = await transactionalEntityManager
.upsert(
MiMeta,
2022-09-17 20:27:08 +02:00
{
id: 'x',
},
['id'],
)
.then((x) => transactionalEntityManager.findOneByOrFail(MiMeta, x.identifiers[0]));
2022-09-18 20:11:50 +02:00
this.cache = saved;
2022-09-17 20:27:08 +02:00
return saved;
}
});
}
@bindThis
public async update(data: Partial<MiMeta>): Promise<MiMeta> {
const updated = await this.db.transaction(async transactionalEntityManager => {
const metas = await transactionalEntityManager.find(MiMeta, {
order: {
id: 'DESC',
},
});
const meta = metas[0];
if (meta) {
await transactionalEntityManager.update(MiMeta, meta.id, data);
const metas = await transactionalEntityManager.find(MiMeta, {
order: {
id: 'DESC',
},
});
return metas[0];
} else {
return await transactionalEntityManager.save(MiMeta, data);
}
});
this.globalEventService.publishInternalEvent('metaUpdated', updated);
return updated;
}
@bindThis
2023-05-29 06:21:26 +02:00
public dispose(): void {
2022-09-18 20:11:50 +02:00
clearInterval(this.intervalId);
this.redisForSub.off('message', this.onMessage);
2022-09-17 20:27:08 +02:00
}
2023-05-29 06:21:26 +02:00
@bindThis
public onApplicationShutdown(signal?: string | undefined): void {
this.dispose();
}
2022-09-17 20:27:08 +02:00
}