这个是A Tour of Go这个教程里面的第八个练习,原题如下:
有种常见的模式是一个
io.Reader
包装另一个io.Reader
,然后通过某种方式修改其数据流。例如,
gzip.NewReader
函数接受一个io.Reader
(已压缩的数据流)并返回一个同样实现了io.Reader
的*gzip.Reader
(解压后的数据流)。编写一个实现了
io.Reader
并从另一个io.Reader
中读取数据的rot13Reader
,通过应用 rot13 代换密码对数据流进行修改。
rot13Reader
类型已经提供。实现Read
方法以满足io.Reader
。
原题模板代码如下:
package main
import (
"io"
"os"
"strings"
)
type rot13Reader struct {
r io.Reader
}
func main() {
s := strings.NewReader("Lbh penpxrq gur pbqr!")
r := rot13Reader{s}
io.Copy(os.Stdout, &r)
}
ROT13转换比较简单,如图:
实现起来就是要覆盖一下原来的Read方法,在Read方法中对每个字符进行替换:
package main
import (
"io"
"os"
"strings"
)
type rot13Reader struct {
r io.Reader
}
//字母转换
func rot13(out byte) byte{
switch{
case out >= 'A' && out <= 'M' || out >= 'a' && out <= 'm':
out += 13
case out >= 'N' && out <= 'Z' || out >= 'n' && out <= 'z':
out -= 13
}
return out
}
//重写Read方法
func (fz rot13Reader) Read(b []byte) (int,error){
n, e := fz.r.Read(b)
for i := 0; i < n; i++ {
b[i] = rot13(b[i])
}
return n,e
}
func main() {
s := strings.NewReader("Lbh penpxrq gur pbqr!")
r := rot13Reader{s}
io.Copy(os.Stdout, &r)
}
输出结果(转换后的内容)如下:
You cracked the code!