heron/spi/src/java/org/apache/heron/spi/common/Context.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

package org.apache.heron.spi.common;

import java.util.Map;

import org.apache.heron.common.basics.ByteAmount;
import org.apache.heron.common.basics.DryRunFormatType;
import org.apache.heron.common.basics.PackageType;

public class Context {

  protected Context() {
  }

  public static String cluster(Config cfg) {
    return cfg.getStringValue(Key.CLUSTER);
  }

  public static String role(Config cfg) {
    return cfg.getStringValue(Key.ROLE);
  }

  public static String environ(Config cfg) {
    return cfg.getStringValue(Key.ENVIRON);
  }

  public static String submitUser(Config cfg) {
    return cfg.getStringValue(Key.SUBMIT_USER);
  }

  public static Boolean dryRun(Config cfg) {
    return cfg.getBooleanValue(Key.DRY_RUN);
  }

  public static DryRunFormatType dryRunFormatType(Config cfg) {
    return cfg.getDryRunFormatType(Key.DRY_RUN_FORMAT_TYPE);
  }

  public static Boolean verbose(Config cfg) {
    return cfg.getBooleanValue(Key.VERBOSE);
  }

  public static Boolean verboseGC(Config cfg) {
    return cfg.getBooleanValue(Key.VERBOSE_GC);
  }

  public static String buildVersion(Config cfg) {
    return cfg.getStringValue(Key.BUILD_VERSION);
  }

  public static String buildTime(Config cfg) {
    return cfg.getStringValue(Key.BUILD_TIME);
  }

  public static Long buildTimeStamp(Config cfg) {
    return cfg.getLongValue(Key.BUILD_TIMESTAMP);
  }

  public static String buildHost(Config cfg) {
    return cfg.getStringValue(Key.BUILD_HOST);
  }

  public static String buildUser(Config cfg) {
    return cfg.getStringValue(Key.BUILD_USER);
  }

  public static String topologyName(Config cfg) {
    return cfg.getStringValue(Key.TOPOLOGY_NAME);
  }

  public static int topologyContainerId(Config cfg) {
    return cfg.getIntegerValue(Key.TOPOLOGY_CONTAINER_ID);
  }

  public static String uploaderClass(Config cfg) {
    return cfg.getStringValue(Key.UPLOADER_CLASS);
  }

  public static String launcherClass(Config cfg) {
    return cfg.getStringValue(Key.LAUNCHER_CLASS);
  }

  public static String schedulerClass(Config cfg) {
    return cfg.getStringValue(Key.SCHEDULER_CLASS);
  }

  public static String packingClass(Config cfg) {
    return cfg.getStringValue(Key.PACKING_CLASS);
  }

  public static String repackingClass(Config cfg) {
    return cfg.getStringValue(Key.REPACKING_CLASS);
  }

  public static String stateManagerClass(Config cfg) {
    return cfg.getStringValue(Key.STATE_MANAGER_CLASS);
  }

  public static Boolean schedulerService(Config cfg) {
    return cfg.getBooleanValue(Key.SCHEDULER_IS_SERVICE);
  }

  public static String clusterFile(Config cfg) {
    return cfg.getStringValue(Key.CLUSTER_YAML);
  }

  public static String statefulConfigFile(Config cfg) {
    return cfg.getStringValue(Key.STATEFUL_YAML);
  }

  public static String clientFile(Config cfg) {
    return cfg.getStringValue(Key.CLIENT_YAML);
  }

  public static String healthmgrFile(Config cfg) {
    return cfg.getStringValue(Key.HEALTHMGR_YAML);
  }

  public static String metricsSinksFile(Config cfg) {
    return cfg.getStringValue(Key.METRICS_YAML);
  }

  public static String packingFile(Config cfg) {
    return cfg.getStringValue(Key.PACKING_YAML);
  }

  public static String schedulerFile(Config cfg) {
    return cfg.getStringValue(Key.SCHEDULER_YAML);
  }

  public static String stateManagerFile(Config cfg) {
    return cfg.getStringValue(Key.STATEMGR_YAML);
  }

  public static String systemFile(Config cfg) {
    return cfg.getStringValue(Key.SYSTEM_YAML);
  }

  public static String uploaderFile(Config cfg) {
    return cfg.getStringValue(Key.UPLOADER_YAML);
  }

  public static String downloaderFile(Config cfg) {
    return cfg.getStringValue(Key.DOWNLOADER_YAML);
  }

  public static String schedulerJar(Config cfg) {
    return cfg.getStringValue(Key.SCHEDULER_JAR);
  }

  public static String schedulerProxyConnectionString(Config cfg) {
    return cfg.getStringValue(Key.SCHEDULER_PROXY_CONNECTION_STRING);
  }

  public static String schedulerProxyConnectionType(Config cfg) {
    return cfg.getStringValue(Key.SCHEDULER_PROXY_CONNECTION_TYPE);
  }

  public static String stateManagerConnectionString(Config cfg) {
    return cfg.getStringValue(Key.STATEMGR_CONNECTION_STRING);
  }

  public static String stateManagerRootPath(Config cfg) {
    return cfg.getStringValue(Key.STATEMGR_ROOT_PATH);
  }

  public static String corePackageUri(Config cfg) {
    return cfg.getStringValue(Key.CORE_PACKAGE_URI);
  }

  public static String corePackageDirectory(Config cfg) {
    return cfg.getStringValue(Key.CORE_PACKAGE_DIRECTORY);
  }

  public static Boolean useCorePackageUri(Config cfg) {
    return cfg.getBooleanValue(Key.USE_CORE_PACKAGE_URI);
  }

  public static String healthMgrMode(Config cfg) {
    return cfg.getStringValue(Key.HEALTHMGR_MODE);
  }

  public static String systemConfigFile(Config cfg) {
    return cfg.getStringValue(Key.SYSTEM_YAML);
  }

  public static String topologyDefinitionFile(Config cfg) {
    return cfg.getStringValue(Key.TOPOLOGY_DEFINITION_FILE);
  }

  public static String topologyBinaryFile(Config cfg) {
    return cfg.getStringValue(Key.TOPOLOGY_BINARY_FILE);
  }

  public static String topologyPackageFile(Config cfg) {
    return cfg.getStringValue(Key.TOPOLOGY_PACKAGE_FILE);
  }

  public static PackageType topologyPackageType(Config cfg) {
    return cfg.getPackageType(Key.TOPOLOGY_PACKAGE_TYPE);
  }

  public static ByteAmount stmgrRam(Config cfg) {
    return cfg.getByteAmountValue(Key.STMGR_RAM);
  }

  public static ByteAmount ckptmgrRam(Config cfg) {
    return cfg.getByteAmountValue(Key.CKPTMGR_RAM);
  }

  public static ByteAmount metricsmgrRam(Config cfg) {
    return cfg.getByteAmountValue(Key.METRICSMGR_RAM);
  }

  public static ByteAmount instanceRam(Config cfg) {
    return cfg.getByteAmountValue(Key.INSTANCE_RAM);
  }

  public static Double instanceCpu(Config cfg) {
    return cfg.getDoubleValue(Key.INSTANCE_CPU);
  }

  public static ByteAmount instanceDisk(Config cfg) {
    return cfg.getByteAmountValue(Key.INSTANCE_DISK);
  }

  public static String heronHome(Config cfg) {
    return cfg.getStringValue(Key.HERON_HOME);
  }

  public static String heronBin(Config cfg) {
    return cfg.getStringValue(Key.HERON_BIN);
  }

  public static String heronConf(Config cfg) {
    return cfg.getStringValue(Key.HERON_CONF);
  }

  public static String heronLib(Config cfg) {
    return cfg.getStringValue(Key.HERON_LIB);
  }

  public static String heronDist(Config cfg) {
    return cfg.getStringValue(Key.HERON_DIST);
  }

  public static String heronEtc(Config cfg) {
    return cfg.getStringValue(Key.HERON_ETC);
  }

  public static String instanceClassPath(Config cfg) {
    return cfg.getStringValue(Key.INSTANCE_CLASSPATH);
  }

  public static String healthMgrClassPath(Config cfg) {
    return cfg.getStringValue(Key.HEALTHMGR_CLASSPATH);
  }

  public static String metricsManagerClassPath(Config cfg) {
    return cfg.getStringValue(Key.METRICSMGR_CLASSPATH);
  }

  public static String metricsCacheManagerClassPath(Config cfg) {
    return cfg.getStringValue(Key.METRICSCACHEMGR_CLASSPATH);
  }

  public static String packingClassPath(Config cfg) {
    return cfg.getStringValue(Key.PACKING_CLASSPATH);
  }

  public static String schedulerClassPath(Config cfg) {
    return cfg.getStringValue(Key.SCHEDULER_CLASSPATH);
  }

  public static String ckptmgrClassPath(Config cfg) {
    return cfg.getStringValue(Key.CKPTMGR_CLASSPATH);
  }

  public static String statefulStoragesClassPath(Config cfg) {
    return cfg.getStringValue(Key.STATEFULSTORAGE_CLASSPATH);
  }

  public static String stateManagerClassPath(Config cfg) {
    return cfg.getStringValue(Key.STATEMGR_CLASSPATH);
  }

  public static String uploaderClassPath(Config cfg) {
    return cfg.getStringValue(Key.UPLOADER_CLASSPATH);
  }

  public static String javaHome(Config cfg) {
    return cfg.getStringValue(Key.JAVA_HOME);
  }

  public static String clusterJavaHome(Config cfg) {
    return cfg.getStringValue(Key.HERON_CLUSTER_JAVA_HOME);
  }

  public static String overrideFile(Config cfg) {
    return cfg.getStringValue(Key.OVERRIDE_YAML);
  }

  public static String apiserverOverrideFile(Config cfg) {
    return cfg.getStringValue(Key.APISERVER_OVERRIDE_YAML);
  }

  public static String executorBinary(Config cfg) {
    return cfg.getStringValue(Key.EXECUTOR_BINARY);
  }

  public static String stmgrBinary(Config cfg) {
    return cfg.getStringValue(Key.STMGR_BINARY);
  }

  public static String tmanagerBinary(Config cfg) {
    return cfg.getStringValue(Key.TMANAGER_BINARY);
  }

  public static String shellBinary(Config cfg) {
    return cfg.getStringValue(Key.SHELL_BINARY);
  }

  public static String pythonInstanceBinary(Config cfg) {
    return cfg.getStringValue(Key.PYTHON_INSTANCE_BINARY);
  }

  public static String cppInstanceBinary(Config cfg) {
    return cfg.getStringValue(Key.CPP_INSTANCE_BINARY);
  }

  public static String downloaderBinary(Config cfg) {
    return cfg.getStringValue(Key.DOWNLOADER_BINARY);
  }

  public static String downloaderConf(Config cfg) {
    return cfg.getStringValue(Key.DOWNLOADER_CONF);
  }

  public static String updatePrompt(Config cfg) {
    return cfg.getStringValue(Key.UPDATE_PROMPT);
  }

  @SuppressWarnings("unchecked")
  public static final String statefulStorageCustomClassPath(Config cfg) {
    Map<String, Object> statefulStorageConfig =
        (Map<String, Object>) cfg.get(Key.STATEFUL_STORAGE_CONF);
    if (statefulStorageConfig == null) {
      return "";
    }
    Object o = statefulStorageConfig.get(Key.STATEFUL_STORAGE_CUSTOM_CLASSPATH.value());
    return o == null ? "" : (String) o;
  }

  public static String metricscacheMgrMode(Config cfg) {
    return cfg.getStringValue(Key.METRICSCACHEMGR_MODE);
  }

}