All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.yamcs.yarch.SqlBuilder Maven / Gradle / Ivy

Go to download

There is a newer version: 5.10.7
Show newest version
package org.yamcs.yarch;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.yamcs.utils.TimeEncoding;

import com.google.protobuf.Timestamp;

public class SqlBuilder {

    private String table;
    private List selectExpressions = new ArrayList<>();
    private List conditions = new ArrayList<>(2);
    private Boolean descend;
    private Long offset;
    private Long limit;

    private List queryArgs = new ArrayList<>();

    public SqlBuilder(String table) {
        this.table = table;
    }

    /**
     * Additive! Calling multiple times will add extra select expressions to the already specified list.
     */
    public SqlBuilder select(String... exprs) {
        for (String expr : exprs) {
            selectExpressions.add(expr);
        }
        return this;
    }

    /**
     * Additive! Calling multiple times will add extra conditions to the already specified list.
     */
    public SqlBuilder where(String whereCondition, Object... args) {
        conditions.add(whereCondition);
        for (Object o : args) {
            queryArgs.add(o);
        }
        return this;
    }

    public SqlBuilder whereColAfter(String colName, Timestamp timestamp) {
        return whereColAfter(colName, TimeEncoding.fromProtobufTimestamp(timestamp), true);
    }

    public SqlBuilder whereColAfter(String colName, long instant) {
        return whereColAfter(colName, instant, true);
    }

    public SqlBuilder whereColAfterOrEqual(String colName, long instant) {
        return whereColAfter(colName, instant, false);
    }

    public SqlBuilder whereColAfterOrEqual(String colName, Timestamp timestamp) {
        return whereColAfter(colName, TimeEncoding.fromProtobufTimestamp(timestamp), false);
    }

    private SqlBuilder whereColAfter(String colName, long instant, boolean strict) {
        StringBuilder cond = new StringBuilder();
        cond.append(colName);
        cond.append(strict ? " > " : " >= ");
        cond.append(instant);
        conditions.add(cond.toString());
        return this;
    }

    public SqlBuilder whereColBeforeOrEqual(String colName, Timestamp timestamp) {
        return whereColBefore(colName, TimeEncoding.fromProtobufTimestamp(timestamp), false);
    }

    public SqlBuilder whereColBeforeOrEqual(String colName, long instant) {
        return whereColBefore(colName, instant, false);
    }

    public SqlBuilder whereColBefore(String colName, Timestamp timestamp) {
        return whereColBefore(colName, TimeEncoding.fromProtobufTimestamp(timestamp), true);
    }

    public SqlBuilder whereColBefore(String colName, long instant) {
        return whereColBefore(colName, instant, true);
    }

    private SqlBuilder whereColBefore(String colName, long instant, boolean strict) {
        StringBuilder cond = new StringBuilder();
        cond.append(colName);
        cond.append(strict ? " < " : " <= ");
        cond.append(instant);
        conditions.add(cond.toString());
        return this;
    }

    public SqlBuilder whereColIn(String colName, Collection values) {
        return whereColInNotIn(colName, values, true);
    }

    public SqlBuilder whereColNotIn(String colName, Collection values) {
        return whereColInNotIn(colName, values, false);
    }

    private SqlBuilder whereColInNotIn(String colName, Collection values, boolean in) {
        StringBuilder cond = new StringBuilder();
        cond.append(colName);
        if (!in) {
            cond.append(" NOT");
        }
        cond.append(" IN (");
        boolean first = true;
        for (Object o : values) {
            if (first) {
                first = false;
            } else {
                cond.append(", ");
            }
            cond.append("?");
            queryArgs.add(o);
        }
        cond.append(")");
        conditions.add(cond.toString());

        return this;
    }

    public SqlBuilder descend(boolean descend) {
        this.descend = descend;
        return this;
    }

    public SqlBuilder limit(long limit) {
        this.limit = limit;
        return this;
    }

    public SqlBuilder limit(long offset, long limit) {
        this.offset = offset;
        this.limit = limit;
        return this;
    }

    public List getQueryArguments() {
        return queryArgs;
    }

    public Object[] getQueryArgumentsArray() {
        return queryArgs.toArray();
    }

    @Override
    public String toString() {
        StringBuilder buf = new StringBuilder("select ");
        if (selectExpressions.isEmpty()) {
            buf.append("*");
        } else {
            boolean first = true;
            for (String expr : selectExpressions) {
                if (!first) {
                    buf.append(", ");
                }
                buf.append(expr);
                first = false;
            }
        }
        buf.append(" from ").append(table);
        if (!conditions.isEmpty()) {
            buf.append(" where ");
            boolean first = true;
            for (String condition : conditions) {
                if (!first) {
                    buf.append(" and ");
                } else {
                    first = false;
                }
                buf.append(condition);
            }
        }
        if (descend != null) {
            buf.append(descend ? " order desc" : " order asc");
        }
        if (limit != null) {
            if (offset != null) {
                buf.append(" limit ").append(offset).append(",").append(limit);
            } else {
                buf.append(" limit ").append(limit);
            }
        }
        return buf.toString();
    }
}
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy