go tcp server client及粘包拆包處理

先上代碼
server.go

package main
import (
    "wingbow.com.cn/server/controller"
)

func main() {
    controller.ServerRun()
}

controller.go

package controller

import (
    "fmt"
    "io"
    "net"

    "wingbow.com.cn/server/network"
)

//ServerRun 服務(wù)
func ServerRun() {
    lister, err := net.Listen("tcp", "localhost:8888")
    fmt.Println("服務(wù)啟動成功:localhost:8888")
    CheckErr(err)
    defer lister.Close()
    for {
        conn, err := lister.Accept()
        CheckErr(err)
        fmt.Println("用戶接入")
        client := network.NewTCPSocket(conn)
        go func() {
            defer client.Close()
            for {
                data, err := client.Read()
                if err == io.EOF {
                    fmt.Println("斷開鏈接")
                    return
                }
                if err != nil {
                    continue
                }
                switchController(data, client)
            }
        }()
    }
}

//CheckErr 異常檢查
func CheckErr(err error) {
    if err != nil {
        panic(err)
    }
}

func switchController(data []byte, c *network.TCPSocket) {
    fmt.Println("讀到的數(shù)據(jù): " + string(data))
    switch string(data) {
    case "ping":
        c.Write([]byte("pong"))
        fmt.Println("發(fā)出的數(shù)據(jù): pong")
        break
    }
}

network.go

package network

import (
    "bufio"
    "bytes"
    "encoding/binary"
    "net"
)

//TCPSocket 連接
type TCPSocket struct {
    tag  string
    conn net.Conn
    r    *bufio.Reader
}

//NewTCPSocket 創(chuàng)建一個TCP客戶端
func NewTCPSocket(conn net.Conn) *TCPSocket {
    return &TCPSocket{conn: conn, r: bufio.NewReader(conn)}
}

//LocalAddr 本地地址
func (c *TCPSocket) LocalAddr() net.Addr {
    return c.conn.LocalAddr()
}

//RemoteAddr 遠(yuǎn)程地址
func (c *TCPSocket) RemoteAddr() net.Addr {
    return c.conn.RemoteAddr()
}

//Close 關(guān)閉
func (c *TCPSocket) Close() error {
    return c.conn.Close()
}

//Write 寫消息
func (c *TCPSocket) Write(message []byte) (int, error) {
    // 讀取消息的長度
    var length = int32(len(message))
    var pkg = new(bytes.Buffer)
    //寫入消息頭
    err := binary.Write(pkg, binary.BigEndian, length)
    if err != nil {
        return 0, err
    }
    //寫入消息體
    err = binary.Write(pkg, binary.BigEndian, message)
    if err != nil {
        return 0, err
    }
    nn, err := c.conn.Write(pkg.Bytes())
    if err != nil {
        return 0, err
    }
    return nn, nil
}

//Read 讀消息
func (c *TCPSocket) Read() ([]byte, error) {
    // Peek 返回緩存的一個切片,該切片引用緩存中前 n 個字節(jié)的數(shù)據(jù),
    // 該操作不會將數(shù)據(jù)讀出,只是引用,引用的數(shù)據(jù)在下一次讀取操作之
    // 前是有效的。如果切片長度小于 n,則返回一個錯誤信息說明原因。
    // 如果 n 大于緩存的總大小,則返回 ErrBufferFull。
    lengthByte, err := c.r.Peek(4)
    if err != nil {
        return nil, err
    }
    //創(chuàng)建 Buffer緩沖器
    lengthBuff := bytes.NewBuffer(lengthByte)
    var length int32
    // 通過Read接口可以將buf中得內(nèi)容填充到data參數(shù)表示的數(shù)據(jù)結(jié)構(gòu)中
    err = binary.Read(lengthBuff, binary.BigEndian, &length)
    if err != nil {
        return nil, err
    }
    // Buffered 返回緩存中未讀取的數(shù)據(jù)的長度
    if int32(c.r.Buffered()) < length+4 {
        return nil, err
    }
    // 讀取消息真正的內(nèi)容
    pack := make([]byte, int(4+length))
    // Read 從 b 中讀出數(shù)據(jù)到 p 中,返回讀出的字節(jié)數(shù)和遇到的錯誤。
    // 如果緩存不為空,則只能讀出緩存中的數(shù)據(jù),不會從底層 io.Reader
    // 中提取數(shù)據(jù),如果緩存為空,則:
    // 1、len(p) >= 緩存大小,則跳過緩存,直接從底層 io.Reader 中讀
    // 出到 p 中。
    // 2、len(p) < 緩存大小,則先將數(shù)據(jù)從底層 io.Reader 中讀取到緩存
    // 中,再從緩存讀取到 p 中。
    _, err = c.r.Read(pack)
    if err != nil {
        return nil, err
    }
    return pack[4:], nil
}

客戶端
client.go

package main

import (
    "bytes"
    "encoding/binary"
    "fmt"
    "log"
    "net"
    "time"
)

func main() {
    conn, err := net.Dial("tcp", "localhost:8888")
    if err != nil {
        log.Println("dial error:", err)
        return
    }
    defer conn.Close()
    for {
        data, _ := Encode(time.Now().Format("2006-01-02 15:04:05"))
        time.Sleep(time.Second)
        _, err := conn.Write(data)
        fmt.Println(err)
    }
}
func Encode(message string) ([]byte, error) {
    // 讀取消息的長度
    var length = int32(len(message))
    var pkg = new(bytes.Buffer)
    // 寫入消息頭
    err := binary.Write(pkg, binary.BigEndian, length)
    if err != nil {
        return nil, err
    }
    // 寫入消息實(shí)體
    err = binary.Write(pkg, binary.BigEndian, []byte(message))
    if err != nil {
        return nil, err
    }
    return pkg.Bytes(), nil
}
  上面處理粘包的方法是用了自定義的非定長的數(shù)據(jù)包,這也是一種常用的粘包處理方式。
   另外還有自定義分隔符、定長分割等方法。
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容