Revert "feat: remove mongo analytics (too slow), make CF KV not expire"

This reverts commit 8a7d457.
Kukei-eu · Dec 26, 2023 · f49e8ce · f49e8ce
1 parent 9227a2f
commit f49e8ce
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/functions/index.js b/functions/index.js
@@ -76,8 +76,12 @@ export const onRequestGet = async (context) => {
 			q,
 			JSON.stringify({
 				hasResults,
-			})
+			}),
+			{
+				expirationTtl: 86400, // 24h
+			}
 		);
+		await trackQuery(context.env, { q, hasResults });
 	}
 	const view = {
 		...viewDefaults,

diff --git a/lib/mongo.js b/lib/mongo.js
@@ -75,3 +75,27 @@ export const getUnchecked = async (envs) => {
 
 	return response?.documents?.[0]?.count ?? null;
 };
+
+export const trackQuery = async (envs, {q, hasResults}) => {
+	const d = Date.now();
+	await callMongo(envs, 'updateOne', {
+		collection: 'queries',
+		database: envs.ATLAS_DB,
+		dataSource: envs.ATLAS_SOURCE,
+		filter: {
+			q,
+		},
+		update: {
+			$inc: {
+				used: 1,
+			},
+			$set: {
+				q,
+				hasResults,
+				lastUse: Date.now()
+			}
+		},
+		upsert: true,
+	});
+	console.log(`Analytics via mongo took ${Date.now() - d}ms`);
+};