github/PreMiD
1
1
Fork 0
mirror of https://github.com/PreMiD/PreMiD.git synced 2026-04-05 20:31:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: add environment variable to disable

Browse Source
This commit is contained in:
Bas950
2024-12-21 12:19:41 +01:00
parent 255f840275
commit d862584a30
2 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
apps/api-worker/environment.d.ts vendored
View File

@@ -3,5 +3,6 @@ declare namespace NodeJS {
NODE_ENV?: "development" | "production" | "test";
DATABASE_URL?: string;
SESSION_KEEP_ALIVE_INTERVAL?: string;
HEARTBEATS?: "true" | "false";
}
}

27
apps/api-worker/src/graphql/resolvers/v5/Mutation/heartbeat.ts
View File

@@ -1,3 +1,4 @@
import process from "node:process";
import { type } from "arktype";
import { GraphQLError } from "graphql";
import type { MutationResolvers } from "../../../../generated/graphql-v5.js";
@@ -32,18 +33,20 @@ const mutation: MutationResolvers["heartbeat"] = async (_parent, input, context)
// * Use Redis Hash with 'service' in the key to store heartbeat data
const redisKey = `pmd-api.heartbeatUpdates.${out.identifier}`;
await redis.hset(redisKey, {
service: out.presence?.service,
version: out.presence?.version,
language: out.presence?.language,
since: out.presence?.since.toString(),
extension_version: out.extension.version,
extension_language: out.extension.language,
extension_connected_app: out.extension.connected?.app?.toString(),
extension_connected_discord: out.extension.connected?.discord?.toString(),
ip_address: userIp,
});
await redis.expire(redisKey, 300);
if (process.env.HEARTBEATS !== "false") {
await redis.hset(redisKey, {
service: out.presence?.service,
version: out.presence?.version,
language: out.presence?.language,
since: out.presence?.since.toString(),
extension_version: out.extension.version,
extension_language: out.extension.language,
extension_connected_app: out.extension.connected?.app?.toString(),
extension_connected_discord: out.extension.connected?.discord?.toString(),
ip_address: userIp,
});
await redis.expire(redisKey, 300);
}
return {
__typename: "HeartbeatResult",