setVirtualCores(3).addResourceValueMap(riProto).build(); Resource res = new ResourcePBImpl(proto); setVirtualCores(3).addResourceValueMap(riProto1).build(); Resource res1 = new ResourcePBImpl(proto1); setVirtualCores(3).addResourceValueMap(riProto2).build(); Resource res2 = new ResourcePBImpl(proto2);
synchronized private void mergeLocalToBuilder() { builder.clearResourceValueMap(); if (resources != null && resources.length != 0) { for (ResourceInformation resInfo : resources) { ResourceInformationProto.Builder e = ResourceInformationProto .newBuilder(); e.setKey(resInfo.getName()); e.setUnits(resInfo.getUnits()); e.setType(ProtoUtils.converToProtoFormat(resInfo.getResourceType())); e.setValue(resInfo.getValue()); builder.addResourceValueMap(e); } } builder.setMemory(this.getMemorySize()); builder.setVirtualCores(this.getVirtualCores()); }
@Test public void testResourcePBInitFromOldPB() throws Exception { YarnProtos.ResourceProto proto = YarnProtos.ResourceProto.newBuilder().setMemory(1024).setVirtualCores(3) .build(); // Assert to check it sets resource value and unit to default. Resource res = new ResourcePBImpl(proto); Assert.assertEquals(1024, res.getMemorySize()); Assert.assertEquals(3, res.getVirtualCores()); Assert.assertEquals(ResourceInformation.MEMORY_MB.getUnits(), res.getResourceInformation(ResourceInformation.MEMORY_MB.getName()) .getUnits()); Assert.assertEquals(ResourceInformation.VCORES.getUnits(), res.getResourceInformation(ResourceInformation.VCORES.getName()) .getUnits()); }
public Builder mergeFrom(org.apache.hadoop.yarn.proto.YarnProtos.ResourceProto other) { if (other == org.apache.hadoop.yarn.proto.YarnProtos.ResourceProto.getDefaultInstance()) return this; if (other.hasMemory()) { setMemory(other.getMemory()); } if (other.hasVirtualCores()) { setVirtualCores(other.getVirtualCores()); } this.mergeUnknownFields(other.getUnknownFields()); return this; }
@Override public void setVirtualCores(int vCores) { maybeInitBuilder(); builder.setVirtualCores((vCores)); }
public static ResourceProto convertToProtoFormat(Resource e) { return YarnProtos.ResourceProto.newBuilder() .setMemory(e.getMemorySize()) .setVirtualCores(e.getVirtualCores()) .build(); }
@Override public void setVirtualCores(int vCores) { maybeInitBuilder(); builder.setVirtualCores((vCores)); }