Elizhium/litlyx
1
0
Fork 0
mirror of https://github.com/Litlyx/litlyx synced 2025-12-10 07:48:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix limits

Browse Source
This commit is contained in:
Emily
2024-08-29 16:31:50 +02:00
parent 79309cc537
commit ab07ffb108
2 changed files with 27 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
broker/src/StreamLoopController.ts
View File

@@ -28,18 +28,18 @@ export async function startStreamLoop() {
async function processStreamEvent(data: Record<string, string>) { export async function processStreamEvent(data: Record<string, string>) {
try { try {
const eventType = data._type; const eventType = data._type;
if (!eventType) return; if (!eventType) return;
const { pid, sessionHash } = data; const { pid, sessionHash } = data;
const project = await ProjectModel.exists({ _id: pid }); const project = await ProjectModel.exists({ _id: pid });
if (!project) return; if (!project) return;
if (eventType === 'event') return await process_event(data, sessionHash); if (eventType === 'event') return await process_event(data, sessionHash);
if (eventType === 'keep_alive') return await process_keep_alive(data, sessionHash); if (eventType === 'keep_alive') return await process_keep_alive(data, sessionHash);
if (eventType === 'visit') return await process_visit(data, sessionHash); if (eventType === 'visit') return await process_visit(data, sessionHash);
@@ -50,17 +50,23 @@ async function processStreamEvent(data: Record<string, string>) {
} }
async function checkLimits(project_id: string) {
const projectLimits = await ProjectLimitModel.findOne({ project_id });
if (!projectLimits) return false;
const TOTAL_COUNT = projectLimits.events + projectLimits.visits;
const COUNT_LIMIT = projectLimits.limit;
if ((TOTAL_COUNT) > COUNT_LIMIT * EVENT_LOG_LIMIT_PERCENT) return false;
await checkLimitsForEmail(projectLimits);
return true;
}
async function process_visit(data: Record<string, string>, sessionHash: string) { async function process_visit(data: Record<string, string>, sessionHash: string) {
const { pid, ip, website, page, referrer, userAgent, flowHash } = data; const { pid, ip, website, page, referrer, userAgent, flowHash } = data;
const projectLimits = await ProjectLimitModel.findOne({ project_id: pid }); const canLog = await checkLimits(pid);
if (!projectLimits) return; if (!canLog) return;
const TOTAL_COUNT = projectLimits.events + projectLimits.visits;
const COUNT_LIMIT = projectLimits.limit;
if ((TOTAL_COUNT) > COUNT_LIMIT * EVENT_LOG_LIMIT_PERCENT) return;
await checkLimitsForEmail(projectLimits);
let referrerParsed; let referrerParsed;
try { try {
@@ -97,7 +103,10 @@ async function process_keep_alive(data: Record<string, string>, sessionHash: str
const { pid, instant, flowHash } = data; const { pid, instant, flowHash } = data;
const existingSession = await SessionModel.findOne({ project_id: pid }, { _id: 1 }); const canLog = await checkLimits(pid);
if (!canLog) return;
const existingSession = await SessionModel.findOne({ project_id: pid, session: sessionHash }, { _id: 1 });
if (!existingSession) { if (!existingSession) {
await ProjectCountModel.updateOne({ project_id: pid }, { $inc: { 'sessions': 1 } }, { upsert: true }); await ProjectCountModel.updateOne({ project_id: pid }, { $inc: { 'sessions': 1 } }, { upsert: true });
} }
@@ -123,6 +132,9 @@ async function process_event(data: Record<string, string>, sessionHash: string)
const { name, metadata, pid, flowHash } = data; const { name, metadata, pid, flowHash } = data;
const canLog = await checkLimits(pid);
if (!canLog) return;
let metadataObject; let metadataObject;
try { try {
if (metadata) metadataObject = JSON.parse(metadata); if (metadata) metadataObject = JSON.parse(metadata);

6
shared/services/DatabaseService.ts
View File

@@ -3,4 +3,8 @@ import mongoose from "mongoose";
export async function connectDatabase(connectionString: string) { export async function connectDatabase(connectionString: string) {
await mongoose.connect(connectionString); await mongoose.connect(connectionString);
} }
export async function disconnectDatabase() {
await mongoose.disconnect();
}