io.Reader & io.Writer 인터페이스

Go code

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"io"
	"strings"
)

// 커스텀 io.Reader 구현
type UpperReader struct {
	r io.Reader
}

func (u *UpperReader) Read(p []byte) (int, error) {
	n, err := u.r.Read(p)
	for i := 0; i < n; i++ {
		if p[i] >= 'a' && p[i] <= 'z' {
			p[i] -= 32
		}
	}
	return n, err
}

// 커스텀 io.Writer 구현
type CountWriter struct {
	w    io.Writer
	n    int64
}

func (c *CountWriter) Write(p []byte) (int, error) {
	n, err := c.w.Write(p)
	c.n += int64(n)
	return n, err
}

// io.TeeReader — 읽으면서 동시에 쓰기
func teeExample() {
	r := strings.NewReader("Hello, Go!")
	var buf bytes.Buffer
	tee := io.TeeReader(r, &buf)

	data, _ := io.ReadAll(tee)
	fmt.Println("읽음:", string(data))
	fmt.Println("복사:", buf.String())
}

// io.MultiWriter — 여러 Writer에 동시 쓰기
func multiWriterExample() {
	var b1, b2 bytes.Buffer
	mw := io.MultiWriter(&b1, &b2)
	fmt.Fprintln(mw, "동시에 두 곳에 쓰기")
	fmt.Println(b1.String() == b2.String()) // true
}

func main() {
	// UpperReader 데코레이터
	r := strings.NewReader("hello world")
	upper := &UpperReader{r: r}
	buf := make([]byte, 64)
	n, _ := upper.Read(buf)
	fmt.Println(string(buf[:n])) // HELLO WORLD

	// CountWriter
	var buf2 bytes.Buffer
	cw := &CountWriter{w: &buf2}
	fmt.Fprintln(cw, "bytes counted")
	fmt.Println("쓴 바이트:", cw.n)

	teeExample()
	multiWriterExample()

	// bufio로 성능 향상
	r2 := strings.NewReader("line1
line2
line3")
	scanner := bufio.NewScanner(r2)
	for scanner.Scan() { fmt.Println(scanner.Text()) }
}