io.airlift.http.server.search.SearchServlet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of webserver Show documentation

The newest version!

package io.airlift.http.server.search;

import com.google.common.base.Strings;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.core.MediaType;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.List;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * Search servlet.
 *
 * @since 0.0.6
 * @see SearchIndex
 */
public class SearchServlet extends HttpServlet {
  private static final Logger log = LoggerFactory.getLogger(SearchServlet.class);

  private final SearchIndex searchIndex;

  private final Gson gson;

  public SearchServlet(final SearchIndex searchIndex) {
    this.searchIndex = checkNotNull(searchIndex);
    this.gson = new GsonBuilder().setPrettyPrinting().create();
  }

  @Override
  protected void doGet(final HttpServletRequest request, final HttpServletResponse response) throws ServletException, IOException {
    String query = request.getParameter("q");
    if (Strings.emptyToNull(query) == null) {
      response.sendError(400, "Missing or invalid query");
      return;
    }

    List results = searchIndex.search(query);
    log.trace("Results: {}", results);

    String body = gson.toJson(results);
    response.setContentType(MediaType.APPLICATION_JSON);
    response.setContentLength(body.length());

    PrintWriter writer = response.getWriter();
    writer.print(body);
    writer.flush();

    response.setStatus(200);
  }
}