private void mergeLocalToBuilder() { if (this.attributesToNodes != null) { addAttributesToNodesToProto(); } }
@Override public Map<NodeAttributeKey, List<NodeToAttributeValue>> getAttributesToNodes() { initAttributesToNodes(); return this.attributesToNodes; } }
private void mergeLocalToProto() { if (viaProto) { maybeInitBuilder(); } mergeLocalToBuilder(); proto = builder.build(); viaProto = true; }
public GetAttributesToNodesResponseProto getProto() { mergeLocalToProto(); proto = viaProto ? proto : builder.build(); viaProto = true; return proto; }
private void initAttributesToNodes() { if (this.attributesToNodes != null) { return; } YarnServiceProtos.GetAttributesToNodesResponseProtoOrBuilder p = viaProto ? proto : builder; List<AttributeToNodesProto> list = p.getAttributesToNodesList(); this.attributesToNodes = new HashMap<>(); for (AttributeToNodesProto c : list) { List<NodeToAttributeValueProto> nodeValueMapList = c.getNodeValueMapList(); List<NodeToAttributeValue> nodeToAttributeValue = new ArrayList<>(); for (NodeToAttributeValueProto valueProto : nodeValueMapList) { nodeToAttributeValue.add(convertFromProtoFormat(valueProto)); } if (!nodeToAttributeValue.isEmpty()) { this.attributesToNodes.put(convertFromProtoFormat(c.getNodeAttribute()), nodeToAttributeValue); } } }
@Override public GetAttributesToNodesResponse getAttributesToNodes( GetAttributesToNodesRequest request) throws YarnException, IOException { YarnServiceProtos.GetAttributesToNodesRequestProto requestProto = ((GetAttributesToNodesRequestPBImpl) request).getProto(); try { return new GetAttributesToNodesResponsePBImpl( proxy.getAttributesToNodes(null, requestProto)); } catch (ServiceException e) { RPCUtil.unwrapAndThrowException(e); return null; } }
private void addAttributesToNodesToProto() { maybeInitBuilder(); builder.clearAttributesToNodes(); if (attributesToNodes == null) {
@Override public void setAttributeToNodes( Map<NodeAttributeKey, List<NodeToAttributeValue>> map) { initAttributesToNodes(); attributesToNodes.clear(); attributesToNodes.putAll(map); }
@Override public GetAttributesToNodesResponseProto getAttributesToNodes( RpcController controller, YarnServiceProtos.GetAttributesToNodesRequestProto proto) throws ServiceException { GetAttributesToNodesRequestPBImpl req = new GetAttributesToNodesRequestPBImpl(proto); try { GetAttributesToNodesResponse resp = real.getAttributesToNodes(req); return ((GetAttributesToNodesResponsePBImpl) resp).getProto(); } catch (YarnException ye) { throw new ServiceException(ye); } catch (IOException ie) { throw new ServiceException(ie); } }