Add LogLevel Hooks to HttpLogOptions and HttpLoggingPolicy (Azure#16088)

Add LogLevel Hooks to HttpLogOptions and HttpLoggingPolicy

Author: alzimmermsft

sdk/core/azure-core/src/main/java/com/azure/core/http/HttpPipelineCallContext.java

@@ -3,13 +3,14 @@
 
 package com.azure.core.http;
 
+import com.azure.core.implementation.http.HttpPipelineCallContextHelper;
 import com.azure.core.util.Context;
 
 import java.util.Objects;
 import java.util.Optional;
 
 /**
- * Type representing context local to a single http request and it's response.
+ * Represents the information used to make a single HTTP request.
  */
 public final class HttpPipelineCallContext {
     private HttpRequest httpRequest;
@@ -46,6 +47,10 @@ public final class HttpPipelineCallContext {
         this.data = data;
     }
 
+    static {
+        HttpPipelineCallContextHelper.setAccessor(HttpPipelineCallContext::getContext);
+    }
+
     /**
      * Stores a key-value data in the context.
      *

sdk/core/azure-core/src/main/java/com/azure/core/http/policy/HttpLogOptions.java

@@ -3,8 +3,8 @@
 
 package com.azure.core.http.policy;
 
-import com.azure.core.util.CoreUtils;
 import com.azure.core.util.ClientOptions;
+import com.azure.core.util.CoreUtils;
 import com.azure.core.util.logging.ClientLogger;
 
 import java.util.Arrays;
@@ -22,6 +22,10 @@ public class HttpLogOptions {
     private Set<String> allowedHeaderNames;
     private Set<String> allowedQueryParamNames;
     private boolean prettyPrintBody;
+
+    private HttpRequestLogger requestLogger;
+    private HttpResponseLogger responseLogger;
+
     private final ClientLogger logger = new ClientLogger(HttpLogOptions.class);
 
     private static final int MAX_APPLICATION_ID_LENGTH = 24;
@@ -96,10 +100,9 @@ public Set<String> getAllowedHeaderNames() {
      *
      * <p>
      * This method sets the provided header names to be the whitelisted header names which will be logged for all HTTP
-     * requests and responses, overwriting any previously configured headers, including the default set. Additionally,
-     * users can use {@link HttpLogOptions#addAllowedHeaderName(String)} or
-     * {@link HttpLogOptions#getAllowedHeaderNames()} to add or remove more headers names to the existing set of
-     * allowed header names.
+     * requests and responses, overwriting any previously configured headers. Additionally, users can use {@link
+     * HttpLogOptions#addAllowedHeaderName(String)} or {@link HttpLogOptions#getAllowedHeaderNames()} to add or remove
+     * more headers names to the existing set of allowed header names.
      * </p>
      *
      * @param allowedHeaderNames The list of whitelisted header names from the user.
@@ -202,12 +205,60 @@ public boolean isPrettyPrintBody() {
     /**
      * Sets flag to allow pretty printing of message bodies.
      *
-     * @param prettyPrintBody If true, pretty prints message bodies when logging. If the detailLevel does not
-     *                        include body logging, this flag does nothing.
+     * @param prettyPrintBody If true, pretty prints message bodies when logging. If the detailLevel does not include
+     * body logging, this flag does nothing.
      * @return The updated HttpLogOptions object.
      */
     public HttpLogOptions setPrettyPrintBody(boolean prettyPrintBody) {
         this.prettyPrintBody = prettyPrintBody;
         return this;
     }
+
+    /**
+     * Gets the {@link HttpRequestLogger} that will be used to log HTTP requests.
+     * <p>
+     * A default {@link HttpRequestLogger} will be used if one isn't supplied.
+     *
+     * @return The {@link HttpRequestLogger} that will be used to log HTTP requests.
+     */
+    public HttpRequestLogger getRequestLogger() {
+        return requestLogger;
+    }
+
+    /**
+     * Sets the {@link HttpRequestLogger} that will be used to log HTTP requests.
+     * <p>
+     * A default {@link HttpRequestLogger} will be used if one isn't supplied.
+     *
+     * @param requestLogger The {@link HttpRequestLogger} that will be used to log HTTP requests.
+     * @return The updated HttpLogOptions object.
+     */
+    public HttpLogOptions setRequestLogger(HttpRequestLogger requestLogger) {
+        this.requestLogger = requestLogger;
+        return this;
+    }
+
+    /**
+     * Gets the {@link HttpResponseLogger} that will be used to log HTTP responses.
+     * <p>
+     * A default {@link HttpResponseLogger} will be used if one isn't supplied.
+     *
+     * @return The {@link HttpResponseLogger} that will be used to log HTTP responses.
+     */
+    public HttpResponseLogger getResponseLogger() {
+        return responseLogger;
+    }
+
+    /**
+     * Sets the {@link HttpResponseLogger} that will be used to log HTTP responses.
+     * <p>
+     * A default {@link HttpResponseLogger} will be used if one isn't supplied.
+     *
+     * @param responseLogger The {@link HttpResponseLogger} that will be used to log HTTP responses.
+     * @return The updated HttpLogOptions object.
+     */
+    public HttpLogOptions setResponseLogger(HttpResponseLogger responseLogger) {
+        this.responseLogger = responseLogger;
+        return this;
+    }
 }