[Internal] Query: Adds support for continuation using SqlQuerySpec

Microsoft.Azure.Cosmos/src/CosmosSqlQuerySpecJsonConverter.cs

@@ -24,7 +24,8 @@ internal CosmosSqlQuerySpecJsonConverter(CosmosSerializer userSerializer)
 
         public override bool CanConvert(Type objectType)
         {
-            return typeof(SqlParameter) == objectType;
+            return typeof(SqlParameter) == objectType
+                || typeof(SqlQueryResumeFilter.ResumeValue).IsAssignableFrom(objectType);
         }
 
         public override object ReadJson(JsonReader reader, Type objectType, object existingValue, JsonSerializer serializer)
@@ -34,32 +35,89 @@ public override object ReadJson(JsonReader reader, Type objectType, object exist
 
         public override void WriteJson(JsonWriter writer, object value, JsonSerializer serializer)
         {
-            SqlParameter sqlParameter = (SqlParameter)value;
-
-            writer.WriteStartObject();
-            writer.WritePropertyName("name");
-            serializer.Serialize(writer, sqlParameter.Name);
-            writer.WritePropertyName("value");
-
-            // if the SqlParameter has stream value we dont pass it through the custom serializer.
-            if (sqlParameter.Value is SerializedParameterValue serializedEncryptedData)
-            {
-                writer.WriteRawValue(serializedEncryptedData.rawSerializedJsonValue);
-            }
-            else
+            if (value is SqlParameter sqlParameter)
             {
-                // Use the user serializer for the parameter values so custom conversions are correctly handled
-                using (Stream str = this.UserSerializer.ToStream(sqlParameter.Value))
+                writer.WriteStartObject();
+                writer.WritePropertyName("name");
+                serializer.Serialize(writer, sqlParameter.Name);
+                writer.WritePropertyName("value");
+
+                // if the SqlParameter has stream value we dont pass it through the custom serializer.
+                if (sqlParameter.Value is SerializedParameterValue serializedEncryptedData)
                 {
-                    using (StreamReader streamReader = new StreamReader(str))
+                    writer.WriteRawValue(serializedEncryptedData.rawSerializedJsonValue);
+                }
+                else
+                {
+                    // Use the user serializer for the parameter values so custom conversions are correctly handled
+                    using (Stream str = this.UserSerializer.ToStream(sqlParameter.Value))
                     {
-                        string parameterValue = streamReader.ReadToEnd();
-                        writer.WriteRawValue(parameterValue);
+                        using (StreamReader streamReader = new StreamReader(str))
+                        {
+                            string parameterValue = streamReader.ReadToEnd();
+                            writer.WriteRawValue(parameterValue);
+                        }
                     }
                 }
+
+                writer.WriteEndObject();
             }
+            else if (value is SqlQueryResumeFilter.ResumeValue resumeValue)
+            {
+                this.WriteResumeValue(writer, resumeValue, serializer);
+            }
+        }
 
-            writer.WriteEndObject();
+        private void WriteResumeValue(JsonWriter writer, SqlQueryResumeFilter.ResumeValue value, JsonSerializer serializer)
+        {
+            switch (value)
+            {
+                case SqlQueryResumeFilter.UndefinedResumeValue:
+                    writer.WriteStartArray();
+                    writer.WriteEndArray();
+                    break;
+
+                case SqlQueryResumeFilter.NullResumeValue:
+                    writer.WriteNull();
+                    break;
+
+                case SqlQueryResumeFilter.BooleanResumeValue booleanValue:
+                    serializer.Serialize(writer, booleanValue.Value);
+                    break;
+
+                case SqlQueryResumeFilter.NumberResumeValue numberValue:
+                    serializer.Serialize(writer, numberValue.Value);
+                    break;
+
+                case SqlQueryResumeFilter.StringResumeValue stringValue:
+                    serializer.Serialize(writer, stringValue.Value.ToString());
+                    break;
+
+                case SqlQueryResumeFilter.ArrayResumeValue arrayValue:
+                    writer.WriteStartObject();
+                    writer.WritePropertyName(SqlQueryResumeFilter.ResumeValue.PropertyNames.Type);
+                    writer.WriteValue(SqlQueryResumeFilter.ResumeValue.PropertyNames.ArrayType);
+                    writer.WritePropertyName(SqlQueryResumeFilter.ResumeValue.PropertyNames.Low);
+                    writer.WriteValue(arrayValue.HashValue.GetLow());
+                    writer.WritePropertyName(SqlQueryResumeFilter.ResumeValue.PropertyNames.High);
+                    writer.WriteValue(arrayValue.HashValue.GetHigh());
+                    writer.WriteEndObject();
+                    break;
+
+                case SqlQueryResumeFilter.ObjectResumeValue objectValue:
+                    writer.WriteStartObject();
+                    writer.WritePropertyName(SqlQueryResumeFilter.ResumeValue.PropertyNames.Type);
+                    writer.WriteValue(SqlQueryResumeFilter.ResumeValue.PropertyNames.ObjectType);
+                    writer.WritePropertyName(SqlQueryResumeFilter.ResumeValue.PropertyNames.Low);
+                    writer.WriteValue(objectValue.HashValue.GetLow());
+                    writer.WritePropertyName(SqlQueryResumeFilter.ResumeValue.PropertyNames.High);
+                    writer.WriteValue(objectValue.HashValue.GetHigh());
+                    writer.WriteEndObject();
+                    break;
+
+                default:
+                    throw new ArgumentException($"Invalid {nameof(SqlQueryResumeFilter.ResumeValue)} type.");
+            }
         }
 
         /// <summary>