controllers.api.v1.Search.java Source code

Introduction

Here is the source code for controllers.api.v1.Search.java

Source

/**
 * 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 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);
    }

}