新增支持物联网时序数据库 TDengine

TommyLemon · TommyLemon · commit 4886ee923659 · 2022-07-16T04:49:58.000+08:00
diff --git a/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java b/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java
@@ -148,6 +148,7 @@ public abstract class AbstractSQLConfig implements SQLConfig {
 		DATABASE_LIST.add(DATABASE_DB2);
 		DATABASE_LIST.add(DATABASE_CLICKHOUSE);
 		DATABASE_LIST.add(DATABASE_HIVE);
+		DATABASE_LIST.add(DATABASE_TDENGINE);
 
 
 		RAW_MAP = new LinkedHashMap<>();  // 保证顺序，避免配置冲突等意外情况
@@ -726,6 +727,13 @@ public abstract class AbstractSQLConfig implements SQLConfig {
 		SQL_FUNCTION_MAP.put("bitAnd", ""); //bitAnd(a,b)
 		SQL_FUNCTION_MAP.put("bitOr", ""); //bitOr(a,b)
 
+    // PostgreSQL 表结构相关 SQL 函数
+    SQL_FUNCTION_MAP.put("obj_description", "");
+    SQL_FUNCTION_MAP.put("col_description", "");
+
+    // SQLServer 相关 SQL 函数
+    SQL_FUNCTION_MAP.put("datalength", "");
+
 	}
 
   private int[] dbVersionNums = null;
@@ -991,7 +999,7 @@ public static boolean isHive(String db) {
 
 	@Override
 	public String getQuote() {
-		return isMySQL()||isClickHouse() ? "`" : "\"";
+		return isMySQL() || isClickHouse() || isTDengine() ? "`" : "\"";
 	}
 
 	@Override
diff --git a/APIJSONORM/src/main/java/apijson/orm/AbstractSQLExecutor.java b/APIJSONORM/src/main/java/apijson/orm/AbstractSQLExecutor.java
@@ -1114,7 +1114,8 @@ public Connection getConnection(@NotNull SQLConfig config) throws Exception {
 			connectionMap.put(connectionKey, connection);
 		}
 
-		int ti = getTransactionIsolation();
+    // TDengine 驱动内部事务处理方法都是空实现，手动 commit 无效
+		int ti = config.isTDengine() ? Connection.TRANSACTION_NONE : getTransactionIsolation();
 		if (ti != Connection.TRANSACTION_NONE) { //java.sql.SQLException: Transaction isolation level NONE not supported by MySQL
 			begin(ti);
 		}
@@ -1143,7 +1144,7 @@ public void begin(int transactionIsolation) throws SQLException {
 		//		}
 		if (! isIsolationStatusSet) { //只设置一次Isolation等级 PG重复设置事务等级会报错
 			isIsolationStatusSet = true;
-			connection.setTransactionIsolation(transactionIsolation);
+			connection.setTransactionIsolation(transactionIsolation);  // 这句导致 TDengine 驱动报错
 		}
 		connection.setAutoCommit(false); //java.sql.SQLException: Can''t call commit when autocommit=true
 	}
@@ -1212,6 +1213,12 @@ public void close() {
 
 	@Override
 	public ResultSet executeQuery(@NotNull SQLConfig config, String sql) throws Exception {
+    if (config.isTDengine()) {
+      Connection conn = getConnection(config);
+      Statement stt = conn.createStatement();
+      return executeQuery(stt, StringUtil.isEmpty(sql) ? config.getSQL(false) : sql);
+    }
+
 		PreparedStatement stt = getStatement(config, sql);
 		ResultSet rs = stt.executeQuery();  //PreparedStatement 不用传 SQL
 		//		if (config.isExplain() && (config.isSQLServer() || config.isOracle())) {
@@ -1221,10 +1228,20 @@ public ResultSet executeQuery(@NotNull SQLConfig config, String sql) throws Exce
 		return rs;
 	}
 
+
 	@Override
 	public int executeUpdate(@NotNull SQLConfig config, String sql) throws Exception {
-		PreparedStatement stt = getStatement(config);
-		int count = stt.executeUpdate();  // PreparedStatement 不用传 SQL
+    Statement stt;
+    int count;
+    if (config.isTDengine()) {
+      Connection conn = getConnection(config);
+      stt = conn.createStatement();
+      count = stt.executeUpdate(StringUtil.isEmpty(sql) ? config.getSQL(false) : sql);
+    }
+    else {
+      stt = getStatement(config);
+      count = ((PreparedStatement) stt).executeUpdate();  // PreparedStatement 不用传 SQL
+    }
 
 		if (count <= 0 && config.isHive()) {
 			count = 1;
diff --git a/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java b/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java
@@ -24,6 +24,7 @@ public interface SQLConfig {
 	String DATABASE_DB2 = "DB2";
 	String DATABASE_CLICKHOUSE = "CLICKHOUSE";
 	String DATABASE_HIVE = "HIVE";
+	String DATABASE_TDENGINE = "TDENGINE";
 
 	String SCHEMA_INFORMATION = "information_schema";  //MySQL, PostgreSQL, SQL Server 都有的系统模式
 	String SCHEMA_SYS = "sys";  //SQL Server 系统模式
@@ -41,7 +42,15 @@ public interface SQLConfig {
 	boolean isDb2();
 	boolean isClickHouse();
 	boolean isHive();
-	//暂时只兼容以上 5 种
+  default boolean isTDengine() {
+    return isTDengine(getDatabase());
+  }
+
+  static boolean isTDengine(String db) {
+    return DATABASE_TDENGINE.equals(db);
+  }
+
+  //暂时只兼容以上几种
 	//	boolean isSQL();
 	//	boolean isTSQL();
 	//	boolean isPLSQL();
@@ -177,7 +186,7 @@ default int[] getDBVersionNums() {
 	SQLConfig setTable(String table);
 
 	/**数据库里的真实Table名
-	 * 通过 {@link #TABLE_KEY_MAP} 映射
+	 * 通过 {@link AbstractSQLConfig.TABLE_KEY_MAP} 映射
 	 * @return
 	 */
 	String getSQLTable();
@@ -285,4 +294,5 @@ default int[] getDBVersionNums() {
 	SQLConfig setProcedure(String procedure);
 
 
+
 }
