log: socket message only prints in trace level

optimism-java · Apr 18, 2024 · be2f0a9 · be2f0a9
1 parent ee9427a
commit be2f0a9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/hildr-utilities/src/main/java/io/optimism/utilities/rpc/Web3jProvider.java b/hildr-utilities/src/main/java/io/optimism/utilities/rpc/Web3jProvider.java
@@ -2,6 +2,8 @@
 
 import java.net.ConnectException;
 import java.util.function.Consumer;
+
+import ch.qos.logback.classic.Level;
 import okhttp3.OkHttpClient;
 import okhttp3.logging.HttpLoggingInterceptor;
 import org.apache.commons.lang3.StringUtils;
@@ -47,7 +49,7 @@ public static Tuple2<Web3j, Web3jService> create(String url) {
         Web3jService web3Srv;
         if (Web3jProvider.isHttp(url)) {
             var okHttpClientBuilder = new OkHttpClient.Builder();
-            if (LOGGER.isDebugEnabled()) {
+            if (LOGGER.isTraceEnabled()) {
                 okHttpClientBuilder.addInterceptor(
                         new HttpLoggingInterceptor(LOGGER::debug).setLevel(HttpLoggingInterceptor.Level.BODY));
             }
@@ -57,6 +59,13 @@ public static Tuple2<Web3j, Web3jService> create(String url) {
             web3Srv = new HttpService(url, okHttpClient);
         } else if (Web3jProvider.isWs(url)) {
             final var web3finalSrv = new WebSocketService(url, true);
+            var logger = LoggerFactory.getLogger(Logger.ROOT_LOGGER_NAME);
+            if (logger instanceof ch.qos.logback.classic.Logger) {
+                var level = LOGGER.isTraceEnabled()
+                    ?  ch.qos.logback.classic.Level.TRACE
+                    : ch.qos.logback.classic.Level.INFO;
+                ((ch.qos.logback.classic.Logger) logger).setLevel(level);
+            }
             wsConnect(web3finalSrv);
             web3Srv = web3finalSrv;
         } else {