From d0033d78c945ce9e1d4c188ffa2649196f3a005f Mon Sep 17 00:00:00 2001
From: godotg <jaysunxiao@gmail.com>
Date: Wed, 15 May 2024 14:59:32 +0800
Subject: [PATCH] perf[orm]: export MongoDatabase method

---
 orm/src/main/java/com/zfoo/orm/manager/IOrmManager.java | 2 ++
 orm/src/main/java/com/zfoo/orm/manager/OrmManager.java  | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/orm/src/main/java/com/zfoo/orm/manager/IOrmManager.java b/orm/src/main/java/com/zfoo/orm/manager/IOrmManager.java
index 7f9ddcb9..757da10a 100644
--- a/orm/src/main/java/com/zfoo/orm/manager/IOrmManager.java
+++ b/orm/src/main/java/com/zfoo/orm/manager/IOrmManager.java
@@ -15,6 +15,7 @@ package com.zfoo.orm.manager;
 
 import com.mongodb.client.MongoClient;
 import com.mongodb.client.MongoCollection;
+import com.mongodb.client.MongoDatabase;
 import com.zfoo.orm.cache.IEntityCache;
 import com.zfoo.orm.model.IEntity;
 import org.bson.Document;
@@ -37,6 +38,7 @@ public interface IOrmManager {
      * CN:获取MongoClient，通过MongoClient可以获取到其它数据库或者做一些其它的复杂操作
      */
     MongoClient mongoClient();
+    MongoDatabase mongoDatabase();
 
     <E extends IEntity<?>> IEntityCache<?, E> getEntityCaches(Class<E> clazz);
 
diff --git a/orm/src/main/java/com/zfoo/orm/manager/OrmManager.java b/orm/src/main/java/com/zfoo/orm/manager/OrmManager.java
index 44a1b1c8..27da2f98 100644
--- a/orm/src/main/java/com/zfoo/orm/manager/OrmManager.java
+++ b/orm/src/main/java/com/zfoo/orm/manager/OrmManager.java
@@ -231,6 +231,11 @@ public class OrmManager implements IOrmManager {
         return mongoClient;
     }
 
+    @Override
+    public MongoDatabase mongoDatabase() {
+        return mongodbDatabase;
+    }
+
     @Override
     public <E extends IEntity<?>> IEntityCache<?, E> getEntityCaches(Class<E> clazz) {
         var usable = allEntityCachesUsableMap.get(clazz);