Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.yamcs.yarch.SqlBuilder Maven / Gradle / Ivy
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();
}
}
