controllers.api.v1.AdvSearch.java Source code

Introduction

Here is the source code for controllers.api.v1.AdvSearch.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.AdvSearchDAO;
import dao.SearchDAO;
import org.apache.commons.lang3.StringUtils;
import play.Logger;
import play.Play;
import play.libs.Json;
import play.mvc.Controller;
import play.mvc.Result;

public class AdvSearch extends Controller {
    public static Result getDatasetSources() {
        ObjectNode result = Json.newObject();

        result.put("status", "ok");
        result.set("sources", Json.toJson(AdvSearchDAO.getDatasetSources()));

        return ok(result);
    }

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

        result.put("status", "ok");
        result.set("scopes", Json.toJson(AdvSearchDAO.getDatasetScopes()));

        return ok(result);
    }

    public static Result getDatasetTableNames() {
        ObjectNode result = Json.newObject();
        String scopes = request().getQueryString("scopes");
        result.put("status", "ok");
        result.set("tables", Json.toJson(AdvSearchDAO.getTableNames(scopes)));

        return ok(result);
    }

    public static Result getDatasetFields() {
        ObjectNode result = Json.newObject();
        String tables = request().getQueryString("tables");
        result.put("status", "ok");
        result.set("fields", Json.toJson(AdvSearchDAO.getFields(tables)));

        return ok(result);
    }

    public static Result getFlowApplicationCodes() {
        ObjectNode result = Json.newObject();
        result.put("status", "ok");
        result.set("appcodes", Json.toJson(AdvSearchDAO.getFlowApplicationCodes()));

        return ok(result);
    }

    public static Result getFlowNames() {
        ObjectNode result = Json.newObject();
        String apps = request().getQueryString("apps");
        result.put("status", "ok");
        result.set("flowNames", Json.toJson(AdvSearchDAO.getFlowNames(apps)));

        return ok(result);
    }

    public static Result getJobNames() {
        ObjectNode result = Json.newObject();
        result.put("status", "ok");
        result.set("jobNames", Json.toJson(AdvSearchDAO.getFlowJobNames()));

        return ok(result);
    }

    public static Result search() {
        ObjectNode result = Json.newObject();
        String searchOptStr = request().getQueryString("searchOpts");
        JsonNode searchOpt = Json.parse(searchOptStr);
        int page = 1;
        int size = 10;
        String pageStr = request().getQueryString("page");
        if (StringUtils.isBlank(pageStr)) {
            page = 1;
        } else {
            try {
                page = Integer.parseInt(pageStr);
            } catch (NumberFormatException e) {
                Logger.error("AdvSearch Controller search 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("AdvSearch Controller search wrong page parameter. Error message: " + e.getMessage());
                size = 10;
            }
        }
        result.put("status", "ok");
        String searchEngine = Play.application().configuration().getString(SearchDAO.WHEREHOWS_SEARCH_ENGINE__KEY);

        if (searchOpt != null && searchOpt.has("category")) {
            String category = searchOpt.get("category").asText();
            if (category.equalsIgnoreCase("flow")) {
                if (StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch")) {
                    result.set("result", Json.toJson(AdvSearchDAO.elasticSearchFlowJobs(searchOpt, page, size)));
                } else {
                    result.set("result", Json.toJson(AdvSearchDAO.searchFlows(searchOpt, page, size)));
                }
                return ok(result);
            } else if (category.equalsIgnoreCase("metric")) {
                if (StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch")) {
                    result.set("result", Json.toJson(AdvSearchDAO.elasticSearchMetric(searchOpt, page, size)));
                } else {
                    result.set("result", Json.toJson(AdvSearchDAO.searchMetrics(searchOpt, page, size)));
                }
                return ok(result);
            }

        }

        if (StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch")) {
            result.set("result", Json.toJson(AdvSearchDAO.elasticSearch(searchOpt, page, size)));
        } else {
            result.set("result", Json.toJson(AdvSearchDAO.search(searchOpt, page, size)));
        }

        return ok(result);
    }

}