From 4657a546ba5b079e0fd767bf8b1b369c96798740 Mon Sep 17 00:00:00 2001
From: Neil Deshpande <ndeshpan@microsoft.com>
Date: Mon, 22 Apr 2024 15:11:28 -0700
Subject: [PATCH] Add a new capability for non streaming order by in
 QueryFeatures and add a new flag to QueryInfo that will be used by
 ServiceInterop to indicate non streaming order by queries

---
 .../src/Query/Core/QueryPlan/QueryFeature.cs               | 3 ++-
 .../src/Query/Core/QueryPlan/QueryInfo.cs                  | 7 +++++++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryFeature.cs b/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryFeature.cs
index 6e20108ea7..0b3ed7fdc1 100644
--- a/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryFeature.cs
+++ b/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryFeature.cs
@@ -27,6 +27,7 @@ enum QueryFeatures : ulong
         OrderBy = 1 << 7,
         Top = 1 << 8,
         NonValueAggregate = 1 << 9,
-        DCount = 1 << 10
+        DCount = 1 << 10,
+        NonStreamingOrderBy = 1 << 11,
     }
 }
\ No newline at end of file
diff --git a/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryInfo.cs b/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryInfo.cs
index d210a6daf6..fa796a4965 100644
--- a/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryInfo.cs
+++ b/Microsoft.Azure.Cosmos/src/Query/Core/QueryPlan/QueryInfo.cs
@@ -102,6 +102,13 @@ public bool HasSelectValue
 
         [JsonProperty("dCountInfo")]
         public DCountInfo DCountInfo
+        {
+            get;
+            set;
+        }
+
+        [JsonProperty("hasNonStreamingOrderBy")]
+        public bool HasNonStreamingOrderBy
         {
             get;
             set;