Question

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

我只是使用io.CopyN（套接字，文件，大小），并且无法看到发出中止信号的方法。有什么想法吗？

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
}

如何停止io.CopyN

2 个答案: