[Feature] adapt limit for sqlserver (#48214) (#48248)

(cherry picked from commit 8184872)
StarRocks · Aug 13, 2024 · 8dfc8c4 · 8dfc8c4
1 parent 19369d0
commit 8dfc8c4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/be/src/connector/jdbc_connector.cpp b/be/src/connector/jdbc_connector.cpp
@@ -51,6 +51,10 @@ static std::string get_jdbc_sql(const Slice jdbc_url, const std::string& table,
                                 const std::vector<std::string>& filters, int64_t limit) {
     std::ostringstream oss;
     oss << "SELECT";
+    if (limit != -1 && jdbc_url.starts_with("jdbc:sqlserver")) {
+        oss << fmt::format(" TOP({}) ", limit);
+        limit = -1;
+    }
     for (size_t i = 0; i < columns.size(); i++) {
         oss << (i == 0 ? "" : ",") << " " << columns[i];
     }