FIND-13403: Support unique cache for query

- Add SingleResultCache for set cache_uniq=true

Author: ManhOptimizely

APIs/src/EpiServer.ContentGraph/Api/Querying/GraphQueryBuilder.cs

@@ -11,10 +11,11 @@
 using EPiServer.ServiceLocation;
 using EPiServer.ContentGraph.Helpers.Text;
 using EPiServer.ContentGraph.Helpers;
+using EPiServer.ContentGraph.Tracing;
 
 namespace EPiServer.ContentGraph.Api.Querying
 {
-    public class GraphQueryBuilder : IQuery
+    public class GraphQueryBuilder : IQuery, ITraceable
     {
         private readonly IHttpClientFactory _httpClientFactory;
         private readonly HttpClient _httpClient;
@@ -24,7 +25,9 @@ public class GraphQueryBuilder : IQuery
         private const string UnCachedPath = "?cache=false";
         private Dictionary<string, IFragmentBuilder> _fragmentBuilders;
         private readonly List<string> typeQueries = new List<string>();
-        public Action<JsonRequest> RequestActions;
+        public Action<OptiGraphOptions> GraphOptionsAction;
+
+        Guid ITraceable.TraceId => Guid.NewGuid();
 
         public GraphQueryBuilder()
         {
@@ -116,6 +119,7 @@ public GraphQueryBuilder OperationName(string op)
         }
         private string GetServiceUrl()
         {
+            UpdateQueryPath();
             return _optiGraphOptions.GatewayAddress + _optiGraphOptions.QueryPath;
         }
         private string GetAuthorization(string body)
@@ -311,19 +315,22 @@ private void AdditionalInformation(JsonRequest request, string body)
             {
                 request.AddRequestHeader("cg-include-deleted", "true");
             }
-            if (!_optiGraphOptions.Cache)
-            {
-                Regex regex = new Regex(@"\?cache=\w*");
-                _optiGraphOptions.QueryPath = _optiGraphOptions.QueryPath.Replace(regex.Match(_optiGraphOptions.QueryPath).Value, UnCachedPath);
-            }
-            //apply actions on request before send
-            ApplyRequestActions(request);
         }
-        internal void ApplyRequestActions(JsonRequest request)
+        private void UpdateQueryPath()
         {
-            if (RequestActions.IsNotNull())
+            Regex regex = new Regex(@"\?cache=\w*");
+            if (regex.IsMatch(_optiGraphOptions.QueryPath))
             {
-                RequestActions(request);
+                _optiGraphOptions.QueryPath = 
+                    _optiGraphOptions.QueryPath
+                    .Replace(regex.Match(_optiGraphOptions.QueryPath).Value, $"?cache={_optiGraphOptions.Cache}");
+            }
+            else
+            {
+                _optiGraphOptions.QueryPath = $"{_optiGraphOptions.QueryPath}?cache={_optiGraphOptions.Cache}";
+            }
+            if (GraphOptionsAction.IsNotNull()) {
+                GraphOptionsAction(_optiGraphOptions);
             }
         }
         public void AddQuery(string typeQuery)

APIs/src/EpiServer.ContentGraph/Extensions/GraphQueryBuilderExtension.cs

@@ -1,4 +1,6 @@
-using EPiServer.ContentGraph.Api.Querying;
+using Azure.Core;
+using EPiServer.ContentGraph.Api.Querying;
+using EPiServer.ContentGraph.Tracing;
 
 namespace EPiServer.ContentGraph.Extensions
 {
@@ -29,7 +31,18 @@ public static TypeQueryBuilder<T> BeginType<T>(this GraphQueryBuilder queryBuild
         /// <returns></returns>
         public static GraphQueryBuilder SingleResultCache(this GraphQueryBuilder queryBuilder)
         {
-            queryBuilder.RequestActions = request => request.AddRequestHeader("cache_uniq", "true");
+            queryBuilder.GraphOptionsAction = options =>
+            {
+                if (options.QueryPath.Contains("cache=true"))
+                {
+                    options.QueryPath += "&cache_uniq=true";
+                }
+                else
+                {
+                    //log warning messge
+                    Trace.Instance.Add(new TraceEvent(queryBuilder, "cache_uniq is set to true but cache is not enable. Please enable it in the config before use."));
+                }
+            };
             return queryBuilder;
         }
     }