67 lines
1.7 KiB
TypeScript
67 lines
1.7 KiB
TypeScript
|
import type { PrismaClient } from "@calcom/prisma";
|
||
|
|
|
||
|
|
import type { AppFlags } from "../config";
|
||
|
|
|
||
|
|
export async function getFeatureFlagMap(prisma: PrismaClient) {
|
||
|
|
const flags = await prisma.feature.findMany({
|
||
|
|
orderBy: { slug: "asc" },
|
||
|
|
cacheStrategy: { swr: 300, ttl: 300 },
|
||
|
|
});
|
||
|
|
return flags.reduce((acc, flag) => {
|
||
|
|
acc[flag.slug as keyof AppFlags] = flag.enabled;
|
||
|
|
return acc;
|
||
|
|
}, {} as Partial<AppFlags>);
|
||
|
|
}
|
||
|
|
|
||
|
|
interface CacheEntry {
|
||
|
|
value: boolean; // adapt to other supported value types in the future
|
||
|
|
expiry: number;
|
||
|
|
}
|
||
|
|
|
||
|
|
interface CacheOptions {
|
||
|
|
ttl: number; // time in ms
|
||
|
|
}
|
||
|
|
|
||
|
|
const featureFlagCache = new Map<keyof AppFlags, CacheEntry>();
|
||
|
|
|
||
|
|
const isExpired = (entry: CacheEntry): boolean => {
|
||
|
|
return Date.now() > entry.expiry;
|
||
|
|
};
|
||
|
|
|
||
|
|
export const getFeatureFlag = async (
|
||
|
|
prisma: PrismaClient,
|
||
|
|
slug: keyof AppFlags,
|
||
|
|
options: CacheOptions = { ttl: 5 * 60 * 1000 }
|
||
|
|
): Promise<boolean> => {
|
||
|
|
// pre-compute all app flags, each one will independelty reload it's own state after expiry.
|
||
|
|
|
||
|
|
if (featureFlagCache.size === 0) {
|
||
|
|
const flags = await prisma.feature.findMany({ orderBy: { slug: "asc" } });
|
||
|
|
flags.forEach((flag) => {
|
||
|
|
featureFlagCache.set(flag.slug as keyof AppFlags, {
|
||
|
|
value: flag.enabled,
|
||
|
|
expiry: Date.now() + options.ttl,
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
const cacheEntry = featureFlagCache.get(slug);
|
||
|
|
|
||
|
|
if (cacheEntry && !isExpired(cacheEntry)) {
|
||
|
|
return cacheEntry.value;
|
||
|
|
}
|
||
|
|
|
||
|
|
const flag = await prisma.feature.findUnique({
|
||
|
|
where: {
|
||
|
|
slug,
|
||
|
|
},
|
||
|
|
});
|
||
|
|
|
||
|
|
const isEnabled = Boolean(flag && flag.enabled);
|
||
|
|
const expiry = Date.now() + options.ttl;
|
||
|
|
|
||
|
|
featureFlagCache.set(slug, { value: isEnabled, expiry });
|
||
|
|
|
||
|
|
return isEnabled;
|
||
|
|
};
|