forked from plusone/plusone-commons
重构 SQL 构建器。
parent
58e085277b
commit
15ed8c908e
|
@ -8,6 +8,10 @@ public class JdbcSql extends SQL<JdbcSql> {
|
||||||
super();
|
super();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static JdbcSql newSql() {
|
||||||
|
return new JdbcSql();
|
||||||
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public JdbcSql getSelf() {
|
public JdbcSql getSelf() {
|
||||||
return this;
|
return this;
|
||||||
|
|
|
@ -0,0 +1,65 @@
|
||||||
|
package xyz.zhouxy.plusone.commons.jdbc;
|
||||||
|
|
||||||
|
import com.google.common.annotations.Beta;
|
||||||
|
|
||||||
|
@Beta
|
||||||
|
public class MyBatisSql extends SQL<MyBatisSql> {
|
||||||
|
|
||||||
|
private final boolean withScript;
|
||||||
|
|
||||||
|
MyBatisSql(boolean withScript) {
|
||||||
|
super();
|
||||||
|
this.withScript = withScript;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static MyBatisSql newSql() {
|
||||||
|
return new MyBatisSql(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static MyBatisSql newScriptSql() {
|
||||||
|
return new MyBatisSql(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MyBatisSql getSelf() {
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static String IN(String col, String paramName) {
|
||||||
|
return new StringBuilder(" ")
|
||||||
|
.append(col)
|
||||||
|
.append(" IN")
|
||||||
|
.append(buildQuestionsList(col, paramName))
|
||||||
|
.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
public static String NOT_IN(String col, String paramName) {
|
||||||
|
return new StringBuilder()
|
||||||
|
.append(col)
|
||||||
|
.append(" NOT IN")
|
||||||
|
.append(buildQuestionsList(col, paramName))
|
||||||
|
.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
private static String buildQuestionsList(String col, String paramName) {
|
||||||
|
return new StringBuilder()
|
||||||
|
.append("<foreach item=\"")
|
||||||
|
.append(col)
|
||||||
|
.append("\" index=\"index\" collection=\"")
|
||||||
|
.append(paramName)
|
||||||
|
.append("\" open=\"(\" separator=\",\" close=\")\">")
|
||||||
|
.append("#{")
|
||||||
|
.append(col)
|
||||||
|
.append("}</foreach>")
|
||||||
|
.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString() {
|
||||||
|
String str = super.toString();
|
||||||
|
if (withScript) {
|
||||||
|
str = "<script>\n" + str + "\n</script>";
|
||||||
|
}
|
||||||
|
return str;
|
||||||
|
}
|
||||||
|
}
|
|
@ -30,10 +30,18 @@ public abstract class SQL<T> extends AbstractSQL<T> {
|
||||||
return new JdbcSql();
|
return new JdbcSql();
|
||||||
}
|
}
|
||||||
|
|
||||||
public T WHERE(boolean condition, String sqlConditions) {
|
public static MyBatisSql newMyBatisSql(boolean withScript) {
|
||||||
|
return new MyBatisSql(withScript);
|
||||||
|
}
|
||||||
|
|
||||||
|
public T WHERE(boolean condition, String sqlCondition) {
|
||||||
if (condition) {
|
if (condition) {
|
||||||
return WHERE(sqlConditions);
|
return WHERE(sqlCondition);
|
||||||
}
|
}
|
||||||
return getSelf();
|
return getSelf();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public T WHERE(boolean condition, String ifSqlCondition, String elseSqlCondition) {
|
||||||
|
return WHERE(condition ? ifSqlCondition : elseSqlCondition);
|
||||||
|
}
|
||||||
}
|
}
|
|
@ -1,29 +0,0 @@
|
||||||
package xyz.zhouxy.plusone.commons.util;
|
|
||||||
|
|
||||||
import java.util.Arrays;
|
|
||||||
|
|
||||||
import com.google.common.annotations.Beta;
|
|
||||||
|
|
||||||
@Beta
|
|
||||||
public class MoreArrays {
|
|
||||||
|
|
||||||
public static Object[] asObjectArray(final Object... objects) {
|
|
||||||
return objects;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static char[] newCharArrayWith(int length, char c) {
|
|
||||||
char[] arr = new char[length];
|
|
||||||
Arrays.fill(arr, c);
|
|
||||||
return arr;
|
|
||||||
}
|
|
||||||
|
|
||||||
public static String[] newStringArrayWith(int length, String c) {
|
|
||||||
String[] arr = new String[length];
|
|
||||||
Arrays.fill(arr, c);
|
|
||||||
return arr;
|
|
||||||
}
|
|
||||||
|
|
||||||
private MoreArrays() {
|
|
||||||
throw new IllegalStateException("Utility class");
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,20 +0,0 @@
|
||||||
package xyz.zhouxy.plusone.commons.util;
|
|
||||||
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.Date;
|
|
||||||
|
|
||||||
import org.junit.jupiter.api.Test;
|
|
||||||
import org.slf4j.Logger;
|
|
||||||
import org.slf4j.LoggerFactory;
|
|
||||||
|
|
||||||
class MoreArraysTests {
|
|
||||||
|
|
||||||
private static final Logger log = LoggerFactory.getLogger(MoreArraysTests.class);
|
|
||||||
|
|
||||||
@Test
|
|
||||||
void testAsObjectArray() {
|
|
||||||
Object[] arr = MoreArrays.asObjectArray("1", "2", 1, 2, new Date());
|
|
||||||
log.info("arr: {}", arr.toString());
|
|
||||||
log.info("arr: {}", Arrays.toString(arr));
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -0,0 +1,23 @@
|
||||||
|
package xyz.zhouxy.plusone.commons.util;
|
||||||
|
|
||||||
|
import static xyz.zhouxy.plusone.commons.jdbc.MyBatisSql.IN;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
|
import xyz.zhouxy.plusone.commons.jdbc.MyBatisSql;
|
||||||
|
|
||||||
|
class MyBatisSqlBuilderTests {
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(MyBatisSqlBuilderTests.class);
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void test() {
|
||||||
|
// List<String> ids = Arrays.asList("2333", "4501477");
|
||||||
|
MyBatisSql sql = MyBatisSql.newScriptSql()
|
||||||
|
.SELECT("*")
|
||||||
|
.FROM("test_table")
|
||||||
|
.WHERE(IN("id", "ids"));
|
||||||
|
log.info("sql: {}", sql);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue