From 55064ee94cf81662e5b1940b0a086f23ac2e9b4b Mon Sep 17 00:00:00 2001
From: TommyLemon <tommylemon@qq.com>
Date: Thu, 4 Jan 2024 23:27:26 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=94=AF=E6=8C=81=20AI=20?=
 =?UTF-8?q?=E5=90=91=E9=87=8F=E6=95=B0=E6=8D=AE=E5=BA=93=20Milvus?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/apijson/orm/AbstractSQLConfig.java   | 37 +++++++++++++++----
 .../src/main/java/apijson/orm/SQLConfig.java  |  5 ++-
 2 files changed, 34 insertions(+), 8 deletions(-)

diff --git a/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java b/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java
index 855e1226..5527e09f 100755
--- a/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java
+++ b/APIJSONORM/src/main/java/apijson/orm/AbstractSQLConfig.java
@@ -195,6 +195,7 @@ public abstract class AbstractSQLConfig<T extends Object> implements SQLConfig<T
 		DATABASE_LIST.add(DATABASE_HIVE);
 		DATABASE_LIST.add(DATABASE_PRESTO);
 		DATABASE_LIST.add(DATABASE_TRINO);
+		DATABASE_LIST.add(DATABASE_MILVUS);
 		DATABASE_LIST.add(DATABASE_INFLUXDB);
 		DATABASE_LIST.add(DATABASE_TDENGINE);
 		DATABASE_LIST.add(DATABASE_REDIS);
@@ -1191,6 +1192,14 @@ public static boolean isPresto(String db) {
 		return DATABASE_PRESTO.equals(db);
 	}
 
+	@Override
+	public boolean isTrino() {
+		return isTrino(getSQLDatabase());
+	}
+	public static boolean isTrino(String db) {
+		return DATABASE_TRINO.equals(db);
+	}
+
 	@Override
 	public boolean isSnowflake() {
 		return isSnowflake(getSQLDatabase());
@@ -1216,11 +1225,11 @@ public static boolean isCassandra(String db) {
 	}
 
 	@Override
-	public boolean isTrino() {
-		return isTrino(getSQLDatabase());
+	public boolean isMilvus() {
+		return isMilvus(getSQLDatabase());
 	}
-	public static boolean isTrino(String db) {
-		return DATABASE_TRINO.equals(db);
+	public static boolean isMilvus(String db) {
+		return DATABASE_MILVUS.equals(db);
 	}
 
 	@Override
@@ -1268,7 +1277,7 @@ public boolean isMQ() {
 		return isMQ(getSQLDatabase());
 	}
 	public static boolean isMQ(String db) {
-		return isKafka(db);
+		return DATABASE_MQ.equals(db) || isKafka(db);
 	}
 
 	@Override
@@ -1276,7 +1285,7 @@ public String getQuote() {
 		if(isElasticsearch()) {
 			return "";
 		}
-		return isMySQL() || isMariaDB() || isTiDB() || isClickHouse() || isTDengine() ? "`" : "\"";
+		return isMySQL() || isMariaDB() || isTiDB() || isClickHouse() || isTDengine() || isMilvus() ? "`" : "\"";
 	}
 
 	public String quote(String s) {
@@ -1290,6 +1299,7 @@ public String getSchema() {
 	}
 
 	@NotNull
+	@Override
 	public String getSQLSchema() {
 		String table = getTable();
 		//强制，避免因为全局默认的 @schema 自动填充进来，导致这几个类的 schema 为 sys 等其它值
@@ -2596,9 +2606,22 @@ public static int getOffset(int page, int count) {
 	 */
 	@JSONField(serialize = false)
 	public String getLimitString() {
-		if (count <= 0 || RequestMethod.isHeadMethod(getMethod(), true)) {
+		int count = getCount();
+
+		if (isMilvus()) {
+			if (count == 0) {
+				Parser<T> parser = getParser();
+				count = parser == null ? AbstractParser.MAX_QUERY_COUNT : parser.getMaxQueryCount();
+			}
+
+			int offset = getOffset(getPage(), count);
+			return " LIMIT " + offset + ", " + count; // 目前 moql-transx 的限制
+		}
+
+		if (count <= 0 || RequestMethod.isHeadMethod(getMethod(), true)) { // TODO HEAD 真的不需要 LIMIT ？
 			return "";
 		}
+
 		return getLimitString(
                 getPage()
                 , getCount()
diff --git a/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java b/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java
index dcb6d497..7372630c 100755
--- a/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java
+++ b/APIJSONORM/src/main/java/apijson/orm/SQLConfig.java
@@ -34,6 +34,7 @@ public interface SQLConfig<T extends Object> {
 	String DATABASE_SNOWFLAKE = "SNOWFLAKE"; // https://www.snowflake.com
 	String DATABASE_DATABRICKS = "DATABRICKS"; // https://www.databricks.com
 	String DATABASE_CASSANDRA = "CASSANDRA"; // https://cassandra.apache.org
+	String DATABASE_MILVUS = "MILVUS"; // https://milvus.io
 	String DATABASE_INFLUXDB = "INFLUXDB"; // https://www.influxdata.com/products/influxdb-overview
 	String DATABASE_TDENGINE = "TDENGINE"; // https://tdengine.com
 	String DATABASE_REDIS = "REDIS"; // https://redisql.com
@@ -79,10 +80,11 @@ public interface SQLConfig<T extends Object> {
 	boolean isClickHouse();
 	boolean isHive();
 	boolean isPresto();
+	boolean isTrino();
 	boolean isSnowflake();
 	boolean isDatabricks();
 	boolean isCassandra();
-	boolean isTrino();
+	boolean isMilvus();
 	boolean isInfluxDB();
 	boolean isTDengine();
 	boolean isRedis();
@@ -219,6 +221,7 @@ default int[] getDBVersionNums() {
 	String getDatabase();
 	SQLConfig setDatabase(String database);
 
+	String getSQLSchema();
 	String getSchema();
 	SQLConfig setSchema(String schema);