FastGPT/packages/service/common/mongo/init.ts

import { delay } from '@fastgpt/global/common/system/utils';
import { addLog } from '../system/log';
import type { Mongoose } from 'mongoose';

const maxConnecting = Math.max(30, Number(process.env.DB_MAX_LINK || 20));

/**
 * connect MongoDB and init data
 */
export async function connectMongo(props: {
  db: Mongoose;
  url: string;
  connectedCb?: () => void;
}): Promise<Mongoose> {
  const { db, url, connectedCb } = props;

  /* Connecting, connected will return */
  if (db.connection.readyState !== 0) {
    return db;
  }

  const RemoveListeners = () => {
    db.connection.removeAllListeners('error');
    db.connection.removeAllListeners('disconnected');
  };

  console.log('MongoDB start connect');
  try {
    // Remove existing listeners to prevent duplicates
    RemoveListeners();
    db.set('strictQuery', 'throw');

    db.connection.on('error', async (error) => {
      console.log('mongo error', error);
      try {
        if (db.connection.readyState !== 0) {
          RemoveListeners();
          await db.disconnect();
          await delay(1000);
          await connectMongo(props);
        }
      } catch (error) {}
    });
    db.connection.on('disconnected', async () => {
      console.log('mongo disconnected');
      try {
        if (db.connection.readyState !== 0) {
          RemoveListeners();
          await db.disconnect();
          await delay(1000);
          await connectMongo(props);
        }
      } catch (error) {}
    });

    const options = {
      bufferCommands: true,
      maxConnecting: maxConnecting,
      maxPoolSize: maxConnecting,
      minPoolSize: 20,
      connectTimeoutMS: 60000,
      waitQueueTimeoutMS: 60000,
      socketTimeoutMS: 60000,
      maxIdleTimeMS: 300000,
      retryWrites: true,
      retryReads: true
    };

    await db.connect(url, options);
    console.log('mongo connected');

    connectedCb?.();

    return db;
  } catch (error) {
    addLog.error('Mongo connect error', error);

    await db.disconnect();

    await delay(1000);
    return connectMongo(props);
  }
}