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.babyfish.jimmer.sql.runtime.SqlBuilder Maven / Gradle / Ivy
package org.babyfish.jimmer.sql.runtime;
import org.babyfish.jimmer.meta.ImmutableProp;
import org.babyfish.jimmer.meta.ImmutableType;
import org.babyfish.jimmer.sql.SqlClient;
import org.babyfish.jimmer.sql.ast.table.Table;
import org.babyfish.jimmer.sql.ast.tuple.*;
import java.util.*;
import java.util.function.Function;
public class SqlBuilder {
private SqlClient sqlClient;
private Set> usedTables;
private SqlBuilder parent;
private int childBuilderCount;
private StringBuilder builder = new StringBuilder();
private List variables = new ArrayList<>();
private boolean terminated;
public SqlBuilder(SqlClient sqlClient) {
this.sqlClient = sqlClient;
this.usedTables = new HashSet<>();
}
private SqlBuilder(SqlBuilder parent) {
this.sqlClient = parent.sqlClient;
this.usedTables = parent.usedTables;
this.parent = parent;
parent.childBuilderCount++;
}
public void useTable(Table table) {
if (table != null) {
usedTables.add(table);
}
}
public boolean isTableUsed(Table table) {
return usedTables.contains(table);
}
public SqlBuilder sql(String sql) {
builder.append(sql);
return this;
}
public SqlBuilder variable(Class type, T value) {
if (value != null) {
return variable(value);
}
return nullVariable(type);
}
public SqlBuilder variable(Object value) {
validate();
if (value instanceof Tuple2) {
Tuple2 tuple = (Tuple2)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(")");
} else if (value instanceof Tuple3) {
Tuple3 tuple = (Tuple3)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(")");
} else if (value instanceof Tuple4) {
Tuple4 tuple = (Tuple4)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._4(), "tuple._4 cannot be null"))
.sql(")");
} else if (value instanceof Tuple5) {
Tuple5 tuple = (Tuple5)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._4(), "tuple._4 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._5(), "tuple._5 cannot be null"))
.sql(")");
} else if (value instanceof Tuple6) {
Tuple6 tuple = (Tuple6)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._4(), "tuple._4 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._5(), "tuple._5 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._6(), "tuple._6 cannot be null"))
.sql(")");
} else if (value instanceof Tuple7) {
Tuple7 tuple = (Tuple7)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._4(), "tuple._4 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._5(), "tuple._5 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._6(), "tuple._6 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._7(), "tuple._7 cannot be null"))
.sql(")");
} else if (value instanceof Tuple8) {
Tuple8 tuple = (Tuple8)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._4(), "tuple._4 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._5(), "tuple._5 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._6(), "tuple._6 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._7(), "tuple._7 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._8(), "tuple._8 cannot be null"))
.sql(")");
} else if (value instanceof Tuple9) {
Tuple9 tuple = (Tuple9)value;
this
.sql("(")
.singleVariable(Objects.requireNonNull(tuple._1(), "tuple._1 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._2(), "tuple._2 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._3(), "tuple._3 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._4(), "tuple._4 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._5(), "tuple._5 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._6(), "tuple._6 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._7(), "tuple._7 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._8(), "tuple._8 cannot be null"))
.sql(", ")
.singleVariable(Objects.requireNonNull(tuple._9(), "tuple._9 cannot be null"))
.sql(")");
} else {
singleVariable(value);
}
return this;
}
@SuppressWarnings("unchecked")
private SqlBuilder singleVariable(Object value) {
if (value instanceof DbNull) {
throw new ExecutionException(
"Cannot add variable whose type is " + DbNull.class.getName()
);
}
ScalarProvider scalarProvider =
sqlClient.getScalarProvider((Class)value.getClass());
Object finalValue;
if (scalarProvider != null) {
finalValue = scalarProvider.toSql(value);
} else {
finalValue = value;
}
builder.append('?');
variables.add(finalValue);
return this;
}
public SqlBuilder nullVariable(ImmutableProp prop) {
validate();
ImmutableType targetType = prop.getTargetType();
if (targetType == null) {
return nullVariable(prop.getElementClass());
}
return nullVariable(targetType.getIdProp().getElementClass());
}
@SuppressWarnings("unchecked")
public SqlBuilder nullVariable(Class type) {
validate();
ScalarProvider scalarProvider =
sqlClient.getScalarProvider((Class)type);
Object finalValue;
if (scalarProvider != null) {
finalValue = new DbNull(scalarProvider.getSqlType());
} else {
finalValue = new DbNull(type);
}
builder.append('?');
variables.add(finalValue);
return this;
}
public SqlBuilder createChildBuilder() {
return new SqlBuilder(this);
}
public Tuple2> build() {
return build(null);
}
public Tuple2> build(
Function>, Tuple2>> transformer
) {
validate();
Tuple2> result = new Tuple2<>(builder.toString(), variables);
if (transformer != null) {
result = transformer.apply(result);
}
SqlBuilder p = this.parent;
if (p != null) {
p.builder.append(result._1());
p.variables.addAll(result._2());
while (p != null) {
--p.childBuilderCount;
p = p.parent;
}
}
terminated = true;
return result;
}
private void validate() {
if (childBuilderCount != 0) {
throw new IllegalStateException(
"Internal bug: Cannot change sqlbuilder because there are some child builders"
);
}
if (terminated) {
throw new IllegalStateException(
"Internal bug: Current build has been terminated"
);
}
}
}
