From 957b711cd42ec3eeeb701e45beb79c75c0ef190b Mon Sep 17 00:00:00 2001
From: bangbang93 <bangbang93@163.com>
Date: Fri, 17 Jul 2020 18:31:26 +0800
Subject: [PATCH] add getModelName

---
 src/index.ts | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 4077941..eaa3472 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -31,15 +31,20 @@ export function getSchema<T extends IMongooseClass>(modelClass: T): Schema {
   return schema
 }
 
-export function getModel<T extends IMongooseClass>(modelClass: T): T {
+export function getModel<T extends IMongooseClass>(modelClass: T): ModelType<T> {
   if (modelCache.has(modelClass)) {
-    return modelCache.get(modelClass) as any
+    return modelCache.get(modelClass) as ModelType<T>
   }
   const meta = getMongooseMeta(modelClass.prototype)
   if (!meta.name) throw new Error(`name not set for model ${modelClass.constructor.name}`)
-  const newModel = model(meta.name, getSchema(modelClass))
-  modelCache.set(modelClass, newModel as any)
-  return newModel as any
+  const newModel: ModelType<T> = model(meta.name, getSchema(modelClass))
+  modelCache.set(modelClass, newModel)
+  return newModel
+}
+
+export function getModelName<T extends IMongooseClass>(modelClass: T): string {
+  const meta = getMongooseMeta(modelClass.prototype)
+  return meta.name
 }
 
 function buildSchema(meta: MongooseMeta): Schema {