從* os.file復制Golang而無需等待EOF

Question

我正在嘗試使用io.Copy從文件復制，它在實際從其內部緩沖區復制字節之前等待EOF，對嗎？ 在我的用例（PTY / SSH會話）中，EOF僅在會話完成時才出現，這意味着我一直在瞎瞎，直到會話確定會話結束為止。

我嘗試一次使用1個字節的CopyN可以正常工作，但是如果我嘗試等待出現一些特定的文本，然后復制過去推送到文件中的內容，則代碼將掛起，我失去會議。 是否有一個功能可以“讀取那里的內容”然后停止，或者有其他標記（例如EOF）可以告訴復制現在停止？

我也嘗試讀取ptyI.pty指向的文件的內容，但它始終返回0字節，因此我無法在那里檢查更新

這是目前處理它的代碼：

type PtyInterface struct {
    pty          *os.File
    buf          *bytes.Buffer
}

func (ptyI *PtyInterface) PrivCmd(cmdStr string) (string, error) {

    // Copy the command provided into the STDIN of the bash shell we opened with
    // the earlier PtyInterface
    _, _ = io.Copy(ptyI.pty, strings.NewReader(string("somecommand")))

    // Assuming everything has gone well, we wait for the prompt to appear

    // We do this by running through individual bytes until the prompt is
    // fully printed (otherwise we might try to send in the answer at the wrong time)
    for !strings.HasSuffix(ptyI.buf.String(), "Prompt question? ") {
        _, _ = io.CopyN(ptyI.buf, ptyI.pty, 1)
    }

    // Once we hit the prompt we throw the answer into STDIN along with a newline
    // and the bash shell should accept this and begin executing the command.
    _, _ = io.Copy(ptyI.pty, strings.NewReader(string("answer\n")))

    // If we dont throw an exit in there then the PTY will never receive an EOF marker and we'll
    // hang on the next copy
    _, _ = io.Copy(ptyI.pty, strings.NewReader(string("exit\n")))

    // Now this copy will wait for an EOF
    _, _ = io.Copy(ptyI.buf, ptyI.pty)

    //Debug info to be printed after
    fmt.Println("\nBytes written to buffer (newone): \n" + ptyI.buf.String())

    return ptyI.buf.String(), nil
}

Answer 1

可以將io.Copy視為用於批量復制或流的io.Copy功能，而不是用於請求/響應模式的正確工具。

只需通過檢查字節是否與每個字節上的消息匹配來將字節累積到消息中即可。 直接使用Read方法。

func Expect(message string, r io.Reader) (resp string, err error) {
    b := []byte{0} // 1 byte buffer
    var n int

    for err == nil {
        n, err = r.Read(b)
        if n == 0 {
            continue
        }
        resp += string(b[0])
        if strings.HasSuffix(resp, message) {
            return resp, err
        }
    }

    return resp, err
}

在您的示例中，您可以這樣使用：

resp, err := Expect("Prompt question? ", ptyI.pty)

下面是它在行動示范與模擬連接io.Reader： 操場 。