go / intermediate
Snippet
Abfangen von Datenströmen mit einem benutzerdefinierten io.Reader-Dekorator
Die Standardbibliothek von Go basiert stark auf Schnittstellen wie io.Reader. Indem Sie einen vorhandenen io.Reader in einem benutzerdefinierten Struct verpacken und die Methode Read([]byte) (int, error) implementieren, können Sie Datenströme direkt transformieren, ohne den gesamten Datensatz in den Speicher zu laden.
snippet.go
go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package mainimport ("fmt""io""strings")type UpperReader struct {reader io.Reader}func NewUpperReader(r io.Reader) *UpperReader {return &UpperReader{reader: r}}func (ur *UpperReader) Read(p []byte) (int, error) {n, err := ur.reader.Read(p)for i := 0; i < n; i++ {if p[i] >= 'a' && p[i] <= 'z' {p[i] = p[i] - 32}}return n, err}func main() {input := strings.NewReader("hello, world!")upper := NewUpperReader(input)buf := make([]byte, 13)_, _ = io.ReadFull(upper, buf)fmt.Println(string(buf))}
Erklärung
1
type UpperReader struct {
Definiert ein Struct, das ein inneres io.Reader-Interface kapselt.
2
func (ur *UpperReader) Read(p []byte) (int, error) {
Implementiert das io.Reader-Interface und fängt Lesezugriffe ab, um die gelesenen Pufferdaten zu ändern.
3
n, err := ur.reader.Read(p)
Liest Daten aus dem zugrunde liegenden Reader in den Slice p und ermittelt die Anzahl der gelesenen Bytes.
4
p[i] = p[i] - 32
Mutiert die gelesenen Bytes an Ort und Stelle, um Kleinbuchstaben in Großbuchstaben umzuwandeln.