Search in sources :

Example 6 with ServerStats

use of io.grpc.InternalChannelz.ServerStats in project grpc-java by grpc.

the class InternalChannelzTest method serverSocket_eachServerSeparate.

@Test
public void serverSocket_eachServerSeparate() {
    InternalInstrumented<ServerStats> server1 = create();
    InternalInstrumented<ServerStats> server2 = create();
    InternalInstrumented<SocketStats> socket1 = create();
    InternalInstrumented<SocketStats> socket2 = create();
    channelz.addServer(server1);
    channelz.addServer(server2);
    channelz.addServerSocket(server1, socket1);
    channelz.addServerSocket(server2, socket2);
    ServerSocketsList list1 = channelz.getServerSockets(id(server1), /*fromId=*/
    0, /*maxPageSize=*/
    2);
    assertNotNull(list1);
    assertTrue(list1.end);
    assertThat(list1.sockets).containsExactly(socket1);
    ServerSocketsList list2 = channelz.getServerSockets(id(server2), /*fromId=*/
    0, /*maxPageSize=*/
    2);
    assertNotNull(list2);
    assertTrue(list2.end);
    assertThat(list2.sockets).containsExactly(socket2);
}
Also used : ServerSocketsList(io.grpc.InternalChannelz.ServerSocketsList) ServerStats(io.grpc.InternalChannelz.ServerStats) SocketStats(io.grpc.InternalChannelz.SocketStats) Test(org.junit.Test)

Example 7 with ServerStats

use of io.grpc.InternalChannelz.ServerStats in project grpc-java by grpc.

the class InternalChannelzTest method serverSocket.

@Test
public void serverSocket() {
    InternalInstrumented<ServerStats> server = create();
    channelz.addServer(server);
    InternalInstrumented<SocketStats> socket = create();
    assertEmptyServerSocketsPage(id(server), id(socket));
    channelz.addServerSocket(server, socket);
    ServerSocketsList page = channelz.getServerSockets(id(server), id(socket), /*maxPageSize=*/
    1);
    assertNotNull(page);
    assertTrue(page.end);
    assertThat(page.sockets).containsExactly(socket);
    channelz.removeServerSocket(server, socket);
    assertEmptyServerSocketsPage(id(server), id(socket));
}
Also used : ServerSocketsList(io.grpc.InternalChannelz.ServerSocketsList) ServerStats(io.grpc.InternalChannelz.ServerStats) SocketStats(io.grpc.InternalChannelz.SocketStats) Test(org.junit.Test)

Example 8 with ServerStats

use of io.grpc.InternalChannelz.ServerStats in project grpc-java by grpc.

the class InternalChannelzTest method getServers_onePage_multi.

@Test
public void getServers_onePage_multi() {
    InternalInstrumented<ServerStats> server1 = create();
    InternalInstrumented<ServerStats> server2 = create();
    channelz.addServer(server1);
    channelz.addServer(server2);
    ServerList page = channelz.getServers(/*fromId=*/
    0, /*maxPageSize=*/
    2);
    assertTrue(page.end);
    assertThat(page.servers).containsExactly(server1, server2);
}
Also used : ServerStats(io.grpc.InternalChannelz.ServerStats) ServerList(io.grpc.InternalChannelz.ServerList) Test(org.junit.Test)

Example 9 with ServerStats

use of io.grpc.InternalChannelz.ServerStats in project grpc-java by grpc.

the class ChannelzService method getServer.

/**
 * Returns a server.
 */
@Override
public void getServer(GetServerRequest request, StreamObserver<GetServerResponse> responseObserver) {
    InternalInstrumented<ServerStats> s = channelz.getServer(request.getServerId());
    if (s == null) {
        responseObserver.onError(Status.NOT_FOUND.withDescription("Can't find server " + request.getServerId()).asRuntimeException());
        return;
    }
    GetServerResponse resp;
    try {
        resp = GetServerResponse.newBuilder().setServer(ChannelzProtoUtil.toServer(s)).build();
    } catch (StatusRuntimeException e) {
        responseObserver.onError(e);
        return;
    }
    responseObserver.onNext(resp);
    responseObserver.onCompleted();
}
Also used : GetServerResponse(io.grpc.channelz.v1.GetServerResponse) ServerStats(io.grpc.InternalChannelz.ServerStats) StatusRuntimeException(io.grpc.StatusRuntimeException)

Example 10 with ServerStats

use of io.grpc.InternalChannelz.ServerStats in project grpc-java by grpc.

the class ChannelzProtoUtilTest method toGetServersResponse.

@Test
public void toGetServersResponse() {
    // empty results
    assertEquals(GetServersResponse.getDefaultInstance(), ChannelzProtoUtil.toGetServersResponse(new ServerList(Collections.<InternalInstrumented<ServerStats>>emptyList(), false)));
    // 1 result, paginated
    assertEquals(GetServersResponse.newBuilder().addServer(serverProto).build(), ChannelzProtoUtil.toGetServersResponse(new ServerList(ImmutableList.<InternalInstrumented<ServerStats>>of(server), false)));
    // 1 result, end
    assertEquals(GetServersResponse.newBuilder().addServer(serverProto).setEnd(true).build(), ChannelzProtoUtil.toGetServersResponse(new ServerList(ImmutableList.<InternalInstrumented<ServerStats>>of(server), true)));
    TestServer server2 = new TestServer();
    // 2 results, end
    assertEquals(GetServersResponse.newBuilder().addServer(serverProto).addServer(ChannelzProtoUtil.toServer(server2)).setEnd(true).build(), ChannelzProtoUtil.toGetServersResponse(new ServerList(ImmutableList.<InternalInstrumented<ServerStats>>of(server, server2), true)));
}
Also used : ServerStats(io.grpc.InternalChannelz.ServerStats) ServerList(io.grpc.InternalChannelz.ServerList) TestServer(io.grpc.protobuf.services.ChannelzTestHelper.TestServer) Test(org.junit.Test)

Aggregations

ServerStats (io.grpc.InternalChannelz.ServerStats)10 Test (org.junit.Test)7 ServerList (io.grpc.InternalChannelz.ServerList)5 SocketStats (io.grpc.InternalChannelz.SocketStats)3 ServerSocketsList (io.grpc.InternalChannelz.ServerSocketsList)2 Metadata (io.grpc.Metadata)1 StatusRuntimeException (io.grpc.StatusRuntimeException)1 GetServerResponse (io.grpc.channelz.v1.GetServerResponse)1 Server (io.grpc.channelz.v1.Server)1 TestServer (io.grpc.protobuf.services.ChannelzTestHelper.TestServer)1