diff --git a/src/util/Database.ts b/src/util/Database.ts index e17eeccb..5d9afab9 100644 --- a/src/util/Database.ts +++ b/src/util/Database.ts @@ -1,5 +1,5 @@ import "./MongoBigInt"; -import mongoose, { Collection, Connection } from "mongoose"; +import mongoose, { Collection, Connection, LeanDocument } from "mongoose"; import { ChangeStream, ChangeEvent, Long } from "mongodb"; import EventEmitter from "events"; import { Document } from "mongoose"; @@ -11,11 +11,13 @@ const connection = mongoose.createConnection(uri, { autoIndex: true, useNewUrlPa export default connection; -function transform(document: T) { +function transform(document: T) { + // @ts-ignore return document.toObject({ virtuals: true }); } -export function toObject(document: T | T[]) { +export function toObject(document: T): LeanDocument { + // @ts-ignore return Array.isArray(document) ? document.map((x) => transform(x)) : transform(document); }