golang smux库源码分析

1、简介

一个socket连接复用的包https://github.com/xtaci/smux

如图所示，多个channel输入通过smux合并在一个连接中，后端服务将连接中的channel分离出来进行处理。简介就不多写了。

1.1 先放客户端和服务端代码， 了解使用方式

package main

import (
	"bytes"
	"encoding/binary"
	"fmt"
	"github.com/rs/zerolog/log"
	"math/rand"
	"zero/smux"

	"net"
	"time"
)

func init() {
   
	rand.Seed(time.Now().UnixNano())
}

/**
一个生成随机数的tcp服务
客户端发送'R', 'A', 'N', 'D'，服务返回一个随机数
*/
func main() {
   
	listener, err := net.Listen("tcp", ":9000")
	if err != nil {
   
		panic(err)
	}
	log.Info().Msg("随机数服务启动，监听9000端口")
	defer listener.Close()
	for {
   
		conn, err := listener.Accept()
		if err != nil {
   
			fmt.Println(err.Error())
			continue
		}
		go SessionHandler(conn)
	}
}

/**
处理会话
每个tcp连接生成一个会话session
*/
func SessionHandler(conn net.Conn) {
   
	session, err := smux.Server(conn, nil)
	if err != nil {
   
		panic(err)
	}
	log.Info().Msgf("收到客户端连接，创建新会话，对端地址：%s", session.RemoteAddr().String())

	for !session.IsClosed() {
   
		stream, err := session.AcceptStream()
		if err != nil {
   
			fmt.Println(err.Error())
			break
		}
		go StreamHandler(stream)
	}
	log.Info().Msgf("客户端连接断开，销毁会话，对端地址：%s", session.RemoteAddr().String())
}

/**
流数据处理
*/
func StreamHandler(stream *smux.Stream) {
   
	buffer := make([]byte, 1024)
	n, err := stream.Read(buffer)
	if err != nil {
   
		log.Error().Msgf("流id：%d，异常信息：%s", stream.ID(), err.Error())
		stream.Close()
		return
	}
	cmd := buffer[:n]
	if bytes.Equal(cmd, []byte{
   'R', 'A', 'N', 'D'}) {
   
		rand := rand.Uint64()
		response := make([]byte, 8)
		binary.BigEndian.PutUint64(response, rand)
		stream.Write(response)
		log.Debug().Msgf("收到客户端数据，流id：%d，随机数：%d， 响应数据：%v", stream.ID(), rand, response)
	} else {
   
		log.Warn().Msgf("收到未知请求命令，流id：%d，请求命令：%v", stream.ID(), cmd)
	}
}

package main

import (
	"encoding/binary"
	"fmt"
	"github.com/rs/zerolog/log"
	"net"
	"net/http"
	"zero/smux"
)

/**
随机数服务客户端连接
*/
var randClient *smux.Session

func init() {
   
	//连接后端随机数服务
	conn, err := net.Dial("tcp", ":9000")
	if err != nil {
   
		log.Warn().