Merge pull request Azure#1012 from abhivijay96/master

ServiceClient that allows setting the UserAgent string
pomortaz · May 26, 2016 · ab5315f · ab5315f
2 parents 73987f0 + ca2b687
commit ab5315f
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/ClientRuntimes/CSharp/Microsoft.Rest.ClientRuntime/ServiceClient.cs b/ClientRuntimes/CSharp/Microsoft.Rest.ClientRuntime/ServiceClient.cs
@@ -211,6 +211,47 @@ protected void InitializeHttpClient(HttpClientHandler httpClientHandler, params
                 GetClientVersion()));
         }
 
+        /// <summary>
+        /// Sets the product name to be used in the user agent header when making requests
+        /// </summary>
+        /// <param name="productName">Name of the product to be used in the user agent</param>
+        public bool SetUserAgent(string productName)
+        {
+            if (!_disposed && HttpClient != null)
+            {
+                // Clear the old user agent
+                HttpClient.DefaultRequestHeaders.UserAgent.Clear();
+                HttpClient.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(productName, GetClientVersion()));
+
+                // Returns true if the user agent was set 
+                return true;
+            }
+
+            // Returns false if the HttpClient was disposed before invoking the method
+            return false;
+        }
+
+        /// <summary>
+        /// Sets the product name and version to be used in the user agent header when making requests
+        /// </summary>
+        /// <param name="productName">Name of the product to be used in the user agent</param>
+        /// <param name="version">Version of the product to be used in the user agent</param>
+        public bool SetUserAgent(string productName, string version)
+        {
+            if (!_disposed && HttpClient != null)
+            {
+                // Clear the old user agent
+                HttpClient.DefaultRequestHeaders.UserAgent.Clear();
+                HttpClient.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(productName, version));
+
+                // Returns true if the user agent was set
+                return true;
+            }
+
+            // Returns false if the HttpClient was disposed before invoking the method
+            return false;
+        }
+
         /// <summary>
         /// Gets the AssemblyInformationalVersion if available
         /// if not it gets the AssemblyFileVerion