conn（net.conn）并不总是在套接字上写入

Question

我用一个服务器和客户端做了一个应用程序，用TCP套接字发送信息

问题是，如果函数zfs。ReceiveSnapshot（在服务器端）不返回错误（err==nil），conn.写（[]byte（"0"））不工作，客户端没有收到任何字节继续，它不能关闭连接...

我给你看服务器和客户端的代码

服务器：

package main

import (
    "net"
    "github.com/mistifyio/go-zfs"
)

func main() {
    // Listen for incoming connections
    l, _ := net.Listen("tcp", ":7766")

    // Close the listener when the application closes
    defer l.Close()
    fmt.Println("Listening on port 7766...")
    for {
        // Listen for an incoming connection.
        conn, _ := l.Accept()
        // Handle connections in a new goroutine.
        go handleRequest(conn)
    }
}

// Handles incoming requests
func handleRequest(conn net.Conn) {
    // Receive snapshot
    _, err := zfs.ReceiveSnapshot(conn, "tank/replication")
    if err != nil {
        conn.Write([]byte("1"))
        zfs.ReceiveSnapshot(conn, "tank/replication")
        conn.Close()
    } else {
        conn.Write([]byte("0"))
        conn.Close()
    }
}

客户：

package main

import (
    "net"
    "github.com/mistifyio/go-zfs"
)

func main() {
    conn, _ := net.Dial("tcp", "192.168.99.5:7766")

    for i := 0; i < 1; i++ {
        // Get all snapshots in tank/test
        take, _ := zfs.Snapshots("tank/test")

        // Select snapshots
        snap := take[0].Name
        ds1, _ := zfs.GetDataset(snap)

        // Send first snapshot
        ds1.SendSnapshot(conn)
        defer conn.Close()

        buff := make([]byte, 1024)
        n, _ := conn.Read(buff)

        if n != 0 {
            snap = take[1].Name
            ds2, _ := zfs.GetDataset(snap)
            zfs.SendSnapshotIncremental(conn, ds1, ds2, zfs.IncrementalStream)
            conn.Close()
        }
    }
}

[编辑]：如果接收快照返回错误，控制。写（[]byte）写"1"，客户端接收到它，执行SendSnapshot增量（如果n！=0）并关闭客户端的连接...但是，如果ReceiveSnapshot不返回错误，则返回。写（[]byte）不写0，只是我用ctrl C关闭服务器端的连接

Answer 1

[编辑

解决这个问题的几个修改。。。

服务器：

func handleRequest(conn net.Conn) {
    // Number of snapshots in server side
    take, _ := zfs.Snapshots("tank/replication")
    count := len(take)
    if count < 1 {
        conn.Write([]byte("0"))
    } else {
        conn.Write([]byte("1"))
    }

    // Receive snapshot
    zfs.ReceiveSnapshot(conn, "tank/replication")
}

count返回服务器拥有的快照数量，因此如果没有快照，客户端将执行该案例，在另一种情况下为默认情况。

客户：

func main() {
    // Get the last two snapshots
    take, _ := zfs.Snapshots("tank/test")
    snap := take[0].Name
    ds1, _ := zfs.GetDataset(snap)
    snap = take[1].Name
    ds2, _ := zfs.GetDataset(snap)

    // New server connection
    conn, _ := net.Dial("tcp", "192.168.99.5:7766")

    // Read data of server side
    buff := bufio.NewReader(conn)
    n, _ := buff.ReadByte()

    // Execute the correct case
    switch string(n) {
    case "0":
        ds1.SendSnapshot(conn)
        conn.Close()
    default :
        zfs.SendSnapshotIncremental(conn, ds1, ds2, zfs.IncrementalStream)
        conn.Close()
    }
}

我不知道这个解决方案是否很好，但它是有效的！

Answer 2

我认为问题在于以下几方面：

buff := make([]byte, 1024)
n, _ := conn.Read(buff)

在本例中，n是读取的字节数，而不是其中任何一个字节的值。

我会：

buff := make([]byte, 1)
n,err := conn.Read(buff)
buff = buff[:n]
if len(buff) == 0 {
   //error
}

if(buff[0] != 0){...