195 lines
6.1 KiB
Java
Raw Normal View History

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;
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);
}
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()
2015-11-19 14:39:21 -08:00
{
ObjectNode result = Json.newObject();
int page = 1;
int size = 10;
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))
{
category = "datasets";
2015-11-19 14:39:21 -08:00
}
if (StringUtils.isBlank(source) || source.equalsIgnoreCase("all") || source.equalsIgnoreCase("default"))
2015-11-19 14:39:21 -08:00
{
source = null;
2016-03-02 15:06:10 -08: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"))
{
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
}
else if (category.toLowerCase().equalsIgnoreCase("flows"))
2016-02-29 18:44:39 -08: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
}
else if (category.toLowerCase().equalsIgnoreCase("jobs"))
2016-03-01 14:49:42 -08: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"))
{
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
{
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
{
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);
}
}