Go gRPC 系列七:讓服務同時提供 HTTP 接口

前言

你們好,我是煎魚,不知道你有沒有想過以下問題:git

  • 接口須要提供給其餘業務組訪問,可是 RPC 協議不一樣沒法內調,對方問可否走 HTTP 接口,怎麼辦?github

  • 微信(公衆號、小程序)等第三方回調接口只支持 HTTP 接口,怎麼辦golang

我相信你在實際工做中都會遇到如上問題,在 gRPC 中都是能夠找到其它解決方案的,本章節將會進行介紹。小程序

爲何能夠同時提供 HTTP 接口

關鍵一點,gRPC 的協議是基於 HTTP/2 的,所以應用程序可以在單個 TCP 端口上提供 HTTP/1.1 和 gRPC 接口服務(兩種不一樣的流量)bash

怎麼同時提供 HTTP 接口

檢測協議

if r.ProtoMajor == 2 && strings.Contains(r.Header.Get("Content-Type"), "application/grpc") {
    server.ServeHTTP(w, r)
} else {
    mux.ServeHTTP(w, r)
}
複製代碼

流程

  1. 檢測請求協議是否爲 HTTP/2
  2. 判斷 Content-Type 是否爲 application/grpc(gRPC 的默認標識位)
  3. 根據協議的不一樣轉發到不一樣的服務處理

gRPC

TLS

在前面的章節,爲了便於展現所以沒有簡單封裝微信

在本節需複用代碼,從新封裝了,可詳見:go-grpc-exampleapp

目錄結構

新建 simple_http_client、simple_http_server 目錄,目錄結構以下:ui

go-grpc-example
├── client
│   ├── simple_client
│   ├── simple_http_client
│   └── stream_client
├── conf
├── pkg
│   └── gtls
├── proto
├── server
│   ├── simple_http_server
│   ├── simple_server
│   └── stream_server
複製代碼

Server

在 simple_http_server 目錄下新建 server.go,寫入文件內容:google

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
}
複製代碼
  • http.NewServeMux:建立一個新的 ServeMux,ServeMux 本質上是一個路由表。它默認實現了 ServeHTTP,所以返回 Handler 後可直接經過 HandleFunc 註冊 pattern 和處理邏輯的方法
  • http.ListenAndServeTLS:可簡單的理解爲提供監聽 HTTPS 服務的方法,重點的協議判斷轉發,也在這裏面

其實,你理解後就會以爲很簡單,核心步驟:判斷 -> 轉發 -> 響應。咱們改變了前兩步的默認邏輯,僅此而已spa

Client

在 simple_http_server 目錄下新建 client.go,寫入文件內容:

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())
}
複製代碼

驗證

gRPC Client

$ go run client.go 
2018/10/04 14:56:56 resp: gRPC HTTP Server
複製代碼

HTTP/1.1 訪問

image

總結

經過本章節,表面上完成了同端口提供雙服務的功能,但實際上,應該是加深了 HTTP/2 的理解和使用,這纔是本質

問題

你覺得這個方案就萬能了嗎,不。Envoy Proxy 的支持就不完美,沒法同時監聽一個端口的兩種流量。

若是有任何疑問或錯誤,歡迎在 issues 進行提問或給予修正意見,若是喜歡或對你有所幫助,歡迎 Star,對做者是一種鼓勵和推動。

個人公衆號

image

參考

本系列示例代碼

相關文章
相關標籤/搜索