xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-10-08 19:16:37 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

getRedisClient

Browse Source
This commit is contained in:
DirkSchlossmacher 2023-11-14 11:46:32 +01:00
parent 993c7be12d
commit cc8b1e35c6
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
app/utils/cloud/redisRestClient.ts
View File

@ -23,11 +23,6 @@ const getRedisClient = () => {
};
// Use getRedisClient() to access the Redis client in your functions
export const incrementSignInCount = async (email: string | undefined, dateKey: string) => {
const redis = getRedisClient();
// ... rest of your function
};
export const incrementSignInCount = async (email: string | undefined, dateKey: string) => {
if (!email) {
@ -35,6 +30,8 @@ export const incrementSignInCount = async (email: string | undefined, dateKey: s
return;
}
const redis = getRedisClient();
try {
await redis.hincrby(`signin_count:${email}`, dateKey, 1);
} catch (error) {
@ -48,6 +45,8 @@ export const incrementSessionRefreshCount = async (email: string | undefined, da
return;
}
const redis = getRedisClient();
try {
await redis.hincrby(`session_refreshes:${email}`, dateKey, 1);
} catch (error) {
@ -66,6 +65,8 @@ export const incrementTokenCounts = async (
return;
}
const redis = getRedisClient();
try {
await redis.hincrby(`tokens:${email}`, `${dateKey}:completion_tokens`, completionTokens);
await redis.hincrby(`tokens:${email}`, `${dateKey}:prompt_tokens`, promptTokens);
@ -76,6 +77,9 @@ export const incrementTokenCounts = async (
export const getAvailableDateKeys = async (): Promise<string[]> => {
const redis = getRedisClient();
try {
const keys = await redis.keys('signin_count:*');
return keys.map(key => key.split(':')[1]);
@ -86,6 +90,9 @@ export const getAvailableDateKeys = async (): Promise<string[]> => {
};
export const getSignInCountForPeriod = async (dateKey: string): Promise<number> => {
const redis = getRedisClient();
try {
const counts = await redis.hgetall(`signin_count:${dateKey}`);
if (counts === null) {
@ -108,6 +115,9 @@ export const getSignInCountForPeriod = async (dateKey: string): Promise<number>
export const getDetailsByUser = async (dateKey: string): Promise<Record<string, number>> => {
const redis = getRedisClient();
try {
const rawCounts = await redis.hgetall(`signin_count:${dateKey}`);
const counts: Record<string, number> = {};