原文地址:带入gRPC:让你的服务同时提供 HTTP 接口
项目地址:https://github.com/EDDYCJY/go...html
我相信你在实际工做中都会遇到如上问题,在 gRPC 中都是有解决方案的,本章节将会进行介绍 🤔git
关键一点,gRPC 的协议是基于 HTTP/2 的,所以应用程序可以在单个 TCP 端口上提供 HTTP/1.1 和 gRPC 接口服务(两种不一样的流量)github
if r.ProtoMajor == 2 && strings.Contains(r.Header.Get("Content-Type"), "application/grpc") { server.ServeHTTP(w, r) } else { mux.ServeHTTP(w, r) }
在前面的章节,为了便于展现所以没有简单封装golang
在本节需复用代码,从新封装了,可详见:go-grpc-example小程序
新建 simple_http_client、simple_http_server 目录,目录结构以下:segmentfault
go-grpc-example ├── client │ ├── simple_client │ ├── simple_http_client │ └── stream_client ├── conf ├── pkg │ └── gtls ├── proto ├── server │ ├── simple_http_server │ ├── simple_server │ └── stream_server
在 simple_http_server 目录下新建 server.go,写入文件内容:微信
package main import ( "context" "log" "net/http" "strings" "github.com/EDDYCJY/go-grpc-example/pkg/gtls" pb "github.com/EDDYCJY/go-grpc-example/proto" "google.golang.org/grpc" ) type SearchService struct{} func (s *SearchService) Search(ctx context.Context, r *pb.SearchRequest) (*pb.SearchResponse, error) { return &pb.SearchResponse{Response: r.GetRequest() + " HTTP Server"}, nil } const PORT = "9003" func main() { certFile := "../../conf/server/server.pem" keyFile := "../../conf/server/server.key" tlsServer := gtls.Server{ CertFile: certFile, KeyFile: keyFile, } c, err := tlsServer.GetTLSCredentials() if err != nil { log.Fatalf("tlsServer.GetTLSCredentials err: %v", err) } mux := GetHTTPServeMux() server := grpc.NewServer(grpc.Creds(c)) pb.RegisterSearchServiceServer(server, &SearchService{}) http.ListenAndServeTLS(":"+PORT, certFile, keyFile, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { if r.ProtoMajor == 2 && strings.Contains(r.Header.Get("Content-Type"), "application/grpc") { server.ServeHTTP(w, r) } else { mux.ServeHTTP(w, r) } return }), ) } func GetHTTPServeMux() *http.ServeMux { mux := http.NewServeMux() mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { w.Write([]byte("eddycjy: go-grpc-example")) }) return mux }
其实,你理解后就会以为很简单,核心步骤:判断 -> 转发 -> 响应。咱们改变了前两步的默认逻辑,仅此而已app
在 simple_http_server 目录下新建 client.go,写入文件内容:google
package main import ( "context" "log" "google.golang.org/grpc" "github.com/EDDYCJY/go-grpc-example/pkg/gtls" pb "github.com/EDDYCJY/go-grpc-example/proto" ) const PORT = "9003" func main() { tlsClient := gtls.Client{ ServerName: "go-grpc-example", CertFile: "../../conf/server/server.pem", } c, err := tlsClient.GetTLSCredentials() if err != nil { log.Fatalf("tlsClient.GetTLSCredentials err: %v", err) } conn, err := grpc.Dial(":"+PORT, grpc.WithTransportCredentials(c)) if err != nil { log.Fatalf("grpc.Dial err: %v", err) } defer conn.Close() client := pb.NewSearchServiceClient(conn) resp, err := client.Search(context.Background(), &pb.SearchRequest{ Request: "gRPC", }) if err != nil { log.Fatalf("client.Search err: %v", err) } log.Printf("resp: %s", resp.GetResponse()) }
$ go run client.go 2018/10/04 14:56:56 resp: gRPC HTTP Server
经过本章节,表面上完成了同端口提供双服务的功能,但实际上,应该是加深了 HTTP/2 的理解和使用,这才是本质spa
若是你有一个需求,是要同时提供 RPC 和 RESTful JSON API 两种接口的,不要犹豫,点进去:gRPC + gRPC Gateway 实践
你觉得这个方案就万能了吗,不。Envoy Proxy 的支持就不完美,没法同时监听一个端口的两种流量 😤