func getStatusClient() (serverpb.StatusClient, *stop.Stopper, error) { conn, stopper, err := getGRPCConn() if err != nil { return nil, nil, err } return serverpb.NewStatusClient(conn), stopper, nil }
func TestSpanStatsGRPCResponse(t *testing.T) { defer leaktest.AfterTest(t)() ts := startServer(t) defer ts.Stopper().Stop() rpcStopper := stop.NewStopper() defer rpcStopper.Stop() rpcContext := rpc.NewContext(ts.RPCContext().Context, ts.Clock(), rpcStopper) request := serverpb.SpanStatsRequest{ NodeID: "1", StartKey: []byte(roachpb.RKeyMin), EndKey: []byte(roachpb.RKeyMax), } url := ts.ServingAddr() conn, err := rpcContext.GRPCDial(url) if err != nil { t.Fatal(err) } client := serverpb.NewStatusClient(conn) response, err := client.SpanStats(context.Background(), &request) if err != nil { t.Fatal(err) } if a, e := int(response.RangeCount), ExpectedInitialRangeCount(); a != e { t.Errorf("expected %d ranges, found %d", e, a) } }
func (s *statusServer) dialNode(nodeID roachpb.NodeID) (serverpb.StatusClient, error) { addr, err := s.gossip.GetNodeIDAddress(nodeID) if err != nil { return nil, err } conn, err := s.rpcCtx.GRPCDial(addr.String()) if err != nil { return nil, err } return serverpb.NewStatusClient(conn), nil }