Example #1
0
func buildMockDiscoveryServer(port uint) (*mockDiscoveryServer, *grpc.Server) {
	lis, err := net.Listen("tcp", fmt.Sprintf(":%d", port))
	if err != nil {
		panic(err)
	}
	d := &mockDiscoveryServer{}
	s := grpc.NewServer()
	pb.RegisterDiscoveryServer(s, d)
	go s.Serve(lis)
	return d, s
}
Example #2
0
// RegisterRPC registers the local discovery with a gRPC server
func (d *discovery) RegisterRPC(s *grpc.Server) {
	server := &discoveryServer{d}
	pb.RegisterDiscoveryServer(s, server)
}