From f10ade87951ba0985e58064b8b16fa9fc2c8345e Mon Sep 17 00:00:00 2001
From: redkale <redkale@qq.com>
Date: Wed, 14 Aug 2024 20:04:56 +0800
Subject: [PATCH] AbstractDataSource

---
 .../java/org/redkale/source/AbstractDataSource.java   | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/main/java/org/redkale/source/AbstractDataSource.java b/src/main/java/org/redkale/source/AbstractDataSource.java
index 50935fa7f..b282332a7 100644
--- a/src/main/java/org/redkale/source/AbstractDataSource.java
+++ b/src/main/java/org/redkale/source/AbstractDataSource.java
@@ -249,6 +249,17 @@ public abstract class AbstractDataSource extends AbstractService implements Data
             BiFunction<DataSource, EntityInfo, CompletableFuture<List>> fullloader) {
         return EntityInfo.load(clazz, cacheForbidden, props, this, fullloader);
     }
+    
+    /**
+     * 加载指定类的EntityInfo
+     *
+     * @param <T> 泛型
+     * @param clazz 类
+     * @return EntityInfo
+     */
+    protected <T> EntityInfo<T> getEntityInfo(Class<T> clazz) {
+        return EntityInfo.get(clazz);
+    }
 
     /**
      * 检查对象是否都是同一个Entity类