mirrors/barkey
1
0
Fork 0
mirror of https://codeberg.org/yeentown/barkey.git synced 2025-07-07 12:36:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

implement userFollowersCache

Browse source
This commit is contained in:
Hazelnoot 2025-06-07 21:32:55 -04:00
parent 853b548a43
commit 372714c9b6
2 changed files with 21 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
packages/backend/src/core/CacheService.ts
View file

@ -47,6 +47,7 @@ export class CacheService implements OnApplicationShutdown {
public userBlockedCache: QuantumKVCache<Set<string>>; // NOTE: 「被」Blockキャッシュ public userBlockedCache: QuantumKVCache<Set<string>>; // NOTE: 「被」Blockキャッシュ
public renoteMutingsCache: QuantumKVCache<Set<string>>; public renoteMutingsCache: QuantumKVCache<Set<string>>;
public userFollowingsCache: QuantumKVCache<Map<string, { withReplies: boolean }>>; public userFollowingsCache: QuantumKVCache<Map<string, { withReplies: boolean }>>;
public userFollowersCache: QuantumKVCache<Set<string>>;
protected userFollowStatsCache = new MemoryKVCache<FollowStats>(1000 * 60 * 10); // 10 minutes protected userFollowStatsCache = new MemoryKVCache<FollowStats>(1000 * 60 * 10); // 10 minutes
protected translationsCache: RedisKVCache<CachedTranslationEntity>; protected translationsCache: RedisKVCache<CachedTranslationEntity>;
@ -115,6 +116,11 @@ export class CacheService implements OnApplicationShutdown {
fetcher: (key) => this.followingsRepository.find({ where: { followerId: key }, select: ['followeeId', 'withReplies'] }).then(xs => new Map(xs.map(f => [f.followeeId, { withReplies: f.withReplies }]))), fetcher: (key) => this.followingsRepository.find({ where: { followerId: key }, select: ['followeeId', 'withReplies'] }).then(xs => new Map(xs.map(f => [f.followeeId, { withReplies: f.withReplies }]))),
}); });
this.userFollowersCache = new QuantumKVCache<Set<string>>(this.internalEventService, 'userFollowers', {
lifetime: 1000 * 60 * 30, // 30m
fetcher: (key) => this.followingsRepository.find({ where: { followeeId: key }, select: ['followerId'] }).then(xs => new Set(xs.map(x => x.followerId))),
});
this.translationsCache = new RedisKVCache<CachedTranslationEntity>(this.redisClient, 'translations', { this.translationsCache = new RedisKVCache<CachedTranslationEntity>(this.redisClient, 'translations', {
lifetime: 1000 * 60 * 60 * 24 * 7, // 1 week, lifetime: 1000 * 60 * 60 * 24 * 7, // 1 week,
memoryCacheLifetime: 1000 * 60, // 1 minute memoryCacheLifetime: 1000 * 60, // 1 minute
@ -154,6 +160,7 @@ export class CacheService implements OnApplicationShutdown {
this.userBlockedCache.delete(body.id), this.userBlockedCache.delete(body.id),
this.renoteMutingsCache.delete(body.id), this.renoteMutingsCache.delete(body.id),
this.userFollowingsCache.delete(body.id), this.userFollowingsCache.delete(body.id),
this.userFollowersCache.delete(body.id),
]); ]);
} }
} else { } else {
@ -193,7 +200,10 @@ export class CacheService implements OnApplicationShutdown {
if (follower) follower.followingCount++; if (follower) follower.followingCount++;
const followee = this.userByIdCache.get(body.followeeId); const followee = this.userByIdCache.get(body.followeeId);
if (followee) followee.followersCount++; if (followee) followee.followersCount++;
await this.userFollowingsCache.delete(body.followerId); await Promise.all([
this.userFollowingsCache.delete(body.followerId),
this.userFollowersCache.delete(body.followeeId),
]);
this.userFollowStatsCache.delete(body.followerId); this.userFollowStatsCache.delete(body.followerId);
this.userFollowStatsCache.delete(body.followeeId); this.userFollowStatsCache.delete(body.followeeId);
break; break;
@ -203,7 +213,10 @@ export class CacheService implements OnApplicationShutdown {
if (follower) follower.followingCount--; if (follower) follower.followingCount--;
const followee = this.userByIdCache.get(body.followeeId); const followee = this.userByIdCache.get(body.followeeId);
if (followee) followee.followersCount--; if (followee) followee.followersCount--;
await this.userFollowingsCache.delete(body.followerId); await Promise.all([
this.userFollowingsCache.delete(body.followerId),
this.userFollowersCache.delete(body.followeeId),
]);
this.userFollowStatsCache.delete(body.followerId); this.userFollowStatsCache.delete(body.followerId);
this.userFollowStatsCache.delete(body.followeeId); this.userFollowStatsCache.delete(body.followeeId);
break; break;

6
packages/backend/test/misc/noOpCaches.ts
View file

@ -112,6 +112,12 @@ export class NoOpCacheService extends CacheService {
onSet: this.userFollowingsCache.onSet, onSet: this.userFollowingsCache.onSet,
onDelete: this.userFollowingsCache.onDelete, onDelete: this.userFollowingsCache.onDelete,
}); });
this.userFollowersCache = new NoOpQuantumKVCache<Set<string>>({
internalEventService: fakeInternalEventService,
fetcher: this.userFollowersCache.fetcher,
onSet: this.userFollowersCache.onSet,
onDelete: this.userFollowersCache.onDelete,
});
this.userFollowStatsCache = new NoOpMemoryKVCache<FollowStats>(); this.userFollowStatsCache = new NoOpMemoryKVCache<FollowStats>();
this.translationsCache = new NoOpRedisKVCache<CachedTranslationEntity>({ this.translationsCache = new NoOpRedisKVCache<CachedTranslationEntity>({
redis: fakeRedis, redis: fakeRedis,