moby--moby/api/server/router/grpc/grpc.go

package grpc // import "github.com/docker/docker/api/server/router/grpc"

import (
	"github.com/docker/docker/api/server/router"
	"github.com/moby/buildkit/util/grpcerrors"
	"golang.org/x/net/http2"
	"google.golang.org/grpc"
)

type grpcRouter struct {
	routes     []router.Route
	grpcServer *grpc.Server
	h2Server   *http2.Server
}

// NewRouter initializes a new grpc http router
func NewRouter(backends ...Backend) router.Router {
	opts := []grpc.ServerOption{grpc.UnaryInterceptor(grpcerrors.UnaryServerInterceptor), grpc.StreamInterceptor(grpcerrors.StreamServerInterceptor)}
	server := grpc.NewServer(opts...)

	r := &grpcRouter{
		h2Server:   &http2.Server{},
		grpcServer: server,
	}
	for _, b := range backends {
		b.RegisterGRPC(r.grpcServer)
	}
	r.initRoutes()
	return r
}

// Routes returns the available routers to the session controller
func (r *grpcRouter) Routes() []router.Route {
	return r.routes
}

func (r *grpcRouter) initRoutes() {
	r.routes = []router.Route{
		router.NewPostRoute("/grpc", r.serveGRPC),
	}
}