com.github.mengweijin.flyway.database.dm.DmParser Maven / Gradle / Ivy

Go to download
package com.github.mengweijin.flyway.database.dm;

import com.github.mengweijin.flyway.database.oracle.OracleParser;
import org.flywaydb.core.api.configuration.Configuration;
import org.flywaydb.core.internal.parser.ParserContext;
import org.flywaydb.core.internal.parser.ParsingContext;
import org.flywaydb.core.internal.parser.PeekingReader;
import org.flywaydb.core.internal.parser.Token;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @author mengweijin
 */
public class DmParser extends OracleParser {

    private static final String IF = "IF";
    private static final String EXISTS = "EXISTS";

    private static final String CREATE_OR_DROP_IF_OR_NOT = "(CREATE|DROP)\\s(TABLE|PROCEDURE|FUNCTION)\\sIF(\\sNOT)?";


    public DmParser(Configuration configuration, ParsingContext parsingContext) {
        super(configuration, parsingContext);
    }

    /**
     * 调整块深度

     *
     * @param context 参数
     * @param tokens  参数
     * @param keyword 参数
     * @param reader  参数
     * @author 作者: Jack Chu E-mail: [email protected]
     * @date 创建时间：2023/12/13 20:52
     */
    @Override
    protected void adjustBlockDepth(ParserContext context, List tokens, Token keyword, PeekingReader reader) {
        String keywordText = keyword.getText();
        if (EXISTS.equals(keywordText) && context.getBlockDepth() > 0) {
            String sql = makeSql(tokens);
            if (sql.matches(CREATE_OR_DROP_IF_OR_NOT)) {
                context.decreaseBlockDepth();
                return;
            }
        }
        super.adjustBlockDepth(context, tokens, keyword, reader);
    }

    private String makeSql(List tokens) {
        return tokens
                .stream()
                .map(Token::getText)
                .collect(Collectors.joining(" "));
    }
}