golang: 技巧---同一个端口监听 http & grpc & websocket 三种不同协议

同一端口支持http & websocket & grpc

package main



var upgrader = websocket.Upgrader{}

func main() {

	l, err := net.Listen("tcp", ":23456")
	if err != nil {
		log.Fatal(err)
	}

	
	m := cmux.New(l)

	httpl := m.Match(cmux.HTTP1Fast())
	grpcl := m.Match(cmux.Any())

	go serveGRPC(grpcl)
	go serveHTTPAndWs(httpl)

	if err := m.Serve(); !strings.Contains(err.Error(), "use of closed network connection") {
		panic(err)
	}

}

type grpcServer struct{}

func (s *grpcServer) SayHello(ctx context.Context, in *grpchello.HelloRequest) (
	*grpchello.HelloReply, error) {

	fmt.Printf("request:%v\n", in)
	return &grpchello.HelloReply{Message: "Hello " + in.Name + " from cmux"}, nil
}

func serveGRPC(l net.Listener) {
	grpcs := grpc.NewServer()
	grpchello.RegisterGreeterServer(grpcs, &grpcServer{})
	if err := grpcs.Serve(l); err != cmux.ErrListenerClosed {
		panic(err)
	}
}

func serveHTTPAndWs(l net.Listener) {
	r := gin.Default()

	// websocket echo
	r.Any("/websocket", func(c *gin.Context) {
		r := c.Request
		w := c.Writer
		conn, err := upgrader.Upgrade(w, r, nil)
		if err != nil {
			fmt.Printf("err = %s\n", err)
			return
		}

		defer func() {
			// 发送websocket结束包
			conn.WriteMessage(websocket.CloseMessage,
				websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
			// 真正关闭conn
			conn.Close()
		}()
		// 读取一个包
		mt, d, err := conn.ReadMessage()
		if err != nil {
			fmt.Printf("read fail = %v\n", err)
			return
		}

		fmt.Printf("data:%s\n", d)
		// 写入一个包
		err = conn.WriteMessage(mt, d)
		if err != nil {
			fmt.Printf("write fail = %v\n", err)
			return
		}
	})

	// http echo
	r.GET("/http", func(c *gin.Context) {
		io.Copy(c.Writer, c.Request.Body)
	})

	s := &http.Server{
		Handler: r,
	}

	if err := s.Serve(l); err != cmux.ErrListenerClosed {
		panic(err)
	}
}

总结

• 适用了cmux实现效果，需要注意上面的m.Match顺序不要修改

github

https://github.com/guonaihong/gout

补下依赖

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"github.com/soheilhy/cmux"
	"log"
	"net/http"
	"strings"

	"google.golang.org/grpc"

	"context"

	"net"

	grpchello "google.golang.org/grpc/examples/helloworld/helloworld"
	"io"
)