hadoop-submarine/hadoop-submarine-core/src/main/java/org/apache/hadoop/yarn/submarine/common/ClientContext.java

/**
 * 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.
 * See the License for the specific language governing permissions and
 * limitations under the License. See accompanying LICENSE file.
 */

package org.apache.hadoop.yarn.submarine.common;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.yarn.client.api.YarnClient;
import org.apache.hadoop.yarn.conf.YarnConfiguration;
import org.apache.hadoop.yarn.submarine.common.conf.SubmarineConfiguration;
import org.apache.hadoop.yarn.submarine.common.fs.DefaultRemoteDirectoryManager;
import org.apache.hadoop.yarn.submarine.common.fs.RemoteDirectoryManager;
import org.apache.hadoop.yarn.submarine.runtimes.RuntimeFactory;

public class ClientContext {
  private Configuration yarnConf = new YarnConfiguration();

  private volatile RemoteDirectoryManager remoteDirectoryManager;
  private YarnClient yarnClient;
  private Configuration submarineConfig;
  private RuntimeFactory runtimeFactory;

  public ClientContext() {
    submarineConfig = new SubmarineConfiguration();
  }

  public synchronized YarnClient getOrCreateYarnClient() {
    if (yarnClient == null) {
      yarnClient = YarnClient.createYarnClient();
      yarnClient.init(yarnConf);
      yarnClient.start();
    }
    return yarnClient;
  }

  public Configuration getYarnConfig() {
    return yarnConf;
  }

  public void setConfiguration(Configuration conf) {
    this.yarnConf = conf;
  }

  public RemoteDirectoryManager getRemoteDirectoryManager() {
    if(remoteDirectoryManager == null) {
      synchronized (this) {
        if(remoteDirectoryManager == null) {
          remoteDirectoryManager = new DefaultRemoteDirectoryManager(this);
        }
      }
    }
    return remoteDirectoryManager;
  }

  public Configuration getSubmarineConfig() {
    return submarineConfig;
  }

  public void setSubmarineConfig(Configuration submarineConfig) {
    this.submarineConfig = submarineConfig;
  }

  public RuntimeFactory getRuntimeFactory() {
    return runtimeFactory;
  }

  public void setRuntimeFactory(RuntimeFactory runtimeFactory) {
    this.runtimeFactory = runtimeFactory;
  }
}