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);
}
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));
}
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);
}
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();
}
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)));
}
Aggregations