mirrors/APIJSON
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add:ORM's support for ClickHouse

Browse Source
This commit is contained in:
chenyanlann 2021-08-01 20:11:14 +08:00
parent cd9827f18d
commit 9b5f3f2597
2 changed files with 22 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java
View File

@ -124,6 +124,7 @@ public abstract class AbstractSQLConfig implements SQLConfig {
DATABASE_LIST.add(DATABASE_SQLSERVER); DATABASE_LIST.add(DATABASE_SQLSERVER);
DATABASE_LIST.add(DATABASE_ORACLE); DATABASE_LIST.add(DATABASE_ORACLE);
DATABASE_LIST.add(DATABASE_DB2); DATABASE_LIST.add(DATABASE_DB2);
DATABASE_LIST.add(DATABASE_CLICKHOUSE);
RAW_MAP = new LinkedHashMap<>(); // 保证顺序，避免配置冲突等意外情况 RAW_MAP = new LinkedHashMap<>(); // 保证顺序，避免配置冲突等意外情况
@ -508,10 +509,17 @@ public abstract class AbstractSQLConfig implements SQLConfig {
public static boolean isDb2(String db) { public static boolean isDb2(String db) {
return DATABASE_DB2.equals(db); return DATABASE_DB2.equals(db);
} }
@Override
public boolean isClickHouse() {
return isClickHouse(getSQLDatabase());
}
public static boolean isClickHouse(String db) {
return DATABASE_CLICKHOUSE.equals(db);
}
@Override @Override
public String getQuote() { public String getQuote() {
return isMySQL() ? "`" : "\""; return isMySQL() ? "`" : ( isClickHouse()? "" : "\"");
} }
@Override @Override
@ -2158,6 +2166,9 @@ public abstract class AbstractSQLConfig implements SQLConfig {
if (isOracle()) { if (isOracle()) {
return "regexp_like(" + getKey(key) + ", " + getValue(value) + (ignoreCase ? ", 'i'" : ", 'c'") + ")"; return "regexp_like(" + getKey(key) + ", " + getValue(value) + (ignoreCase ? ", 'i'" : ", 'c'") + ")";
} }
if (isClickHouse()) {
return "match(" + (ignoreCase ? "lower(" : "") + getKey(key) + (ignoreCase ? ")" : "") + ", " + (ignoreCase ? "lower(" : "") + getValue(value) + (ignoreCase ? ")" : "") + ")";
}
return getKey(key) + " REGEXP " + (ignoreCase ? "" : "BINARY ") + getValue(value); return getKey(key) + " REGEXP " + (ignoreCase ? "" : "BINARY ") + getValue(value);
} }
//~ regexp >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> //~ regexp >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
@ -2448,7 +2459,12 @@ public abstract class AbstractSQLConfig implements SQLConfig {
else { else {
boolean isNum = c instanceof Number; boolean isNum = c instanceof Number;
String v = (isNum ? "" : "\"") + childs[i] + (isNum ? "" : "\""); String v = (isNum ? "" : "\"") + childs[i] + (isNum ? "" : "\"");
condition += ("json_contains(" + getKey(key) + ", " + getValue(v) + ")"); if (isClickHouse()) {
condition += condition + "has(JSONExtractArrayRaw(assumeNotNull(" + getKey(key) + "))" + ", " + getValue(v) + ")";
}
else {
condition += ("json_contains(" + getKey(key) + ", " + getValue(v) + ")");
}
} }
} }
} }
@ -2649,9 +2665,9 @@ public abstract class AbstractSQLConfig implements SQLConfig {
case POST: case POST:
return "INSERT INTO " + tablePath + config.getColumnString() + " VALUES" + config.getValuesString(); return "INSERT INTO " + tablePath + config.getColumnString() + " VALUES" + config.getValuesString();
case PUT: case PUT:
return "UPDATE " + tablePath + config.getSetString() + config.getWhereString(true) + (config.isMySQL() ? config.getLimitString() : ""); return "UPDATE " + tablePath + config.getSetString() + config.getWhereString(true) + (config.isMySQL()||config.isClickHouse() ? config.getLimitString() : "");
case DELETE: case DELETE:
return "DELETE FROM " + tablePath + config.getWhereString(true) + (config.isMySQL() ? config.getLimitString() : ""); // PostgreSQL 不允许 LIMIT return "DELETE FROM " + tablePath + config.getWhereString(true) + (config.isMySQL()||config.isClickHouse() ? config.getLimitString() : ""); // PostgreSQL 不允许 LIMIT
default: default:
String explain = (config.isExplain() ? (config.isSQLServer() || config.isOracle() ? "SET STATISTICS PROFILE ON " : "EXPLAIN ") : ""); String explain = (config.isExplain() ? (config.isSQLServer() || config.isOracle() ? "SET STATISTICS PROFILE ON " : "EXPLAIN ") : "");
if (config.isTest() && RequestMethod.isGetMethod(config.getMethod(), true)) { if (config.isTest() && RequestMethod.isGetMethod(config.getMethod(), true)) {

2
APIJSONORM/src/main/java/apijson/orm/SQLConfig.java
View File

@ -22,6 +22,7 @@ public interface SQLConfig {
String DATABASE_SQLSERVER = "SQLSERVER"; String DATABASE_SQLSERVER = "SQLSERVER";
String DATABASE_ORACLE = "ORACLE"; String DATABASE_ORACLE = "ORACLE";
String DATABASE_DB2 = "DB2"; String DATABASE_DB2 = "DB2";
String DATABASE_CLICKHOUSE = "CLICKHOUSE";
String SCHEMA_INFORMATION = "information_schema"; //MySQL, PostgreSQL, SQL Server 都有的系统模式 String SCHEMA_INFORMATION = "information_schema"; //MySQL, PostgreSQL, SQL Server 都有的系统模式
String SCHEMA_SYS = "sys"; //SQL Server 系统模式 String SCHEMA_SYS = "sys"; //SQL Server 系统模式
@ -37,6 +38,7 @@ public interface SQLConfig {
boolean isSQLServer(); boolean isSQLServer();
boolean isOracle(); boolean isOracle();
boolean isDb2(); boolean isDb2();
boolean isClickHouse();
//暂时只兼容以上 5 种 //暂时只兼容以上 5 种
// boolean isSQL(); // boolean isSQL();
// boolean isTSQL(); // boolean isTSQL();