go - 如何停止 io.CopyN

Question

我有一些从文件复制到 tcp 套接字（如 ftp 服务器）的代码，并且希望能够在需要时中止此复制。

我只是使用 io.CopyN(socket, file, size) 并且看不到发出中止信号的方法。有任何想法吗？

Answer 1

关闭输入文件怎么样？ io.CopyN然后将返回错误并中止。

这是一个演示（如果不在 Linux 上运行，请更改/dev/zero并/dev/null为您的操作系统等效！）

package main

import (
    "fmt"
    "io"
    "log"
    "os"
    "time"
)

func main() {
    in, err := os.Open("/dev/zero")
    if err != nil {
        log.Fatal(err)
    }

    out, err := os.Create("/dev/null")
    if err != nil {
        log.Fatal(err)
    }

    go func() {
        time.Sleep(time.Second)
        in.Close()
    }()

    written, err := io.CopyN(out, in, 1E12)
    fmt.Printf("%d bytes written with error %s\n", written, err)
}

运行时它将打印类似

9756147712 bytes written with error read /dev/zero: bad file descriptor

Answer 2

CopyN努力复制 N 个字节。如果您想选择性地复制少于 N 个字节，那么首先不要使用 CopyN。我可能会将原始代码改编为（未经测试的代码）：

func copyUpToN(dst Writer, src Reader, n int64, signal chan int) (written int64, err error) {
    buf := make([]byte, 32*1024)
    for written < n {
        select {
        default:
        case <-signal:
            return 0, fmt.Errorf("Aborted") // or whatever
        }

        l := len(buf)
        if d := n - written; d < int64(l) {
            l = int(d)
        }
        nr, er := src.Read(buf[0:l])
        if nr > 0 {
            nw, ew := dst.Write(buf[0:nr])
            if nw > 0 {
                written += int64(nw)
            }
            if ew != nil {
                err = ew
                break
            }
            if nr != nw {
                err = io.ErrShortWrite
                break
            }
        }
        if er != nil {
            err = er
            break
        }
    }
    return written, err
}