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

nicoxix

已于 2023-02-02 11:58:35 修改

阅读量812

点赞数

分类专栏： Go gogogo 文章标签： golang go http grpc https

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

本文链接：https://blog.csdn.net/nxb593427560/article/details/118516757

版权

Go gogogo 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

核心部分

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")
	}
}