com.github.harbby.dsxparser.tree.SubStringExpression Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dsx-parser Show documentation

dsx parser

The newest version!

package com.github.harbby.dsxparser.tree;

import com.github.harbby.dsxparser.Optimizer;

import java.util.List;
import java.util.Objects;

public class SubStringExpression extends Expression {

    private final Expression base;
    private final Expression pos;
    private final Expression len;

    private SubStringExpression(Expression base, Expression pos, Expression len) {
        this(null, base, pos, len);
    }

    public SubStringExpression(NodeLocation location, Expression base, Expression pos, Expression len) {
        super(location);
        this.base = base;
        this.pos = pos;
        this.len = len;
    }

    public Expression getPos() {
        return pos;
    }

    public Expression getLen() {
        return len;
    }

    public Expression getBase() {
        return base;
    }

    @Override
    public String doGenSql() {
        return String.format("substring(%s,%s,%s)", base.doGenSql(), pos.doGenSql(), len.doGenSql());
    }

    @Override
    public List getChildren() {
        return List.of(base, pos, len);
    }

    @Override
    public int hashCode() {
        return Objects.hash(base, pos, len);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }

        SubStringExpression that = (SubStringExpression) o;
        return Objects.equals(base, that.base) &&
                Objects.equals(pos, that.pos) &&
                Objects.equals(len, that.len);
    }

    @Override
    public Expression visit(Optimizer optimizer) {
        return new SubStringExpression(optimizer.optimize(base), optimizer.optimize(pos), optimizer.optimize(len));
    }
}