Golang:一个服务同时处理普通http请求和gRPC请求

原创已于 2023-02-02 11:58:35 修改 · 911 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#golang #go #http #grpc #https

于 2021-07-06 11:35:26 首次发布

Go gogogo 专栏收录该内容

7 篇文章

订阅专栏

该博客展示了如何使用Go语言搭建一个同时支持HTTP/2和gRPC服务的服务器。通过golang.org/x/net/http2和golang.org/x/net/http2/h2c包，实现了HTTP/2协议的支持，并利用google.golang.org/grpc库处理gRPC请求。服务器配置了TLS选项，根据启用情况使用ListenAndServeTLS或ListenAndServe进行监听。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

核心部分

package main

import (
	"log"
	"net/http"
	"strings"

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

var (
	EnableTLS bool
	CertFile  string
	KeyFile   string
	Address   string
)

type MyServer struct {
	grpcServer *grpc.Server
	httpServer *http.ServeMux
}

func (ms *MyServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	if r.ProtoMajor == 2 && strings.HasPrefix(r.Header.Get("Content-Type"), "application/grpc") {
		ms.grpcServer.ServeHTTP(w, r)
	} else {
		ms.httpServer.ServeHTTP(w, r)
	}
}

func main() {
	ms := &MyServer{
		grpcServer: grpc.NewServer(),
		httpServer: http.NewServeMux(),
	}

	server := &http.Server{
		Addr: Address,
	}
	var err error
	if EnableTLS {
		server.Handler = ms
		err = server.ListenAndServeTLS(CertFile, KeyFile)
	} else {
		server.Handler = h2c.NewHandler(ms, &http2.Server{})
		err = server.ListenAndServe()
	}
	if err != nil {
		log.Printf("serve failed")
	}
}