Skip to content
Snippets Groups Projects
Commit dc43fc68 authored by syuilo's avatar syuilo
Browse files

enhance(backend): metaのポーリング頻度を減らし、redisで更新を受け取るように

parent 55f8a641
No related branches found
No related tags found
No related merge requests found
import { Inject, Injectable } from '@nestjs/common';
import { DataSource } from 'typeorm';
import type { UsersRepository } from '@/models/index.js';
import Redis from 'ioredis';
import { DI } from '@/di-symbols.js';
import { Meta } from '@/models/entities/Meta.js';
import type { OnApplicationShutdown } from '@nestjs/common';
......@@ -11,19 +11,27 @@ export class MetaService implements OnApplicationShutdown {
private intervalId: NodeJS.Timer;
constructor(
@Inject(DI.redisSubscriber)
private redisSubscriber: Redis.Redis,
@Inject(DI.db)
private db: DataSource,
) {
this.onMessage = this.onMessage.bind(this);
if (process.env.NODE_ENV !== 'test') {
this.intervalId = setInterval(() => {
this.fetch(true).then(meta => {
// fetch内でもセットしてるけど仕様変更の可能性もあるため一応
this.cache = meta;
});
}, 1000 * 10);
}, 1000 * 60 * 5);
}
this.redisSubscriber.on('message', this.onMessage);
}
async fetch(noCache = false): Promise<Meta> {
public async fetch(noCache = false): Promise<Meta> {
if (!noCache && this.cache) return this.cache;
return await this.db.transaction(async transactionalEntityManager => {
......@@ -56,8 +64,25 @@ export class MetaService implements OnApplicationShutdown {
}
});
}
private async onMessage(_, data) {
const obj = JSON.parse(data);
if (obj.channel === 'internal') {
const { type, body } = obj.message;
switch (type) {
case 'metaUpdated': {
this.cache = body;
break;
}
default:
break;
}
}
}
public onApplicationShutdown(signal?: string | undefined) {
clearInterval(this.intervalId);
this.redisSubscriber.off('message', this.onMessage);
}
}
......@@ -5,6 +5,7 @@ import { ModerationLogService } from '@/core/ModerationLogService.js';
import { DB_MAX_NOTE_TEXT_LENGTH } from '@/misc/hard-limits.js';
import { Endpoint } from '@/server/api/endpoint-base.js';
import { DI } from '@/di-symbols.js';
import { GlobalEventService } from '@/core/GlobalEventService.js';
export const meta = {
tags: ['admin'],
......@@ -115,6 +116,7 @@ export default class extends Endpoint<typeof meta, typeof paramDef> {
@Inject(DI.db)
private db: DataSource,
private globalEventService: GlobalEventService,
private moderationLogService: ModerationLogService,
) {
super(meta, paramDef, async (ps, me) => {
......@@ -436,7 +438,7 @@ export default class extends Endpoint<typeof meta, typeof paramDef> {
set.enableActiveEmailValidation = ps.enableActiveEmailValidation;
}
await this.db.transaction(async transactionalEntityManager => {
const updated = await this.db.transaction(async transactionalEntityManager => {
const metas = await transactionalEntityManager.find(Meta, {
order: {
id: 'DESC',
......@@ -447,11 +449,21 @@ export default class extends Endpoint<typeof meta, typeof paramDef> {
if (meta) {
await transactionalEntityManager.update(Meta, meta.id, set);
const metas = await transactionalEntityManager.find(Meta, {
order: {
id: 'DESC',
},
});
return metas[0];
} else {
await transactionalEntityManager.save(Meta, set);
return await transactionalEntityManager.save(Meta, set);
}
});
this.globalEventService.publishInternalEvent('metaUpdated', updated);
this.moderationLogService.insertModerationLog(me, 'updateMeta');
});
}
......
......@@ -13,6 +13,7 @@ import type { Signin } from '@/models/entities/Signin.js';
import type { Page } from '@/models/entities/Page.js';
import type { Packed } from '@/misc/schema.js';
import type { Webhook } from '@/models/entities/Webhook.js';
import type { Meta } from '@/models/entities/Meta.js';
import type Emitter from 'strict-event-emitter-types';
import type { EventEmitter } from 'events';
......@@ -29,6 +30,7 @@ export interface InternalStreamTypes {
antennaCreated: Antenna;
antennaDeleted: Antenna;
antennaUpdated: Antenna;
metaUpdated: Meta,
}
export interface BroadcastTypes {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment