Go指南练习 rot13Reader

题目：
练习：rot13Reader
有种常见的模式是一个 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 (
    // "errors"

    "io"
    "os"
    "strings"
)

type rot13Reader struct {
    r io.Reader
}

func (r13 rot13Reader) Read(b []byte) (int, error) {
    // temp_b := []byte{} 这里新造一个字节切片接受，会导致程序卡住
    r, err := r13.r.Read(b)
    for i, v := range b {
        switch {
        case 65 <= v && v <= 90:
            new_v := v + 13
            if new_v <= 90 {
                b[i] = new_v
            } else {
                new_v = new_v - 26
                b[i] = new_v
            }
        case 97 <= v && v <= 122:
            new_v := v + 13
            if new_v <= 122 {
                b[i] = new_v
            } else {
                new_v = new_v - 26
                b[i] = new_v
            }
                        // 这里抛出异常也会卡住执行很久
            // default:
            //  return 0, errors.New("not alphabet")
        }
    }

    return r, err

}

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

Go指南练习 rot13Reader

猜你喜欢