Ejemplo n.º 1
0
// serve accepts incoming connections on the listener l,
// creating a new service goroutine for each. The service goroutines
// read requests and then call handler to reply to them.
func serve(l net.Listener, grpcS *grpc.Server, handler http.Handler, readTimeout time.Duration) error {
	// TODO: assert net.Listener type? Arbitrary listener might break HTTPS server which
	// expect a TLS Conn type.
	httpl := l
	if grpcS != nil {
		m := cmux.New(l)
		grpcl := m.Match(cmux.HTTP2HeaderField("content-type", "application/grpc"))
		httpl = m.Match(cmux.Any())
		go func() { plog.Fatal(m.Serve()) }()
		go plog.Fatal(grpcS.Serve(grpcl))
	}

	logger := defaultLog.New(ioutil.Discard, "etcdhttp", 0)
	// TODO: add debug flag; enable logging when debug flag is set
	srv := &http.Server{
		Handler:     handler,
		ReadTimeout: readTimeout,
		ErrorLog:    logger, // do not log user error
	}
	return srv.Serve(httpl)
}
Ejemplo n.º 2
0
func tearDown(s *grpc.Server, cc *grpc.ClientConn) {
	cc.Close()
	s.Stop()
}
Ejemplo n.º 3
0
func RegisterTestServiceServer(s *grpc.Server, srv TestServiceServer) {
	s.RegisterService(&_TestService_serviceDesc, srv)
}
Ejemplo n.º 4
0
func RegisterRouteGuideServer(s *grpc.Server, srv RouteGuideServer) {
	s.RegisterService(&_RouteGuide_serviceDesc, srv)
}
Ejemplo n.º 5
0
func RegisterHealthServer(s *grpc.Server, srv HealthServer) {
	s.RegisterService(&_Health_serviceDesc, srv)
}
Ejemplo n.º 6
0
func RegisterWorkerServer(s *grpc.Server, srv WorkerServer) {
	s.RegisterService(&_Worker_serviceDesc, srv)
}
Ejemplo n.º 7
0
func RegisterGreeterServer(s *grpc.Server, srv GreeterServer) {
	s.RegisterService(&_Greeter_serviceDesc, srv)
}