Add LogLevel Hooks to HttpLogOptions and HttpLoggingPolicy

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

@@ -3,10 +3,14 @@
 
 package com.azure.core.http.policy;
 
-import com.azure.core.util.CoreUtils;
+import com.azure.core.http.HttpPipelineCallContext;
+import com.azure.core.http.HttpResponse;
 import com.azure.core.util.ClientOptions;
+import com.azure.core.util.CoreUtils;
 import com.azure.core.util.logging.ClientLogger;
+import com.azure.core.util.logging.LogLevel;
 
+import java.time.Duration;
 import java.util.Arrays;
 import java.util.HashSet;
 import java.util.List;
@@ -22,6 +26,11 @@ public class HttpLogOptions {
     private Set<String> allowedHeaderNames;
     private Set<String> allowedQueryParamNames;
     private boolean prettyPrintBody;
+
+    private LogLevel defaultLogLevel;
+    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 +105,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 +210,92 @@ 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 LogLevel} used by default when logging requests and responses.
+     * <p>
+     * {@link HttpRequestLogger#getLogLevel(LogLevel, HttpPipelineCallContext)} and {@link
+     * HttpResponseLogger#getLogLevel(LogLevel, HttpResponse, Duration)} can be used to set the {@link LogLevel} for
+     * each request and response being logged.
+     * <p>
+     * By default {@link LogLevel#INFORMATIONAL} is used.
+     *
+     * @return The {@link LogLevel} used by default when logging requests and responses.
+     */
+    public LogLevel getDefaultLogLevel() {
+        return defaultLogLevel;
+    }
+
+    /**
+     * Sets the {@link LogLevel} used by default when logging requests and responses.
+     * <p>
+     * {@link HttpRequestLogger#getLogLevel(LogLevel, HttpPipelineCallContext)} and {@link
+     * HttpResponseLogger#getLogLevel(LogLevel, HttpResponse, Duration)} can be used to set the {@link LogLevel} for
+     * each request and response being logged.
+     * <p>
+     * By default {@link LogLevel#INFORMATIONAL} is used.
+     *
+     * @param defaultLogLevel The default log level.
+     * @return The updated HttpLogOptions object.
+     */
+    public HttpLogOptions setDefaultLogLevel(LogLevel defaultLogLevel) {
+        this.defaultLogLevel = defaultLogLevel;
+        return this;
+    }
+
+    /**
+     * Gets the {@link HttpRequestLogger} that will be used to log requests.
+     * <p>
+     * A default logger will be used if one isn't supplied.
+     *
+     * @return The {@link HttpRequestLogger} that will be used to log requests.
+     */
+    public HttpRequestLogger getRequestLogger() {
+        return requestLogger;
+    }
+
+    /**
+     * Sets the {@link HttpRequestLogger} that will be used to log requests.
+     * <p>
+     * A default logger will be used if one isn't supplied.
+     *
+     * @param requestLogger The {@link HttpRequestLogger} that will be used to log 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 responses.
+     * <p>
+     * A default logger will be used if one isn't supplied.
+     *
+     * @return The {@link HttpResponseLogger} that will be used to log responses.
+     */
+    public HttpResponseLogger getResponseLogger() {
+        return responseLogger;
+    }
+
+    /**
+     * Sets the {@link HttpResponseLogger} that will be used to log responses.
+     * <p>
+     * A default logger will be sued if one isn't supplied.
+     *
+     * @param responseLogger The {@link HttpResponseLogger} that will be used to log responses.
+     * @return The updated HttpLogOptions object.
+     */
+    public HttpLogOptions setResponseLogger(HttpResponseLogger responseLogger) {
+        this.responseLogger = responseLogger;
+        return this;
+    }
 }