2015-11-19 14:39:21 -08:00
|
|
|
/**
|
|
|
|
* 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;
|
2016-05-20 17:59:28 -07:00
|
|
|
import play.Play;
|
2015-11-19 14:39:21 -08:00
|
|
|
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);
|
|
|
|
}
|
|
|
|
|
2017-03-23 17:26:29 -07:00
|
|
|
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);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-04-21 16:25:08 -07:00
|
|
|
public static Result searchByKeyword()
|
2015-11-19 14:39:21 -08:00
|
|
|
{
|
|
|
|
ObjectNode result = Json.newObject();
|
|
|
|
|
|
|
|
int page = 1;
|
|
|
|
int size = 10;
|
2016-04-21 16:25:08 -07:00
|
|
|
String keyword = request().getQueryString("keyword");
|
2015-11-19 14:39:21 -08:00
|
|
|
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)
|
|
|
|
{
|
2016-03-02 15:06:10 -08:00
|
|
|
Logger.error("Dataset Controller searchByKeyword wrong page parameter. Error message: " +
|
2015-11-19 14:39:21 -08:00
|
|
|
e.getMessage());
|
|
|
|
page = 1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
String sizeStr = request().getQueryString("size");
|
|
|
|
if (StringUtils.isBlank(sizeStr))
|
|
|
|
{
|
|
|
|
size = 10;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
size = Integer.parseInt(sizeStr);
|
|
|
|
}
|
|
|
|
catch(NumberFormatException e)
|
|
|
|
{
|
2016-03-02 15:06:10 -08:00
|
|
|
Logger.error("Dataset Controller searchByKeyword wrong page parameter. Error message: " +
|
2015-11-19 14:39:21 -08:00
|
|
|
e.getMessage());
|
|
|
|
size = 10;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
result.put("status", "ok");
|
2016-03-02 15:06:10 -08:00
|
|
|
Boolean isDefault = false;
|
2015-11-19 14:39:21 -08:00
|
|
|
if (StringUtils.isBlank(category))
|
|
|
|
{
|
2016-03-02 00:49:15 -08:00
|
|
|
category = "datasets";
|
2015-11-19 14:39:21 -08:00
|
|
|
}
|
2016-05-20 17:59:28 -07:00
|
|
|
if (StringUtils.isBlank(source) || source.equalsIgnoreCase("all") || source.equalsIgnoreCase("default"))
|
2015-11-19 14:39:21 -08:00
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
source = null;
|
2016-03-02 15:06:10 -08:00
|
|
|
}
|
2016-05-20 17:59:28 -07:00
|
|
|
|
|
|
|
String searchEngine = Play.application().configuration().getString(SearchDAO.WHEREHOWS_SEARCH_ENGINE__KEY);
|
|
|
|
|
2015-11-19 14:39:21 -08:00
|
|
|
if (category.toLowerCase().equalsIgnoreCase("metric"))
|
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
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));
|
|
|
|
}
|
2015-11-19 14:39:21 -08:00
|
|
|
}
|
2016-03-02 00:49:15 -08:00
|
|
|
else if (category.toLowerCase().equalsIgnoreCase("flows"))
|
2016-02-29 18:44:39 -08:00
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
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));
|
|
|
|
}
|
2016-02-29 18:44:39 -08:00
|
|
|
}
|
2016-03-02 00:49:15 -08:00
|
|
|
else if (category.toLowerCase().equalsIgnoreCase("jobs"))
|
2016-03-01 14:49:42 -08:00
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
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));
|
|
|
|
}
|
2016-03-01 14:49:42 -08:00
|
|
|
}
|
2015-11-19 14:39:21 -08:00
|
|
|
else if (category.toLowerCase().equalsIgnoreCase("comments"))
|
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
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));
|
|
|
|
}
|
|
|
|
|
2015-11-19 14:39:21 -08:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
if(StringUtils.isNotBlank(searchEngine) && searchEngine.equalsIgnoreCase("elasticsearch"))
|
|
|
|
{
|
|
|
|
result.set("result", SearchDAO.elasticSearchDatasetByKeyword(category, keyword, source, page, size));
|
|
|
|
}
|
|
|
|
else
|
2016-03-02 15:06:10 -08:00
|
|
|
{
|
2016-05-20 17:59:28 -07:00
|
|
|
result.set("result", SearchDAO.getPagedDatasetByKeyword(category, keyword, source, page, size));
|
2016-03-02 15:06:10 -08:00
|
|
|
}
|
2015-11-19 14:39:21 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
return ok(result);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|