private BlockWorkerInfo worker(long capacity, String node, String rack) { WorkerNetAddress address = new WorkerNetAddress(); List<LocalityTier> tiers = new ArrayList<>(); if (node != null && !node.isEmpty()) { address.setHost(node); tiers.add(new LocalityTier(Constants.LOCALITY_NODE, node)); } if (rack != null && !rack.isEmpty()) { tiers.add(new LocalityTier(Constants.LOCALITY_RACK, rack)); } address.setTieredIdentity(new TieredIdentity(tiers)); return new BlockWorkerInfo(address, capacity, 0); } }
private BlockWorkerInfo worker(long capacity, long used, String node, String rack) { WorkerNetAddress address = new WorkerNetAddress(); List<LocalityTier> tiers = new ArrayList<>(); if (node != null && !node.isEmpty()) { address.setHost(node); tiers.add(new LocalityTier(Constants.LOCALITY_NODE, node)); } if (rack != null && !rack.isEmpty()) { tiers.add(new LocalityTier(Constants.LOCALITY_RACK, rack)); } address.setTieredIdentity(new TieredIdentity(tiers)); return new BlockWorkerInfo(address, capacity, used); } }
public static WorkerNetAddress createRandom() { WorkerNetAddress result = new WorkerNetAddress(); Random random = new Random(); String host = CommonUtils.randomAlphaNumString(random.nextInt(10)); int rpcPort = random.nextInt(); int dataPort = random.nextInt(); int webPort = random.nextInt(); TieredIdentity identity = TieredIdentityTest.createRandomTieredIdentity(); result.setHost(host); result.setRpcPort(rpcPort); result.setDataPort(dataPort); result.setWebPort(webPort); result.setTieredIdentity(identity); return result; } }
@Override public WorkerNetAddress getAddress() { return new WorkerNetAddress() .setHost(NetworkAddressUtils.getConnectHost(ServiceType.WORKER_RPC, ServerConfiguration.global())) .setRpcPort(mRpcAddress.getPort()) .setDataPort(getDataLocalPort()) .setDomainSocketPath(getDataDomainSocketPath()) .setWebPort(mWebServer.getLocalPort()) .setTieredIdentity(mTieredIdentitiy); }
/** * Converts a proto type to a wire type. * * @param workerNetPAddress the proto type to convert * @return the converted wire type */ public static WorkerNetAddress fromProto(alluxio.grpc.WorkerNetAddress workerNetPAddress) { WorkerNetAddress workerNetAddress = new WorkerNetAddress(); workerNetAddress.setHost(workerNetPAddress.getHost()); workerNetAddress.setRpcPort(workerNetPAddress.getRpcPort()); workerNetAddress.setDataPort(workerNetPAddress.getDataPort()); workerNetAddress.setWebPort(workerNetPAddress.getWebPort()); workerNetAddress.setDomainSocketPath(workerNetPAddress.getDomainSocketPath()); workerNetAddress.setTieredIdentity(fromProto(workerNetPAddress.getTieredIdentity())); return workerNetAddress; }
BlockWorkerInfo workerInfo = new BlockWorkerInfo(new WorkerNetAddress().setHost("localhost") .setTieredIdentity(TieredIdentityFactory.fromString("node=localhost", sConf)) .setRpcPort(1).setDataPort(2).setWebPort(3), Constants.GB, 0); when(mBlockStore.getEligibleWorkers()).thenReturn(Lists.newArrayList(workerInfo));
/** * Creates a new instance of {@link WorkerNetAddress} from thrift representation. * * @param address the thrift net address * @return the instance */ public static WorkerNetAddress fromThrift(alluxio.thrift.WorkerNetAddress address) { TieredIdentity tieredIdentity = TieredIdentity.fromThrift(address.getTieredIdentity()); if (tieredIdentity == null) { // This means the worker is pre-1.7.0. We handle this in post-1.7.0 clients by filling out // the tiered identity using the hostname field. tieredIdentity = new TieredIdentity( Arrays.asList(new LocalityTier(Constants.LOCALITY_NODE, address.getHost()))); } return new WorkerNetAddress() .setDataPort(address.getDataPort()) .setDomainSocketPath(address.getDomainSocketPath()) .setHost(address.getHost()) .setRpcPort(address.getRpcPort()) .setTieredIdentity(tieredIdentity) .setWebPort(address.getWebPort()); }