hadoop AMInfoPBImpl 源码

haddop AMInfoPBImpl 代码

文件路径：/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-common/src/main/java/org/apache/hadoop/mapreduce/v2/api/records/impl/pb/AMInfoPBImpl.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.hadoop.mapreduce.v2.api.records.impl.pb;

import org.apache.hadoop.mapreduce.v2.api.records.AMInfo;
import org.apache.hadoop.mapreduce.v2.proto.MRProtos.AMInfoProto;
import org.apache.hadoop.mapreduce.v2.proto.MRProtos.AMInfoProtoOrBuilder;
import org.apache.hadoop.yarn.api.records.ApplicationAttemptId;
import org.apache.hadoop.yarn.api.records.ContainerId;
import org.apache.hadoop.yarn.api.records.impl.pb.ApplicationAttemptIdPBImpl;
import org.apache.hadoop.yarn.api.records.impl.pb.ContainerIdPBImpl;
import org.apache.hadoop.yarn.api.records.impl.pb.ProtoBase;
import org.apache.hadoop.yarn.proto.YarnProtos.ApplicationAttemptIdProto;
import org.apache.hadoop.yarn.proto.YarnProtos.ContainerIdProto;

public class AMInfoPBImpl extends ProtoBase<AMInfoProto> implements AMInfo {

  AMInfoProto proto = AMInfoProto.getDefaultInstance();
  AMInfoProto.Builder builder = null;
  boolean viaProto = false;
  
  private ApplicationAttemptId appAttemptId;
  private ContainerId containerId;
  
  public AMInfoPBImpl() {
    builder = AMInfoProto.newBuilder();
  }

  public AMInfoPBImpl(AMInfoProto proto) {
    this.proto = proto;
    viaProto = true;
  }

  public synchronized AMInfoProto getProto() {
    mergeLocalToProto();
    proto = viaProto ? proto : builder.build();
    viaProto = true;
    return proto;
  }

  private synchronized void mergeLocalToBuilder() {
    if (this.appAttemptId != null
        && !((ApplicationAttemptIdPBImpl) this.appAttemptId).getProto().equals(
            builder.getApplicationAttemptId())) {
      builder.setApplicationAttemptId(convertToProtoFormat(this.appAttemptId));
    }
    if (this.getContainerId() != null
        && !((ContainerIdPBImpl) this.containerId).getProto().equals(
            builder.getContainerId())) {
      builder.setContainerId(convertToProtoFormat(this.containerId));
    }
  }

  private synchronized void mergeLocalToProto() {
    if (viaProto)
      maybeInitBuilder();
    mergeLocalToBuilder();
    proto = builder.build();
    viaProto = true;
  }

  private synchronized void maybeInitBuilder() {
    if (viaProto || builder == null) {
      builder = AMInfoProto.newBuilder(proto);
    }
    viaProto = false;
  }

  @Override
  public synchronized ApplicationAttemptId getAppAttemptId() {
    AMInfoProtoOrBuilder p = viaProto ? proto : builder;
    if (appAttemptId != null) {
      return appAttemptId;
    } // Else via proto
    if (!p.hasApplicationAttemptId()) {
      return null;
    }
    appAttemptId = convertFromProtoFormat(p.getApplicationAttemptId());
    return appAttemptId;
  }

  @Override
  public synchronized void setAppAttemptId(ApplicationAttemptId appAttemptId) {
    maybeInitBuilder();
    if (appAttemptId == null) {
      builder.clearApplicationAttemptId();
    }
    this.appAttemptId = appAttemptId;
  }

  @Override
  public synchronized long getStartTime() {
    AMInfoProtoOrBuilder p = viaProto ? proto : builder;
    return (p.getStartTime());
  }
  
  @Override
  public synchronized void setStartTime(long startTime) {
    maybeInitBuilder();
    builder.setStartTime(startTime);
  }

  @Override
  public synchronized ContainerId getContainerId() {
    AMInfoProtoOrBuilder p = viaProto ? proto : builder;
    if (containerId != null) {
      return containerId;
    } // Else via proto
    if (!p.hasContainerId()) {
      return null;
    }
    containerId = convertFromProtoFormat(p.getContainerId());
    return containerId;
  }
  
  @Override
  public synchronized void setContainerId(ContainerId containerId) {
    maybeInitBuilder();
    if (containerId == null) {
      builder.clearContainerId();
    }
    this.containerId = containerId;
  }

  @Override
  public synchronized String getNodeManagerHost() {
    AMInfoProtoOrBuilder p = viaProto ? proto : builder;
    if (!p.hasNodeManagerHost()) {
      return null;
    }
    return p.getNodeManagerHost();
  }

  @Override
  public synchronized void setNodeManagerHost(String nmHost) {
    maybeInitBuilder();
    if (nmHost == null) {
      builder.clearNodeManagerHost();
      return;
    }
    builder.setNodeManagerHost(nmHost);
  }

  @Override
  public synchronized int getNodeManagerPort() {
    AMInfoProtoOrBuilder p = viaProto ? proto : builder;
    return (p.getNodeManagerPort());
  }
  
  @Override
  public synchronized void setNodeManagerPort(int nmPort) {
    maybeInitBuilder();
    builder.setNodeManagerPort(nmPort);
  }

  @Override
  public synchronized int getNodeManagerHttpPort() {
    AMInfoProtoOrBuilder p = viaProto ? proto : builder;
    return p.getNodeManagerHttpPort();
  }

  @Override
  public synchronized void setNodeManagerHttpPort(int httpPort) {
    maybeInitBuilder();
    builder.setNodeManagerHttpPort(httpPort);
  }

  private ApplicationAttemptIdPBImpl convertFromProtoFormat(
      ApplicationAttemptIdProto p) {
    return new ApplicationAttemptIdPBImpl(p);
  }

  private ContainerIdPBImpl convertFromProtoFormat(ContainerIdProto p) {
    return new ContainerIdPBImpl(p);
  }

  private
      ApplicationAttemptIdProto convertToProtoFormat(ApplicationAttemptId t) {
    return ((ApplicationAttemptIdPBImpl) t).getProto();
  }

  private ContainerIdProto convertToProtoFormat(ContainerId t) {
    return ((ContainerIdPBImpl) t).getProto();
  }
}

相关信息

hadoop 源码目录

相关文章

hadoop CounterGroupPBImpl 源码

hadoop CounterPBImpl 源码

hadoop CountersPBImpl 源码

hadoop JobIdPBImpl 源码

hadoop JobReportPBImpl 源码

hadoop TaskAttemptCompletionEventPBImpl 源码

hadoop TaskAttemptIdPBImpl 源码

hadoop TaskAttemptReportPBImpl 源码

hadoop TaskIdPBImpl 源码

hadoop TaskReportPBImpl 源码