m/umami
1
0
Fork 0
mirror of https://github.com/kremalicious/umami.git synced 2025-02-14 21:10:34 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add increment to usage.

Browse Source
This commit is contained in:
Brian Cao 2023-06-01 11:54:24 -07:00
parent dc6da1e41f
commit a3639d59a3
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/cache.ts
View File

@ -54,6 +54,11 @@ async function fetchUserBlock(userId: string) {
return redis.get(key);
}
async function incrementUserBlock(userId: string) {
const key = `user:block:${userId}`;
return redis.incr(key);
}
export default {
fetchWebsite,
storeWebsite,
@ -65,5 +70,6 @@ export default {
storeSession,
deleteSession,
fetchUserBlock,
incrementUserBlock,
enabled: redis.enabled,
};

2
lib/session.ts
View File

@ -80,6 +80,8 @@ export async function findSession(req: NextApiRequestCollect) {
async function checkUserBlock(userId: string) {
if (process.env.ENABLE_BLOCKER && (await cache.fetchUserBlock(userId))) {
await cache.incrementUserBlock(userId);
throw new Error('Usage Limit.');
}
}