Go语言指南练习: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 (
    "io"
    "os"
    "strings"
)

type rot13Reader struct {
    r io.Reader
}

func rot13(a []byte, n int) []byte {
    for i := 0; i < n; i++ {
        switch {
        case (a[i] >= 'A' && a[i] <= 'M') || (a[i] >= 'a' && a[i] <= 'm'):
            a[i] += 13
        case (a[i] >= 'N' && a[i] <= 'Z') || (a[i] >= 'n' && a[i] <= 'z'):
            a[i] -= 13
        }
    }
    return a
}

func (a rot13Reader) Read(p []byte) (n int, err error) {
    n, err = a.r.Read(p)
    p = rot13(p, n)	
    return
}

func main() {
    s := strings.NewReader("Lbh penpxrq gur pbqr!")
    r := rot13Reader{s}
    io.Copy(os.Stdout, &r)
}
posted @ 2024-02-17 12:53  Nephna  阅读(125)  评论(0)    收藏  举报