Add LogLevel Hooks to HttpLogOptions and HttpLoggingPolicy

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

@@ -3,9 +3,10 @@
 
 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 com.azure.core.util.logging.LogLevel;
 
 import java.util.Arrays;
 import java.util.HashSet;
@@ -22,6 +23,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 +102,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 +207,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 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;
+    }
 }