com.netflix.search.query.engine.BaseSearcher Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2016 Netflix, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.netflix.search.query.engine;

import java.io.IOException;
import java.util.List;
import java.util.Set;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.netflix.search.query.Properties;
import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.WebResource;

public abstract class BaseSearcher {

    public static final Logger logger = LoggerFactory.getLogger(BaseSearcher.class);

	private Client client = Client.create();

	public BaseSearcher() {
	}

	public Set getResults(String q, List languages, String dataSetId) throws Throwable
	{
		String urlForGettingDoc = getUrlForGettingDoc(q, languages, dataSetId);

		if (Properties.isPrintUrl.get())
			logger.info(urlForGettingDoc);

		String jsonString = getJsonForQuery(q, languages, dataSetId);

		WebResource webResource = client.resource(urlForGettingDoc);
		ClientResponse response = null;

		if (jsonString != null)
			response = webResource.type("application/json").post(ClientResponse.class, jsonString);
		else
			response = webResource.get(ClientResponse.class);
		if (response == null || (response.getStatus() != 201 && response.getStatus() != 200))
			throw new RuntimeException("Failed : HTTP error code : " + response.getStatus());
		String output = response.getEntity(String.class);
		Set results = getResultsFromServerResponse(output);
		return results;
	}

	public static String getPhraseQueryString(String q)
	{
		if (q == null)
			return null;
		return "\"" + q.replaceAll("[\"|\\\\]", "") + "\"";
	}

	public static String getQueryFields(List localeList)
	{
		StringBuffer sb = new StringBuffer();
		if (localeList != null)
		{
			for (String fieldName : Properties.titleFields.get())
				addNonDefaultLocaleTitleFieldName(localeList, sb, fieldName);
			for (String fieldName : Properties.titleAkaFields.get())
				addNonDefaultLocaleTitleFieldName(localeList, sb, fieldName);
		}
		return sb.toString().substring(0, sb.length()).trim();
	}

	protected static void addNonDefaultLocaleTitleFieldName(List localeList, StringBuffer sb, String fieldName)
	{
		for (String locale : localeList)
		{
			sb.append(fieldName + "_" + locale + " ");
		}
	}

	public abstract String getUrlForGettingDoc(String q, List languages, String dataSetId);
	public abstract Set getResultsFromServerResponse(String output) throws IOException, JsonProcessingException;

	public String getJsonForQuery(String q, List languages, String dataSetId) throws JsonProcessingException
	{
		return null;
	}

	public String getServerUrl()
	{
		return "http://" + Properties.engineHost.get() + ":" + Properties.enginePort.get() + "/" + Properties.engineServlet.get() + "/" + Properties.engineIndexName.get();
	}

}