Hibernate6根据xml获取ddl sql语句

java 复制代码
package com.kongjs.dbmo.jdbc;

import org.hibernate.boot.Metadata;
import org.hibernate.boot.MetadataSources;
import org.hibernate.boot.registry.StandardServiceRegistry;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.engine.config.spi.ConfigurationService;
import org.hibernate.tool.schema.SourceType;
import org.hibernate.tool.schema.TargetType;
import org.hibernate.tool.schema.internal.ExceptionHandlerHaltImpl;
import org.hibernate.tool.schema.spi.*;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.util.StringUtils;

import java.io.*;
import java.util.*;

public class Hibernate6DDL {

    public static String getDDLSql(String dialect, File dir) {
        StringBuilder sb = new StringBuilder();
        StandardServiceRegistry serviceRegistry = new StandardServiceRegistryBuilder()
                .applySetting("hibernate.dialect", dialect)
                .applySetting("hibernate.temp.use_jdbc_metadata_defaults", false)
                .build();
        try {
            Metadata metadata = new MetadataSources(serviceRegistry)
                    .addDirectory(dir)
                    .buildMetadata();

            Map<String, Object> config = new HashMap<>(Objects.requireNonNull(serviceRegistry.getService(ConfigurationService.class)).getSettings());
            ExceptionHandler exceptionHandler = ExceptionHandlerHaltImpl.INSTANCE;
            ExecutionOptions executionOptions = SchemaManagementToolCoordinator.buildExecutionOptions(config, exceptionHandler);
            SchemaManagementTool tool = serviceRegistry.getService(SchemaManagementTool.class);

            SourceDescriptor sourceDescriptor = new SourceDescriptor() {
                @Override
                public SourceType getSourceType() {
                    return SourceType.METADATA;
                }

                @Override
                public ScriptSourceInput getScriptSourceInput() {
                    return null;
                }
            };

            TargetDescriptor targetDescriptor = new TargetDescriptor() {
                @Override
                public EnumSet<TargetType> getTargetTypes() {
                    return EnumSet.of(TargetType.SCRIPT);
                }

                @Override
                public ScriptTargetOutput getScriptTargetOutput() {
                    return new ScriptTargetOutput() {
                        @Override
                        public void prepare() {

                        }

                        @Override
                        public void accept(String command) {
                            sb.append(command);
                            sb.append("\n");
                        }

                        @Override
                        public void release() {

                        }
                    };
                }
            };

            Objects.requireNonNull(tool);

            SchemaCreator schemaCreator = tool.getSchemaCreator(config);
            schemaCreator.doCreation(metadata, executionOptions, ContributableMatcher.ALL, sourceDescriptor, targetDescriptor);

            SchemaDropper schemaDropper = tool.getSchemaDropper(config);
            schemaDropper.doDrop(metadata, executionOptions, ContributableMatcher.ALL, sourceDescriptor, targetDescriptor);

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            StandardServiceRegistryBuilder.destroy(serviceRegistry);
        }
        return sb.toString();
    }

    public static final Map<String, String> dialects = new LinkedHashMap<>();

    public static void initDialect(String classpath) {
        try {
            ResourcePatternResolver resourcePatternResolver = new PathMatchingResourcePatternResolver();
            Resource[] resources = resourcePatternResolver.getResources(classpath);
            for (Resource resource : resources) {
                String filename = resource.getFilename();
                if (filename == null) {
                    continue;
                }
                if (filename.contains("Dialect") && !filename.contains("Abstract")) {
                    int indexOf = filename.indexOf("Dialect");
                    String name = filename.substring(0, indexOf);
                    if (!StringUtils.hasLength(name)) {
                        continue;
                    }
                    if (!filename.contains("community")) {
                        dialects.put(name, "org.hibernate.dialect." + name + "Dialect");
                    } else {
                        dialects.put(name, "org.hibernate.community.dialect." + name + "Dialect");
                    }
                }
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    static {
        String path1 = "classpath*:/org/hibernate/dialect/**";
        //initDialect(path1);
        String path2 = "classpath*:/org/hibernate/community/dialect/**";
        //initDialect(path2);
    }

    public static void main(String[] args) throws FileNotFoundException {
        //System.out.println(dialects);
        //System.out.println(getCreatorSql("org.hibernate.dialect.PostgreSQLDialect",
                //new FileInputStream("C:\\Users\\Administrator\\IdeaProjects\\dbmo\\dbmo-start\\src\\test\\resources\\Event.hbm.xml")));
    }
}
相关推荐
夕除4 小时前
redis--IO 多路复用(select + poll)
数据库·sql
wangchunyu1146 小时前
JAVA CompletableFuture异步编程实战
java·开发语言·python
paopaokaka_luck6 小时前
流浪动物救助与收养系统(DeepSeek AI对话、 ECharts多维分析、救助流程衔接、领养与回访衔接、 捐赠确认与公示、志愿者协作)
java·前端·javascript·人工智能·mysql·echarts·推荐算法
Thomas.Sir6 小时前
第9课:SpringBoot4.x Web开发基础:WebMvc自动配置、静态资源、请求参数接收
java·spring boot
程序 代码狂人7 小时前
切片表与分区表的区别(数据开发面试版)
java·大数据·开发语言·数据仓库·算法
2501_944521597 小时前
Deepmerge 跨端三方库仓颉适配全流程
java·python·eclipse·github
lzhdim8 小时前
100条常用SQL语句
数据库·sql·mysql·mybatis
SEO_juper9 小时前
Node.js 重定向链检测工具:301 跳了几跳、有没有成环,一次看清
xml·开发语言·python·node.js·json·bash
南归北隐9 小时前
Spring AI Alibaba Graph框架实现MCP数据库查询实战
java·后端·spring ai·mcp·spring ai mcp
古法老登瞎折腾9 小时前
大数据智能运维实战(九):让 AI 查日志——找到异常,也保留上下文
java·大数据·运维·人工智能·spring ai·mcp