datahub/wherehows-frontend/app/controllers/api/v2/Dataset.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.v2;

import controllers.Application;
import java.util.Collections;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.math.NumberUtils;
import play.Logger;
import play.libs.F.Promise;
import play.libs.Json;
import play.mvc.Controller;
import play.mvc.Result;
import wherehows.dao.view.DataTypesViewDao;
import wherehows.dao.view.DatasetViewDao;

import static utils.Dataset.*;


public class Dataset extends Controller {

  private static final DatasetViewDao DATASET_VIEW_DAO = Application.DAO_FACTORY.getDatasetViewDao();

  private static final DataTypesViewDao DATA_TYPES_DAO = Application.DAO_FACTORY.getDataTypesViewDao();

  private static final int _dataset_search_page_size = 20;

  private Dataset() {
  }

  public static Promise<Result> listNames() {
    try {
      String platform = request().getQueryString("platform");
      String prefix = request().getQueryString("prefix");
      if (StringUtils.isBlank(platform)) {
        throw new IllegalArgumentException("Missing platform");
      }

      // if prefix is a dataset name, then return itself
      if (listNamePrefixIsDataset(platform, prefix)) {
        return Promise.promise(() -> ok(Json.toJson(Collections.singletonList(prefix))));
      }

      return Promise.promise(
          () -> ok(Json.toJson(DATASET_VIEW_DAO.listNames(platform, getPlatformPrefix(platform, prefix)))));
    } catch (Exception e) {
      Logger.error("Fail to list dataset names/sections", e);
      return Promise.promise(() -> internalServerError("Fetch data Error: " + e.toString()));
    }
  }

  public static Promise<Result> listDatasets() {
    try {
      String platform = request().getQueryString("platform");
      String prefix = request().getQueryString("prefix");

      if (StringUtils.isBlank(platform) || StringUtils.isBlank(prefix)) {
        throw new IllegalArgumentException("Missing platform or prefix");
      }

      int page = NumberUtils.toInt(request().getQueryString("page"), 0);
      int start = page * _dataset_search_page_size;

      return Promise.promise(
          () -> ok(Json.toJson(DATASET_VIEW_DAO.listDatasets(platform, prefix, start, _dataset_search_page_size))));
    } catch (Exception e) {
      Logger.error("Fail to list datasets", e);
      return Promise.promise(() -> internalServerError("Fetch data Error: " + e.toString()));
    }
  }

  public static Promise<Result> totalDatasets() {
    try {
      String platform = request().getQueryString("platform");
      String prefix = StringUtils.defaultIfBlank(request().getQueryString("prefix"), "");

      return Promise.promise(
          () -> ok(String.valueOf(DATASET_VIEW_DAO.listDatasets(platform, prefix, 0, 1).getTotal())));
    } catch (Exception e) {
      Logger.error("Fail to count total datasets", e);
      return Promise.promise(() -> internalServerError("Fetch data Error: " + e.toString()));
    }
  }

  public static Promise<Result> getComplianceDataTypes() {
    try {
      return Promise.promise(() -> ok(
          Json.newObject().set("complianceDataTypes", Json.toJson(DATA_TYPES_DAO.getAllComplianceDataTypes()))));
    } catch (Exception e) {
      Logger.error("Fail to get compliance data types", e);
      return Promise.promise(() -> notFound("Fetch data Error: " + e.toString()));
    }
  }

  public static Promise<Result> getDataPlatforms() {
    try {
      return Promise.promise(
          () -> ok(Json.newObject().set("platforms", Json.toJson(DATA_TYPES_DAO.getAllPlatforms()))));
    } catch (Exception e) {
      Logger.error("Fail to get data platforms", e);
      return Promise.promise(() -> notFound("Fetch data Error: " + e.toString()));
    }
  }
}