datahub/web/app/controllers/api/v1/Search.java

/**
 * Copyright 2015 LinkedIn Corp. All rights reserved.
 *
 * 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.
 */
package controllers.api.v1;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import dao.SearchDAO;
import models.DatasetColumn;
import play.Play;
import play.api.libs.json.JsValue;
import play.libs.Json;
import play.mvc.Controller;
import play.mvc.Result;
import play.Logger;
import org.apache.commons.lang3.StringUtils;
import dao.DatasetsDAO;

import java.util.List;

public class Search extends Controller
{
    public static Result getSearchAutoComplete()
    {
        ObjectNode result = Json.newObject();
        result.put("status", "ok");
        result.set("source", Json.toJson(SearchDAO.getAutoCompleteList()));

        return ok(result);
    }

    public static Result getSearchAutoCompleteForDataset()
    {
        ObjectNode result = Json.newObject();
        result.put("status", "ok");
        result.set("source", Json.toJson(SearchDAO.getAutoCompleteListForDataset()));

        return ok(result);
    }

    public static Result getSearchAutoCompleteForMetric()
    {
        ObjectNode result = Json.newObject();
        result.put("status", "ok");
        result.set("source", Json.toJson(SearchDAO.getAutoCompleteListForMetric()));

        return ok(result);
    }

    public static Result getSearchAutoCompleteForFlow()
    {
        ObjectNode result = Json.newObject();
        result.put("status", "ok");
        result.set("source", Json.toJson(SearchDAO.getAutoCompleteListForFlow()));

        return ok(result);
    }


    public static Result searchByKeyword()
    {
        ObjectNode result = Json.newObject();

        int page = 1;
        int size = 10;
        String keyword = request().getQueryString("keyword");
        String category = request().getQueryString("category");
        String source = request().getQueryString("source");
        String pageStr = request().getQueryString("page");
        if (StringUtils.isBlank(pageStr))
        {
            page = 1;
        }
        else
        {
            try
            {
                page = Integer.parseInt(pageStr);
            }
            catch(NumberFormatException e)
            {
                Logger.error("Dataset Controller searchByKeyword wrong page parameter. Error message: " +
                        e.getMessage());
                page = 1;
            }
        }


        String sizeStr = request().getQueryString("size");
        if (StringUtils.isBlank(sizeStr))
        {
            size = 10;
        }
        else
        {
            try
            {
                size = Integer.parseInt(sizeStr);
            }
            catch(NumberFormatException e)
            {
                Logger.error("Dataset Controller searchByKeyword wrong page parameter. Error message: " +
                        e.getMessage());
                size = 10;
            }
        }

        result.put("status", "ok");
        Boolean isDefault = false;
        if (StringUtils.isBlank(category))
        {
            category = "datasets";
        }
        if (StringUtils.isBlank(source) || source.equalsIgnoreCase("all") || source.equalsIgnoreCase("default"))
        {
            source = null;
        }

        String searchEngine = Play.application().configuration().getString(SearchDAO.WHEREHOWS_SEARCH_ENGINE__KEY);

        if (category.toLowerCase().equalsIgnoreCase("metric"))
        {
            if(StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch"))
            {
                result.set("result", SearchDAO.elasticSearchMetricByKeyword(category, keyword, page, size));

            }
            else
            {
                result.set("result", SearchDAO.getPagedMetricByKeyword(category, keyword, page, size));
            }
        }
        else if (category.toLowerCase().equalsIgnoreCase("flows"))
        {
            if(StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch"))
            {
                result.set("result", SearchDAO.elasticSearchFlowByKeyword(category, keyword, page, size));

            }
            else
            {
                result.set("result", SearchDAO.getPagedFlowByKeyword(category, keyword, page, size));
            }
        }
        else if (category.toLowerCase().equalsIgnoreCase("jobs"))
        {
            if(StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch"))
            {
                result.set("result", SearchDAO.elasticSearchFlowByKeyword(category, keyword, page, size));

            }
            else
            {
                result.set("result", SearchDAO.getPagedJobByKeyword(category, keyword, page, size));
            }
        }
        else if (category.toLowerCase().equalsIgnoreCase("comments"))
        {
            if(StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch"))
            {
                result.set("result", SearchDAO.elasticSearchDatasetByKeyword(category, keyword, null, page, size));
            }
            else
            {
                result.set("result", SearchDAO.getPagedCommentsByKeyword(category, keyword, page, size));
            }

        }
        else
        {
            if(StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch"))
            {
                result.set("result", SearchDAO.elasticSearchDatasetByKeyword(category, keyword, source, page, size));
            }
            else
            {
                result.set("result", SearchDAO.getPagedDatasetByKeyword(category, keyword, source, page, size));
            }
        }

        return ok(result);
    }


}