Propagate nodes too.

2 files changed, 10 insertions, 3 deletions

diff --git a/container-search/src/main/java/com/yahoo/prelude/fastsearch/VespaBackEndSearcher.java b/container-search/src/main/java/com/yahoo/prelude/fastsearch/VespaBackEndSearcher.java
index ea354e1291f..3320c167702 100644
--- a/container-search/src/main/java/com/yahoo/prelude/fastsearch/VespaBackEndSearcher.java
+++ b/container-search/src/main/java/com/yahoo/prelude/fastsearch/VespaBackEndSearcher.java
@@ -465,9 +465,10 @@ public abstract class VespaBackEndSearcher extends PingableSearcher {
         }
 
         if (resultPacket.getCoverageFeature()) {
-            result.setCoverage(new Coverage(resultPacket.getCoverageDocs(), resultPacket.getActiveDocs())
+            result.setCoverage(new Coverage(resultPacket.getCoverageDocs(), resultPacket.getActiveDocs(), resultPacket.getNodesReplied())
                     .setSoonActive(resultPacket.getSoonActiveDocs())
-                    .setDegradedReason(resultPacket.getDegradedReason()));
+                    .setDegradedReason(resultPacket.getDegradedReason())
+                    .setNodesTried(resultPacket.getNodesQueried()));
         }
     }
 
diff --git a/container-search/src/main/java/com/yahoo/search/result/Coverage.java b/container-search/src/main/java/com/yahoo/search/result/Coverage.java
index bd1245ea2ff..81aa4a08f18 100644
--- a/container-search/src/main/java/com/yahoo/search/result/Coverage.java
+++ b/container-search/src/main/java/com/yahoo/search/result/Coverage.java
@@ -12,7 +12,11 @@ import com.google.common.annotations.Beta;
 public class Coverage extends com.yahoo.container.handler.Coverage {
 
     public Coverage(long docs, long active) {
-        super(docs, active, 0, 1);
+        this(docs, active, 0);
+    }
+
+    public Coverage(long docs, long active, int nodes) {
+        super(docs, active, nodes, 1);
     }
 
     @Deprecated
@@ -40,4 +44,6 @@ public class Coverage extends com.yahoo.container.handler.Coverage {
      */
     public Coverage setDegradedReason(int degradedReason) { this.degradedReason = degradedReason; return this; }
 
+    public Coverage setNodesTried(int nodesTried) { super.setNodesTried(nodesTried); return this; }
+
 }