diff options
| -rw-r--r-- | src/util/Database.ts | 8 |
1 files changed, 5 insertions, 3 deletions
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>connection; -function transform<T extends Document>(document: T) { +function transform<T>(document: T) { + // @ts-ignore return document.toObject({ virtuals: true }); } -export function toObject<T extends Document>(document: T | T[]) { +export function toObject<T>(document: T): LeanDocument<T> { + // @ts-ignore return Array.isArray(document) ? document.map((x) => transform<T>(x)) : transform(document); } |