datahub/wherehows-frontend/app/controllers/api/v1/AdvSearch.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.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 getDashboardNames() {
    ObjectNode result = Json.newObject();
    result.put("status", "ok");
    result.set("dashboardNames", Json.toJson(AdvSearchDAO.getMetricDashboardNames()));

    return ok(result);
  }

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

    return ok(result);
  }

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

    return ok(result);
  }

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

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