import mongoose from "mongoose";

export async function queryDB(callback: (db: mongoose.Connection) => Promise<any>): Promise<any> {
  const SSO_DB = mongoose.createConnection(process.env.MONGO_SSO_URI ?? '');

  await new Promise<void>((resolve, reject) => {
    SSO_DB.once('open', () => resolve());
    SSO_DB.once('error', err => reject(err));
  });

  const queryResult = await callback(SSO_DB);

  await SSO_DB.close();
  return queryResult;
}