Examples with InetSocketAddress java.net.InetSocketAddress used on opensource projects

Example 96 with InetSocketAddress

use of java.net.InetSocketAddress in project hadoop by apache.

the class TestServer method testBindError.

@Test
public void testBindError() throws Exception {
    Configuration conf = new Configuration();
    ServerSocket socket = new ServerSocket();
    InetSocketAddress address = new InetSocketAddress("0.0.0.0", 0);
    socket.bind(address);
    try {
        int min = socket.getLocalPort();
        conf.set("TestRange", min + "-" + min);
        ServerSocket socket2 = new ServerSocket();
        InetSocketAddress address2 = new InetSocketAddress("0.0.0.0", 0);
        boolean caught = false;
        try {
            Server.bind(socket2, address2, 10, conf, "TestRange");
        } catch (BindException e) {
            caught = true;
        } finally {
            socket2.close();
        }
        assertTrue("Failed to catch the expected bind exception", caught);
    } finally {
        socket.close();
    }
}

Example 97 with InetSocketAddress

use of java.net.InetSocketAddress in project hadoop by apache.

the class TestIPC method testRTEDuringConnectionSetup.

/**
   * Test that, if a RuntimeException is thrown after creating a socket
   * but before successfully connecting to the IPC server, that the
   * failure is handled properly. This is a regression test for
   * HADOOP-7428.
   */
@Test(timeout = 60000)
public void testRTEDuringConnectionSetup() throws IOException {
    // Set up a socket factory which returns sockets which
    // throw an RTE when setSoTimeout is called.
    SocketFactory spyFactory = spy(NetUtils.getDefaultSocketFactory(conf));
    Mockito.doAnswer(new Answer<Socket>() {

        @Override
        public Socket answer(InvocationOnMock invocation) throws Throwable {
            Socket s = spy((Socket) invocation.callRealMethod());
            doThrow(new RuntimeException("Injected fault")).when(s).setSoTimeout(anyInt());
            return s;
        }
    }).when(spyFactory).createSocket();
    Server server = new TestServer(1, true);
    Client client = new Client(LongWritable.class, conf, spyFactory);
    server.start();
    try {
        // Call should fail due to injected exception.
        InetSocketAddress address = NetUtils.getConnectAddress(server);
        try {
            call(client, RANDOM.nextLong(), address, conf);
            fail("Expected an exception to have been thrown");
        } catch (Exception e) {
            LOG.info("caught expected exception", e);
            assertTrue(StringUtils.stringifyException(e).contains("Injected fault"));
        }
        // Resetting to the normal socket behavior should succeed
        // (i.e. it should not have cached a half-constructed connection)
        Mockito.reset(spyFactory);
        call(client, RANDOM.nextLong(), address, conf);
    } finally {
        client.stop();
        server.stop();
    }
}

Example 98 with InetSocketAddress

use of java.net.InetSocketAddress in project hadoop by apache.

the class TestIPC method testSocketFactoryException.

/**
   * Test that, if the socket factory throws an IOE, it properly propagates
   * to the client.
   */
@Test(timeout = 60000)
public void testSocketFactoryException() throws IOException {
    SocketFactory mockFactory = mock(SocketFactory.class);
    doThrow(new IOException("Injected fault")).when(mockFactory).createSocket();
    Client client = new Client(LongWritable.class, conf, mockFactory);
    InetSocketAddress address = new InetSocketAddress("127.0.0.1", 10);
    try {
        call(client, RANDOM.nextLong(), address, conf);
        fail("Expected an exception to have been thrown");
    } catch (IOException e) {
        assertTrue(e.getMessage().contains("Injected fault"));
    } finally {
        client.stop();
    }
}

Example 99 with InetSocketAddress

use of java.net.InetSocketAddress in project hadoop by apache.

the class RPCCallBenchmark method createRpcClient.

/**
   * Create a client proxy for the specified engine.
   */
private RpcServiceWrapper createRpcClient(MyOptions opts) throws IOException {
    InetSocketAddress addr = NetUtils.createSocketAddr(opts.host, opts.getPort());
    if (opts.rpcEngine == ProtobufRpcEngine.class) {
        final TestRpcService proxy = RPC.getProxy(TestRpcService.class, 0, addr, conf);
        return new RpcServiceWrapper() {

            @Override
            public String doEcho(String msg) throws Exception {
                EchoRequestProto req = EchoRequestProto.newBuilder().setMessage(msg).build();
                EchoResponseProto responseProto = proxy.echo(null, req);
                return responseProto.getMessage();
            }
        };
    } else {
        throw new RuntimeException("unsupported engine: " + opts.rpcEngine);
    }
}

Example 100 with InetSocketAddress

use of java.net.InetSocketAddress in project hadoop by apache.

the class TestAsyncIPC method internalTestAsyncCallLimit.

public void internalTestAsyncCallLimit(int handlerCount, boolean handlerSleep, int clientCount, int callerCount, int callCount) throws IOException, InterruptedException, ExecutionException {
    Configuration conf = new Configuration();
    conf.setInt(CommonConfigurationKeys.IPC_CLIENT_ASYNC_CALLS_MAX_KEY, 100);
    Client.setPingInterval(conf, TestIPC.PING_INTERVAL);
    Server server = new TestIPC.TestServer(handlerCount, handlerSleep, conf);
    InetSocketAddress addr = NetUtils.getConnectAddress(server);
    server.start();
    Client[] clients = new Client[clientCount];
    for (int i = 0; i < clientCount; i++) {
        clients[i] = new Client(LongWritable.class, conf);
    }
    AsyncLimitlCaller[] callers = new AsyncLimitlCaller[callerCount];
    for (int i = 0; i < callerCount; i++) {
        callers[i] = new AsyncLimitlCaller(i, clients[i % clientCount], addr, callCount);
        callers[i].start();
    }
    for (int i = 0; i < callerCount; i++) {
        callers[i].join();
        callers[i].waitForReturnValues(callers[i].getStart(), callers[i].getCount());
        String msg = String.format("Expected not failed for caller-%d: %s.", i, callers[i]);
        assertFalse(msg, callers[i].failed);
    }
    for (int i = 0; i < clientCount; i++) {
        clients[i].stop();
    }
    server.stop();
}