datahub/web/app/controllers/api/v1/Search.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.SearchDAO;
import models.DatasetColumn;
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(String keyword)
    {
        ObjectNode result = Json.newObject();

        int page = 1;
        int size = 10;
        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 getPagedDatasets 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 getPagedDatasets wrong page parameter. Error message: " +
                        e.getMessage());
                size = 10;
            }
        }

        result.put("status", "ok");
        if (StringUtils.isBlank(category))
        {
            category = "dataset";
        }
        if (StringUtils.isBlank(source))
        {
            source = "all";
        }
        if (category.toLowerCase().equalsIgnoreCase("metric"))
        {
            result.set("result", SearchDAO.getPagedMetricByKeyword(keyword, page, size));
        }
        else if (category.toLowerCase().equalsIgnoreCase("flow"))
        {
            result.set("result", SearchDAO.getPagedFlowByKeyword(keyword, page, size));
        }
        else if (category.toLowerCase().equalsIgnoreCase("comments"))
        {
            result.set("result", SearchDAO.getPagedCommentsByKeyword(keyword, page, size));
        }
        else
        {
            result.set("result", SearchDAO.getPagedDatasetByKeyword(keyword, source, page, size));
        }

        return ok(result);
    }


}