[数据模型] 新增了 SessionUtil 中 删除 根据表名删除metadataSources 中的表 的方法

src/main/java/com/tykj/datawarehouse/model/utils/SessionUtil.java

@@ -7,6 +7,9 @@ import org.hibernate.Session;
 import org.hibernate.SessionFactory;
 import org.hibernate.boot.Metadata;
 import org.hibernate.boot.MetadataSources;
+import org.hibernate.boot.jaxb.hbm.spi.JaxbHbmHibernateMapping;
+import org.hibernate.boot.jaxb.hbm.spi.JaxbHbmRootEntityType;
+import org.hibernate.boot.jaxb.spi.Binding;
 import org.hibernate.boot.registry.StandardServiceRegistry;
 import org.hibernate.tool.hbm2ddl.SchemaUpdate;
 import org.hibernate.tool.schema.TargetType;
@@ -76,5 +79,31 @@ public class SessionUtil {
     public Metadata getMetadata(){
         return metadataSources.buildMetadata();
     }
+
+    public MetadataSources getMetadataSources() {
+        return metadataSources;
+    }
+
+    public void deleteTableInMeteData(String tableName){
+        Integer needDelete = -1;
+        List<Binding> xmlBindings = metadataSources.getXmlBindings();
+        for (int i = 0; i < xmlBindings.size(); i++) {
+            Object root = xmlBindings.get(i).getRoot();
+            if (root instanceof JaxbHbmHibernateMapping){
+                List<JaxbHbmRootEntityType> listEntity = ((JaxbHbmHibernateMapping) root).getClazz();
+                for (JaxbHbmRootEntityType jaxbHbmRootEntityType : listEntity) {
+                    String currentTableName = jaxbHbmRootEntityType.getTable();
+                    if (currentTableName.equals(tableName)){
+                        needDelete = i;
+                        break;
+                    }
+                }
+
+            }
+        }
+        if (needDelete>-1){
+            xmlBindings.remove(needDelete);
+        }
+    }
 }