Go Tour练习rot13Reader

本篇是”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 (rr rot13Reader)Read(b []byte)(int,error){
	n,err := rr.r.Read(b)
	if err != nil{
		return 0,err
	}
	for i := 0; i < n;i++{
		b[i] = rr.rot13(b[i])
	}
	return n,nil
}

//rot13 rot13Reader的内部方法，用于单个字符的rot13
func (rr rot13Reader)rot13(b byte) byte{
	if b >= 'a' && b <= 'z'{
		b += 13
		if b > 'z'{
			b -= 26
		}
	}else if b >='A' && b <= 'Z'{
		b += 13
		if b > 'Z'{
			b -= 26
		}
	}
	return b
}

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