mongodb-memory-bank-mcp-v2/dist/db/connection.js

MongoDB-powered Memory Bank MCP server with hybrid search capabilities for AI assistants

import { MongoClient } from 'mongodb';
import { logger } from '../utils/logger.js';
let client = null;
let db = null;
export async function connectToMongoDB() {
    const uri = process.env.MONGODB_URI;
    if (!uri) {
        throw new Error('MONGODB_URI environment variable is not set');
    }
    try {
        client = new MongoClient(uri, {
            maxPoolSize: parseInt(process.env.MONGODB_MAX_POOL_SIZE || '10'),
            minPoolSize: parseInt(process.env.MONGODB_MIN_POOL_SIZE || '2'),
        });
        await client.connect();
        const dbName = process.env.MEMORY_BANK_DB || 'memory_bank';
        db = client.db(dbName);
        // Verify connection
        await db.admin().ping();
        logger.info(`Connected to MongoDB database: ${dbName}`);
    }
    catch (error) {
        logger.error('MongoDB connection error:', error);
        throw error;
    }
}
export async function closeMongoDBConnection() {
    if (client) {
        try {
            await client.close();
            logger.info('MongoDB connection closed');
        }
        catch (error) {
            logger.error('Error closing MongoDB connection:', error);
        }
    }
}
export function getDb() {
    if (!db) {
        throw new Error('Database not connected. Call connectToMongoDB() first.');
    }
    return db;
}
export function getMemoryCollection() {
    const collectionName = process.env.MEMORY_BANK_COLLECTION || 'memory_bank_documents';
    return getDb().collection(collectionName);
}
//# sourceMappingURL=connection.js.map