YueMi-Development/RakunNakun-AI
3
0
Fork 0
mirror of https://github.com/NekoMonci12/RakunNakun-AI.git synced 2025-12-19 14:59:15 +00:00
Code Issues Packages Releases Activity

MongoDB Index + Pagination

Browse Source
This commit is contained in:
NekoMonci12
2025-06-04 07:26:07 +07:00
parent 437893b342
commit 22211e5987
4 changed files with 117 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
mongoCacheManager.js
View File

@@ -23,7 +23,8 @@ class MongoCacheManager {
this.collection = this.client.db(this.dbName).collection(this.collectionName);
this.connected = true;
console.log("[MongoCache] Connected to MongoDB for caching.");
await this.collection.createIndex({ hash: 1 }, { unique: true });
await this.collection.createIndex({ key: 1 }, { unique: true });
// Try a dry-run write to detect read-only access
try {
await this.collection.insertOne({ _test: true });
@@ -74,18 +75,21 @@ class MongoCacheManager {
return await this.collection.find({ embedding: { $exists: true } }).toArray();
}
async getEmbeddingsPage(page = 0, pageSize = 100) {
await this.connect();
return await this.collection
.find({ embedding: { $exists: true } })
.skip(page * pageSize)
.limit(pageSize)
.toArray();
}
async setCache(input, value, embedding, hash) {
await this.connect();
if (!this.connected || !this.collection || this.readOnly) return;
const key = this.normalize(input);
const count = await this.collection.estimatedDocumentCount();
const maxSize = 5000;
if (count >= maxSize) {
await this.collection.deleteOne({}, { sort: { updatedAt: 1 } });
}
await this.collection.updateOne(
{ key },
{