1 files changed, 173 insertions, 0 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/handler/HttpSearchResponse.java b/container-search/src/main/java/com/yahoo/search/handler/HttpSearchResponse.java
new file mode 100644
index 00000000000..f844b5dd940
--- /dev/null
+++ b/container-search/src/main/java/com/yahoo/search/handler/HttpSearchResponse.java
@@ -0,0 +1,173 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.handler;
+
+import java.io.IOException;
+import java.io.OutputStream;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
+import com.google.common.util.concurrent.ListenableFuture;
+import com.yahoo.collections.ListMap;
+import com.yahoo.container.jdisc.ExtendedResponse;
+import com.yahoo.container.handler.Coverage;
+import com.yahoo.container.handler.Timing;
+import com.yahoo.container.jdisc.HttpRequest;
+import com.yahoo.container.logging.AccessLogEntry;
+import com.yahoo.container.logging.HitCounts;
+import com.yahoo.jdisc.HeaderFields;
+import com.yahoo.jdisc.handler.CompletionHandler;
+import com.yahoo.jdisc.handler.ContentChannel;
+import com.yahoo.processing.execution.Execution.Trace.LogValue;
+import com.yahoo.processing.rendering.AsynchronousSectionedRenderer;
+import com.yahoo.processing.rendering.Renderer;
+import com.yahoo.search.Query;
+import com.yahoo.search.Result;
+import com.yahoo.search.query.context.QueryContext;
+
+/**
+ * Wrap the result of a query as an HTTP response.
+ *
+ * @author <a href="mailto:steinar@yahoo-inc.com">Steinar Knutsen</a>
+ */
+public class HttpSearchResponse extends ExtendedResponse {
+
+    private final Result result;
+    private final Query query;
+    private final Renderer<Result> rendererCopy;
+    private final Timing timing;
+    private final HitCounts hitCounts;
+
+    public HttpSearchResponse(int status, Result result, Query query, Renderer renderer) {
+        super(status);
+        this.query = query;
+        this.result = result;
+        this.rendererCopy = renderer;
+
+        this.timing = SearchResponse.createTiming(query, result);
+        this.hitCounts = SearchResponse.createHitCounts(query, result);
+        populateHeaders(headers(), result.getHeaders(false));
+    }
+
+    /**
+     * Copy custom HTTP headers from the search result over to the HTTP
+     * response.
+     *
+     * @param outputHeaders
+     *            the headers which will be sent to a client
+     * @param searchHeaders
+     *            the headers from the search result, or null
+     */
+    private static void populateHeaders(HeaderFields outputHeaders,
+            ListMap<String, String> searchHeaders) {
+        if (searchHeaders == null) {
+            return;
+        }
+        for (Map.Entry<String, List<String>> header : searchHeaders.entrySet()) {
+            for (String value : header.getValue()) {
+                outputHeaders.add(header.getKey(), value);
+            }
+        }
+    }
+
+    public ListenableFuture<Boolean> waitableRender(OutputStream stream) throws IOException {
+        return waitableRender(result, query, rendererCopy, stream);
+    }
+
+    public static ListenableFuture<Boolean> waitableRender(Result result,
+                                                           Query query,
+                                                           Renderer<Result> renderer,
+                                                           OutputStream stream) throws IOException {
+        SearchResponse.trimHits(result);
+        SearchResponse.removeEmptySummaryFeatureFields(result);
+        return renderer.render(stream, result, query.getModel().getExecution(), query);
+
+    }
+
+    @Override
+    public void render(OutputStream output, ContentChannel networkChannel, CompletionHandler handler) throws IOException {
+        if (rendererCopy instanceof AsynchronousSectionedRenderer) {
+            AsynchronousSectionedRenderer<Result> renderer = (AsynchronousSectionedRenderer<Result>) rendererCopy;
+            renderer.setNetworkWiring(networkChannel, handler);
+        }
+        try {
+            try {
+                waitableRender(output);
+            } finally {
+                if (!(rendererCopy instanceof AsynchronousSectionedRenderer)) {
+                    output.flush();
+                }
+            }
+        } finally {
+            if (networkChannel != null && !(rendererCopy instanceof AsynchronousSectionedRenderer)) {
+                networkChannel.close(handler);
+            }
+        }
+    }
+
+    @Override
+    public void populateAccessLogEntry(final AccessLogEntry accessLogEntry) {
+        super.populateAccessLogEntry(accessLogEntry);
+        populateAccessLogEntry(accessLogEntry, getHitCounts());
+    }
+
+    /* package-private */
+    static void populateAccessLogEntry(AccessLogEntry jdiscRequestAccessLogEntry, HitCounts hitCounts) {
+        // This entry will be logged at Jetty level. Here we just populate with tidbits from this context.
+
+        jdiscRequestAccessLogEntry.setHitCounts(hitCounts);
+    }
+
+    @Override
+    public String getParsedQuery() {
+        return query.toString();
+    }
+
+    @Override
+    public Timing getTiming() {
+        return timing;
+    }
+
+    @Override
+    public Coverage getCoverage() {
+        return result.getCoverage(false);
+    }
+
+    @Override
+    public HitCounts getHitCounts() {
+        return hitCounts;
+    }
+
+    /**
+     * Returns MIME type of this response
+     */
+    @Override
+    public String getContentType() {
+        return rendererCopy.getMimeType();
+    }
+
+    /**
+     * Returns expected character encoding of this response
+     */
+    @Override
+    public String getCharacterEncoding() {
+        String encoding = result.getQuery().getModel().getEncoding();
+        return (encoding != null) ? encoding : rendererCopy.getEncoding();
+    }
+
+    /** Returns the query wrapped by this */
+    public Query getQuery() { return query; }
+
+    /** Returns the result wrapped by this */
+    public Result getResult() { return result; }
+
+    @Override
+    public Iterable<LogValue> getLogValues() {
+        QueryContext context = query.getContext(false);
+        return context == null
+                ? Collections::emptyIterator
+                : context::logValueIterator;
+    }
+
+}