Examples with NodeResourceMapProto org.apache.hadoop.yarn.proto.YarnProtos.NodeResourceMapProto used on opensource projects

Example 1 with NodeResourceMapProto

use of org.apache.hadoop.yarn.proto.YarnProtos.NodeResourceMapProto in project hadoop by apache.

the class UpdateNodeResourceRequestPBImpl method addNodeResourceMap.

private void addNodeResourceMap() {
    maybeInitBuilder();
    builder.clearNodeResourceMap();
    if (nodeResourceMap == null) {
        return;
    }
    Iterable<? extends NodeResourceMapProto> values = new Iterable<NodeResourceMapProto>() {

        @Override
        public Iterator<NodeResourceMapProto> iterator() {
            return new Iterator<NodeResourceMapProto>() {

                Iterator<NodeId> nodeIterator = nodeResourceMap.keySet().iterator();

                @Override
                public boolean hasNext() {
                    return nodeIterator.hasNext();
                }

                @Override
                public NodeResourceMapProto next() {
                    NodeId nodeId = nodeIterator.next();
                    return NodeResourceMapProto.newBuilder().setNodeId(convertToProtoFormat(nodeId)).setResourceOption(convertToProtoFormat(nodeResourceMap.get(nodeId))).build();
                }

                @Override
                public void remove() {
                    throw new UnsupportedOperationException();
                }
            };
        }
    };
    this.builder.addAllNodeResourceMap(values);
}

Example 2 with NodeResourceMapProto

use of org.apache.hadoop.yarn.proto.YarnProtos.NodeResourceMapProto in project hadoop by apache.

the class UpdateNodeResourceRequestPBImpl method initNodeResourceMap.

private void initNodeResourceMap() {
    if (this.nodeResourceMap != null) {
        return;
    }
    UpdateNodeResourceRequestProtoOrBuilder p = viaProto ? proto : builder;
    List<NodeResourceMapProto> list = p.getNodeResourceMapList();
    this.nodeResourceMap = new HashMap<NodeId, ResourceOption>(list.size());
    for (NodeResourceMapProto nodeResourceProto : list) {
        this.nodeResourceMap.put(convertFromProtoFormat(nodeResourceProto.getNodeId()), convertFromProtoFormat(nodeResourceProto.getResourceOption()));
    }
}