com.github.mengweijin.flyway.database.gbase8s.Gbase8sParser Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of db-migration Show documentation

Flyway、Liquibase 扩展支持达梦（DM）数据库、南大通用（GBase 8s）数据库。

There is a newer version: 2.0.2

package com.github.mengweijin.flyway.database.gbase8s;

import org.flywaydb.core.api.configuration.Configuration;
import org.flywaydb.core.internal.parser.Parser;
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.io.IOException;
import java.util.List;

/**
 * @author mengweijin
 */
public class Gbase8sParser extends Parser {
    public Gbase8sParser(Configuration configuration, ParsingContext parsingContext) {
        super(configuration, parsingContext, 2);
    }

    @Override
    protected void adjustBlockDepth(ParserContext context, List tokens, Token keyword, PeekingReader reader) throws IOException {
        int lastKeywordIndex = getLastKeywordIndex(tokens);
        if (lastKeywordIndex < 0) {
            return;
        }

        String current = keyword.getText();
        if ("FUNCTION".equals(current) || "PROCEDURE".equals(current)) {
            String previous = tokens.get(lastKeywordIndex).getText();

            // CREATE( DBA)? (FUNCTION|PROCEDURE)
            if ("CREATE".equals(previous) || "DBA".equals(previous)) {
                context.increaseBlockDepth(previous);
            } else if ("END".equals(previous)) {
                context.decreaseBlockDepth();
            }
        }
    }
}