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.Copyc.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" )