de.mhus.lib.sql.parser.FunctionPart Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2018 Mike Hummel
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
 * except in compliance with the License. You may obtain a copy of the License at
 *
 * 
http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the
 * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied. See the License for the specific language governing permissions and
 * limitations under the License.
 */
package de.mhus.lib.sql.parser;

import java.io.IOException;
import java.util.LinkedList;
import java.util.Map;

import de.mhus.lib.core.MString;
import de.mhus.lib.core.parser.ParseException;
import de.mhus.lib.core.parser.ParseReader;
import de.mhus.lib.core.parser.ParsingPart;
import de.mhus.lib.errors.MException;
import de.mhus.lib.errors.MRuntimeException;

public class FunctionPart implements ParsingPart {

    private String name;
    private LinkedList parts = new LinkedList();
    private ICompiler compiler;

    public FunctionPart(ICompiler compiler, String name) {
        this.compiler = compiler;
        this.name = name;
    }

    @Override
    public void execute(StringBuilder out, Map attributes) {
        try {
            out.append(name).append("(");
            boolean first = true;
            for (ParsingPart p : parts) {
                if (!first) out.append(",");
                p.execute(out, attributes);
                first = false;
            }
            out.append(")");
        } catch (MException e) {
            throw new MRuntimeException(e);
        }
    }

    @Override
    public void dump(int level, StringBuilder out) {
        MString.appendRepeating(level, ' ', out);
        out.append(getClass().getCanonicalName())
                .append(" ")
                .append(name)
                .append(" (")
                .append("\n");
        for (ParsingPart p : parts) {
            p.dump(level + 1, out);
        }
        MString.appendRepeating(level, ' ', out);
        out.append(")").append("\n");
    }

    @Override
    public void parse(ParseReader str) throws ParseException {
        try {
            while (true) {
                removeSpaces(str);
                MainPart pp = new MainPart(compiler);
                pp.setStopOnComma(true);
                pp.parse(str);
                add(pp);
                if (str.character() == ')') {
                    str.consume();
                    return;
                }
                if (str.character() == ',') {
                    str.consume();
                }
            }
        } catch (IOException e) {
            throw new ParseException(e);
        }
    }

    private void removeSpaces(ParseReader str) throws IOException {
        do {
            if (str.isClosed()) return;
            char c = str.character();
            if (c != ' ' && c != '\n' && c != '\n' && c != '\r') return;
            str.consume();
        } while (true);
    }

    public void add(ParsingPart pp) {
        parts.add(pp);
    }
}