Add:ORM's support for Hive

chenyanlann · chenyanlann · commit ed05ae37afdc · 2021-12-31T17:53:06.000+08:00
diff --git a/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java b/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java
@@ -126,6 +126,7 @@ public abstract class AbstractSQLConfig implements SQLConfig {
 		DATABASE_LIST.add(DATABASE_ORACLE);
 		DATABASE_LIST.add(DATABASE_DB2);
 		DATABASE_LIST.add(DATABASE_CLICKHOUSE);
+		DATABASE_LIST.add(DATABASE_HIVE);
 
 
 		RAW_MAP = new LinkedHashMap<>();  // 保证顺序，避免配置冲突等意外情况
@@ -903,6 +904,13 @@ public boolean isClickHouse() {
 	public static boolean isClickHouse(String db) {
 		return DATABASE_CLICKHOUSE.equals(db);
 	}
+	@Override
+	public boolean isHive() {
+		return isHive(getSQLDatabase());
+	}
+	public static boolean isHive(String db) {
+		return DATABASE_HIVE.equals(db);
+	}
 
 	@Override
 	public String getQuote() {
@@ -2727,6 +2735,9 @@ public String getRegExpString(String key, String value, boolean ignoreCase) {
 		if (isClickHouse()) {
 			return "match(" + (ignoreCase ? "lower(" : "") + getKey(key) + (ignoreCase ? ")" : "") + ", " + (ignoreCase ? "lower(" : "") + getValue(value) + (ignoreCase ? ")" : "") + ")";
 		}
+		if (isHive()) {
+			return (ignoreCase ? "lower(" : "") + getKey(key) + (ignoreCase ? ")" : "") + " REGEXP " + (ignoreCase ? "lower(" : "") + getValue(value) + (ignoreCase ? ")" : "");
+		}
 		return getKey(key) + " REGEXP " + (ignoreCase ? "" : "BINARY ") + getValue(value);
 	}
 	//~ regexp >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>
diff --git a/APIJSONORM/src/main/java/apijson/orm/AbstractSQLExecutor.java b/APIJSONORM/src/main/java/apijson/orm/AbstractSQLExecutor.java
@@ -27,6 +27,7 @@
 import java.util.Map.Entry;
 import java.util.Objects;
 import java.util.Set;
+import java.util.regex.Pattern;
 
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
@@ -36,6 +37,7 @@
 import apijson.NotNull;
 import apijson.RequestMethod;
 import apijson.StringUtil;
+import apijson.orm.AbstractSQLConfig;
 
 /**executor for query(read) or update(write) MySQL database
  * @author Lemon
@@ -718,7 +720,15 @@ protected List<JSONObject> onPutTable(@NotNull SQLConfig config, @NotNull Result
 
 	protected String getKey(@NotNull SQLConfig config, @NotNull ResultSet rs, @NotNull ResultSetMetaData rsmd
 			, final int tablePosition, @NotNull JSONObject table, final int columnIndex, Map<String, JSONObject> childMap) throws Exception {
-		return rsmd.getColumnLabel(columnIndex);  // dotIndex < 0 ? lable : lable.substring(dotIndex + 1);
+		String key = rsmd.getColumnLabel(columnIndex);// dotIndex < 0 ? lable : lable.substring(dotIndex + 1);
+		if (config.isHive()) {
+			String table_name = config.getTable();
+			if(AbstractSQLConfig.TABLE_KEY_MAP.containsKey(table_name)) table_name = AbstractSQLConfig.TABLE_KEY_MAP.get(table_name);
+			String pattern = "^" + table_name + "\\." + "[a-zA-Z]+$";
+			boolean isMatch = Pattern.matches(pattern, key);
+			if(isMatch) key = key.split("\\.")[1];
+		}
+		return key;
 	}
 
 	protected Object getValue(@NotNull SQLConfig config, @NotNull ResultSet rs, @NotNull ResultSetMetaData rsmd
@@ -974,6 +984,7 @@ public ResultSet executeQuery(@NotNull SQLConfig config) throws Exception {
 	public int executeUpdate(@NotNull SQLConfig config) throws Exception {
 		PreparedStatement s = getStatement(config);
 		int count = s.executeUpdate(); //PreparedStatement 不用传 SQL
+		if (config.isHive() && count==0) count = 1;
 
 		if (config.getMethod() == RequestMethod.POST && config.getId() == null) { //自增id
 			ResultSet rs = s.getGeneratedKeys();
diff --git a/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java b/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java
@@ -22,6 +22,7 @@ public interface SQLConfig {
 	String DATABASE_ORACLE = "ORACLE";
 	String DATABASE_DB2 = "DB2";
 	String DATABASE_CLICKHOUSE = "CLICKHOUSE";
+	String DATABASE_HIVE = "HIVE";
 
 	String SCHEMA_INFORMATION = "information_schema";  //MySQL, PostgreSQL, SQL Server 都有的系统模式
 	String SCHEMA_SYS = "sys";  //SQL Server 系统模式
@@ -38,6 +39,7 @@ public interface SQLConfig {
 	boolean isOracle();
 	boolean isDb2();
 	boolean isClickHouse();
+	boolean isHive();
 	//暂时只兼容以上 5 种
 	//	boolean isSQL();
 	//	boolean isTSQL();
