gRPC + HTTP handler

grpc_http_mux.go

package mux

import (
	"net/http"
	"strings"

	"golang.org/x/net/http2"
	"golang.org/x/net/http2/h2c"
)

// NewHandler create new HTTP handler to serve both gRPC and regular HTTP requests
func NewHandler(httpHandler http.Handler, grpcHandler http.Handler) http.Handler {
	muxHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.ProtoMajor == 2 && strings.HasPrefix(r.Header.Get("content-type"), "application/grpc") {
			grpcHandler.ServeHTTP(w, r)
			return
		}

		httpHandler.ServeHTTP(w, r)
	})

	return h2c.NewHandler(muxHandler, &http2.Server{})
}