/** * 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.protocolrecords.impl.pb; import org.apache.hadoop.mapreduce.v2.api.protocolrecords.FailTaskAttemptRequest; import org.apache.hadoop.mapreduce.v2.api.records.TaskAttemptId; import org.apache.hadoop.mapreduce.v2.api.records.impl.pb.TaskAttemptIdPBImpl; import org.apache.hadoop.mapreduce.v2.proto.MRProtos.TaskAttemptIdProto; import org.apache.hadoop.mapreduce.v2.proto.MRServiceProtos.FailTaskAttemptRequestProto; import org.apache.hadoop.mapreduce.v2.proto.MRServiceProtos.FailTaskAttemptRequestProtoOrBuilder; import org.apache.hadoop.yarn.api.records.ProtoBase; public class FailTaskAttemptRequestPBImpl extends ProtoBase<FailTaskAttemptRequestProto> implements FailTaskAttemptRequest { FailTaskAttemptRequestProto proto = FailTaskAttemptRequestProto.getDefaultInstance(); FailTaskAttemptRequestProto.Builder builder = null; boolean viaProto = false; private TaskAttemptId taskAttemptId = null; public FailTaskAttemptRequestPBImpl() { builder = FailTaskAttemptRequestProto.newBuilder(); } public FailTaskAttemptRequestPBImpl(FailTaskAttemptRequestProto proto) { this.proto = proto; viaProto = true; } public FailTaskAttemptRequestProto getProto() { mergeLocalToProto(); proto = viaProto ? proto : builder.build(); viaProto = true; return proto; } private void mergeLocalToBuilder() { if (this.taskAttemptId != null) { builder.setTaskAttemptId(convertToProtoFormat(this.taskAttemptId)); } } private void mergeLocalToProto() { if (viaProto) maybeInitBuilder(); mergeLocalToBuilder(); proto = builder.build(); viaProto = true; } private void maybeInitBuilder() { if (viaProto || builder == null) { builder = FailTaskAttemptRequestProto.newBuilder(proto); } viaProto = false; } @Override public TaskAttemptId getTaskAttemptId() { FailTaskAttemptRequestProtoOrBuilder p = viaProto ? proto : builder; if (this.taskAttemptId != null) { return this.taskAttemptId; } if (!p.hasTaskAttemptId()) { return null; } this.taskAttemptId = convertFromProtoFormat(p.getTaskAttemptId()); return this.taskAttemptId; } @Override public void setTaskAttemptId(TaskAttemptId taskAttemptId) { maybeInitBuilder(); if (taskAttemptId == null) builder.clearTaskAttemptId(); this.taskAttemptId = taskAttemptId; } private TaskAttemptIdPBImpl convertFromProtoFormat(TaskAttemptIdProto p) { return new TaskAttemptIdPBImpl(p); } private TaskAttemptIdProto convertToProtoFormat(TaskAttemptId t) { return ((TaskAttemptIdPBImpl)t).getProto(); } }