From cd5d1e7f02ab3f770f00474c32ee5364e0c51618 Mon Sep 17 00:00:00 2001
From: Mohamed Shaban <v-moshaban@microsoft.com>
Date: Mon, 12 Jul 2021 20:23:22 +0200
Subject: [PATCH] adding headers and query parms for logging (#22537)

---
 .../src/DocumentTranslationClientOptions.cs   | 27 +++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/sdk/translation/Azure.AI.Translation.Document/src/DocumentTranslationClientOptions.cs b/sdk/translation/Azure.AI.Translation.Document/src/DocumentTranslationClientOptions.cs
index 7f1737afcd6f..ee140068ffaa 100644
--- a/sdk/translation/Azure.AI.Translation.Document/src/DocumentTranslationClientOptions.cs
+++ b/sdk/translation/Azure.AI.Translation.Document/src/DocumentTranslationClientOptions.cs
@@ -30,6 +30,7 @@ public class DocumentTranslationClientOptions : ClientOptions
         public DocumentTranslationClientOptions(ServiceVersion version = LatestVersion)
         {
             Version = version;
+            AddLoggedHeadersAndQueryParameters();
         }
 
         internal string GetVersionString()
@@ -52,5 +53,31 @@ public enum ServiceVersion
             /// </summary>
             V1_0 = 1
         }
+
+        /// <summary>
+        /// Add headers and query parameters that are considered safe for logging or including in
+        /// error messages by default.
+        /// </summary>
+        private void AddLoggedHeadersAndQueryParameters()
+        {
+            Diagnostics.LoggedHeaderNames.Add("Operation-Location");
+            Diagnostics.LoggedHeaderNames.Add("Content-Encoding");
+            Diagnostics.LoggedHeaderNames.Add("Vary");
+            Diagnostics.LoggedHeaderNames.Add("apim-request-id");
+            Diagnostics.LoggedHeaderNames.Add("X-RequestId");
+            Diagnostics.LoggedHeaderNames.Add("Set-Cookie");
+            Diagnostics.LoggedHeaderNames.Add("X-Powered-By");
+            Diagnostics.LoggedHeaderNames.Add("Strict-Transport-Security");
+            Diagnostics.LoggedHeaderNames.Add("x-content-type-options");
+
+            Diagnostics.LoggedQueryParameters.Add("$top");
+            Diagnostics.LoggedQueryParameters.Add("$skip");
+            Diagnostics.LoggedQueryParameters.Add("$maxpagesize");
+            Diagnostics.LoggedQueryParameters.Add("ids");
+            Diagnostics.LoggedQueryParameters.Add("statuses");
+            Diagnostics.LoggedQueryParameters.Add("createdDateTimeUtcStart");
+            Diagnostics.LoggedQueryParameters.Add("createdDateTimeUtcEnd");
+            Diagnostics.LoggedQueryParameters.Add("$orderBy");
+        }
     }
 }