depends/libyarn/src/protocolrecords/RegisterApplicationMasterResponse.cpp

/*
 * 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.
 */

#include "RegisterApplicationMasterResponse.h"

namespace libyarn {

RegisterApplicationMasterResponse::RegisterApplicationMasterResponse() {
	responseProto = RegisterApplicationMasterResponseProto::default_instance();
}

RegisterApplicationMasterResponse::RegisterApplicationMasterResponse(
		const RegisterApplicationMasterResponseProto &proto) :
		responseProto(proto) {
}

RegisterApplicationMasterResponse::~RegisterApplicationMasterResponse() {
}

RegisterApplicationMasterResponseProto& RegisterApplicationMasterResponse::getProto() {
	return responseProto;
}

void RegisterApplicationMasterResponse::setMaximumResourceCapability(
		Resource &capability) {
	ResourceProto* rProto = new ResourceProto();
	rProto->CopyFrom(capability.getProto());
	responseProto.set_allocated_maximumcapability(rProto);
}

Resource RegisterApplicationMasterResponse::getMaximumResourceCapability() {
	return Resource(responseProto.maximumcapability());
}

void RegisterApplicationMasterResponse::setClientToAMTokenMasterKey(
		string &key) {
	responseProto.set_client_to_am_token_master_key(key);
}

string RegisterApplicationMasterResponse::getClientToAMTokenMasterKey() {
	return responseProto.client_to_am_token_master_key();
}

void RegisterApplicationMasterResponse::setApplicationACLs(
		list<ApplicationACLMap> &aclMapList) {
	list<ApplicationACLMap>::iterator it = aclMapList.begin();
	for (; it != aclMapList.end(); it++) {
		ApplicationACLMapProto* aclMapProto =
				responseProto.add_application_acls();
		aclMapProto->CopyFrom((*it).getProto());
	}
}

list<ApplicationACLMap> RegisterApplicationMasterResponse::getApplicationACLs() {
	list<ApplicationACLMap> list;
	for (int i = 0; i < responseProto.application_acls_size(); i++) {
		list.push_back(ApplicationACLMap(responseProto.application_acls(i)));
	}
	return list;
}

} /* namespace libyarn */