Refactor: Pull tracing into its own class

1 files changed, 3 insertions, 3 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/dispatch/rpc/ProtobufSerialization.java b/container-search/src/main/java/com/yahoo/search/dispatch/rpc/ProtobufSerialization.java
index 82c570a9975..256ea0185a0 100644
--- a/container-search/src/main/java/com/yahoo/search/dispatch/rpc/ProtobufSerialization.java
+++ b/container-search/src/main/java/com/yahoo/search/dispatch/rpc/ProtobufSerialization.java
@@ -89,7 +89,7 @@ public class ProtobufSerialization {
     }
 
     public static int getTraceLevelForBackend(Query query) {
-        int traceLevel = query.getTraceLevel();
+        int traceLevel = query.getTrace().getLevel();
         if (query.getModel().getExecution().trace().getForceTimestamps()) {
             traceLevel = Math.max(traceLevel, 5); // Backend produces timing information on level 4 and 5
         }
@@ -157,7 +157,7 @@ public class ProtobufSerialization {
         if (includeQueryData) {
             mergeQueryDataToDocsumRequest(query, builder);
         }
-        if (query.getTraceLevel() >= 3) {
+        if (query.getTrace().getLevel() >= 3) {
             query.trace((includeQueryData ? "ProtoBuf: Resending " : "Not resending ") + "query during document summary fetching", 3);
         }
 
@@ -250,7 +250,7 @@ public class ProtobufSerialization {
         if ( ! slimeTrace.isEmpty()) {
             var traces = new Value.ArrayValue();
             traces.add(new SlimeAdapter(BinaryFormat.decode(slimeTrace.toByteArray()).get()));
-            query.trace(traces, query.getTraceLevel());
+            query.trace(traces, query.getTrace().getLevel());
         }
         return result;
     }