4

我试图io.Reader在 Go 中实现一个非常简单的:

package main

import (
    "io"
    "os"
    "strings"
)

type rot13Reader struct {
    r io.Reader
}

// Very trivial function I implemented.
func (r *rot13Reader) Read(p []byte) (int, error) {
    return 5, nil // Return some trivial values for now.
}

func main() {
    s := strings.NewReader(
        "Lbh penpxrq gur pbqr!")
    r := rot13Reader{s}
    io.Copy(os.Stdout, &r)
}

最后,我想rot13ReaderROT13应用于它正在读取的字符串,但现在,我只是想创建一个io.Reader与正确接口匹配的非常简单的方法。

当我运行这个程序时,它永远不会停止。为什么?我在哪里有无限循环?


更新:我尝试data通过下面的 for 循环改变拼接,但它似乎并没有真正改变拼接。我需要以某种方式复制data吗?

package main

import (

    "io"
    "os"
    "strings"
)

type rot13Reader struct {
    r io.Reader
}

func (r *rot13Reader) Read(data []byte) (int, error) {
    bytesRead, err := r.r.Read(data)

    // Try to alter data... only without this for loop, text prints in standard output... odd.
    for i := 0; i < bytesRead; i++ {
        data[i] += 13   
    }

    return bytesRead, err
}

func main() {
    s := strings.NewReader(
        "Lbh penpxrq gur pbqr!")
    r := rot13Reader{s}
    io.Copy(os.Stdout, &r)
}
4

1 回答 1

8

io.Copy 连续调用您的 Read() 方法。

从文档:

“将副本从 src 复制到 dst,直到在 src 上达到 EOF 或发生错误。它返回复制的字节数和复制时遇到的第一个错误,如果有的话。”

您需要发送 EOF 或返回错误。

这是一个不会无限循环的更新版本。

http://play.golang.org/p/UCfcw4S8yf

package main

import (
    "io"
    "os"
    "strings"
    "fmt"
)

type rot13Reader struct {
    r io.Reader
}

// Very trivial function I implemented.
func (r *rot13Reader) Read(p []byte) (int, error) {
    return 5, io.EOF // Return some trivial values for now.
}

func main() {
    s := strings.NewReader(
        "Lbh penpxrq gur pbqr!")
    r := rot13Reader{s}
    io.Copy(os.Stdout, &r)
    fmt.Printf("Done copying...\n")
}

如果您想实际将字符串复制到输出。您需要从 rot13Reader.r 中读取并将其复制到 Read() 方法中的 p 中。

http://play.golang.org/p/dSCauz0uTw

package main

import (
    "io"
    "os"
    "strings"
    "fmt"
)

type rot13Reader struct {
    r io.Reader
}

// Very trivial function I implemented.
func (r *rot13Reader) Read(p []byte) (int, error) {
    i, err := r.r.Read(p)
    return i, err
}

func main() {
    s := strings.NewReader("Lbh penpxrq gur pbqr!")
    r := rot13Reader{s}
    io.Copy(os.Stdout, &r)
    fmt.Printf("\nDone copying...\n")
}
于 2012-12-24T19:40:12.493 回答