public class UpdateNodeResourceRequestPBImpl extends UpdateNodeResourceRequest { UpdateNodeResourceRequestProto proto = UpdateNodeResourceRequestProto.getDefaultInstance(); UpdateNodeResourceRequestProto.Builder builder = null; boolean viaProto = false; Map<NodeId, ResourceOption> nodeResourceMap = null; public UpdateNodeResourceRequestPBImpl() { builder = UpdateNodeResourceRequestProto.newBuilder(); } public UpdateNodeResourceRequestPBImpl(UpdateNodeResourceRequestProto proto) { this.proto = proto; viaProto = true; } @Override public Map<NodeId, ResourceOption> getNodeResourceMap() { initNodeResourceMap(); return this.nodeResourceMap; } @Override public void setNodeResourceMap(Map<NodeId, ResourceOption> nodeResourceMap) { if (nodeResourceMap == null) { return; } initNodeResourceMap(); this.nodeResourceMap.clear(); this.nodeResourceMap.putAll(nodeResourceMap); } public UpdateNodeResourceRequestProto getProto() { mergeLocalToProto(); proto = viaProto ? proto : builder.build(); viaProto = true; return proto; } private void mergeLocalToBuilder() { if (this.nodeResourceMap != null) { addNodeResourceMap(); } } private void mergeLocalToProto() { if (viaProto) maybeInitBuilder(); mergeLocalToBuilder(); proto = builder.build(); viaProto = true; } 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())); } } private void maybeInitBuilder() { if (viaProto || builder == null) { builder = UpdateNodeResourceRequestProto.newBuilder(proto); } viaProto = false; } private NodeIdProto convertToProtoFormat(NodeId nodeId) { return ((NodeIdPBImpl) nodeId).getProto(); } private NodeId convertFromProtoFormat(NodeIdProto proto) { return new NodeIdPBImpl(proto); } private ResourceOptionPBImpl convertFromProtoFormat(ResourceOptionProto c) { return new ResourceOptionPBImpl(c); } private ResourceOptionProto convertToProtoFormat(ResourceOption c) { return ((ResourceOptionPBImpl) c).getProto(); } 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); } @Override public int hashCode() { return getProto().hashCode(); } @Override public boolean equals(Object other) { if (other == null) return false; if (other.getClass().isAssignableFrom(this.getClass())) { return this.getProto().equals(this.getClass().cast(other).getProto()); } return false; } }
public UpdateNodeResourceRequestPBImpl() { builder = UpdateNodeResourceRequestProto.newBuilder(); }
private void maybeInitBuilder() { if (viaProto || builder == null) { builder = UpdateNodeResourceRequestProto.newBuilder(proto); } viaProto = false; }