Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-18 14:56:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: check quota correctly (#6561)

Browse Source
This commit is contained in:
darkskygit
2024-04-16 09:41:48 +00:00
parent 0ca8a23dd8
commit 1b0864eb60
26 changed files with 309 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
packages/backend/server/src/data/migrations/utils/user-quotas.ts
View File

@@ -1,7 +1,7 @@
import { PrismaClient } from '@prisma/client';
import { FeatureKind } from '../../../core/features';
import { Quotas } from '../../../core/quota/schema';
import { getLatestQuota } from '../../../core/quota/schema';
import { Quota, QuotaType } from '../../../core/quota/types';
import { upsertFeature } from './user-features';
@@ -21,10 +21,10 @@ export async function upgradeQuotaVersion(
});
// find all users that have old free plan
const userIds = await db.user.findMany({
const userIds = await tx.user.findMany({
where: {
features: {
every: {
some: {
feature: {
type: FeatureKind.Quota,
feature: quota.feature,
@@ -65,13 +65,19 @@ export async function upgradeQuotaVersion(
});
}
export async function upsertLatestQuotaVersion(
db: PrismaClient,
type: QuotaType
) {
const latestQuota = getLatestQuota(type);
await upsertFeature(db, latestQuota);
}
export async function upgradeLatestQuotaVersion(
db: PrismaClient,
type: QuotaType,
reason: string
) {
const quota = Quotas.filter(f => f.feature === type);
quota.sort((a, b) => b.version - a.version);
const latestQuota = quota[0];
const latestQuota = getLatestQuota(type);
await upgradeQuotaVersion(db, latestQuota, reason);
}