golang gRPC 和 HTTP 共用端口

最新推荐文章于 2024-10-03 13:03:16 发布

jnpher

最新推荐文章于 2024-10-03 13:03:16 发布

阅读量2.7k

点赞数 1

分类专栏： go 文章标签： golang grpc http 端口 gin

本文链接：https://blog.csdn.net/xie1xiao1jun/article/details/114031570

版权

go 专栏收录该内容

27 篇文章 4 订阅

订阅专栏

前言

接口需要提供给其他业务组访问，但是 RPC 协议不同无法内调，对方问能否走 HTTP 接口
有时候，我们服务都需要同时支持grpc跟http(restful)协议,但是我们又不想多端口开放

原理

原理主要是通过共用listener 模式通过Accept()入口来区分协议内容

golang 实现

package main

import (
	"github.com/soheilhy/cmux"
)

func Run(){
	// 起服务
	// Create the main listener.
	l, err := net.Listen("tcp", ":23456")
	if err != nil {
		log.Fatal(err)
	}

	// Create a cmux.
	m := cmux.New(l)

	// First grpc, then HTTP, and otherwise Go RPC/TCP.
	grpcL := m.MatchWithWriters(cmux.HTTP2MatchHeaderFieldSendSettings("content-type", "application/grpc"))
	go grpcS.Serve(grpcL)// Use the muxed listeners for your servers.

	httpL := m.Match(cmux.HTTP1Fast())
	httpS := &http.Server{
		Handler: &helloHTTP1Handler{},
	}
	go httpS.Serve(httpL)

	// Start serving!
	m.Serve()	
}

gin+grpc 实现

package main

import (
	"github.com/soheilhy/cmux"
)

func Run(){
	// 起服务
	l, err := net.Listen("tcp", ":23456")

	m := cmux.New(l)

	// grpc
	grpcL := m.MatchWithWriters(cmux.HTTP2MatchHeaderFieldSendSettings("content-type", "application/grpc"))
	grpcS := grpc.NewServer()
	grpchello.RegisterGreeterServer(grpcS, &server{})
	go grpcS.Serve(grpcL)
	
	// http
	router := gin.Default()
	go func (){
		httpL := m.Match(cmux.HTTP1Fast())
		router.RunListener(httpL)
	}()


	// Start serving!
	m.Serve()	
}