gaspersic/learngo
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

refactor: io png detector ifaces

Browse Source
This commit is contained in:
Inanc Gumus
2019-11-06 12:28:26 +03:00
parent 09d6c3a110
commit d889fbae51
1 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
interfaces/15-png-detector/main.go
View File

@ -23,12 +23,6 @@ import (
// go run . < rosie.unknown // go run . < rosie.unknown
func main() { func main() {
const pngSign = "\x89PNG\r\n\x1a\n"
const pngSignLen = 8
// create an in-memory buffer (bytes.Buffer is an io.Reader and an io.Writer).
memory := bytes.Buffer{}
// get it from the web server // get it from the web server
resp, err := http.Get("https://inancgumus.github.com/x/rosie.unknown") resp, err := http.Get("https://inancgumus.github.com/x/rosie.unknown")
if err != nil { if err != nil {
@ -36,10 +30,22 @@ func main() {
} }
defer resp.Body.Close() defer resp.Body.Close()
if err := transfer(resp.Body); err != nil {
log.Fatal(err)
}
}
func transfer(r io.Reader) error {
const pngSign = "\x89PNG\r\n\x1a\n"
const pngSignLen = 8
// create an in-memory buffer (bytes.Buffer is an io.Reader and an io.Writer).
memory := bytes.Buffer{}
// stream from the web server to the in-memory buffer in 32KB data chunks. // stream from the web server to the in-memory buffer in 32KB data chunks.
// resp.Body.Read(...) -> memory.Write(...) // resp.Body.Read(...) -> memory.Write(...)
if _, err := io.Copy(&memory, resp.Body); err != nil { if _, err := io.Copy(&memory, r); err != nil {
log.Fatal(err) return err
} }
// get the accumulated bytes from the in-memory buffer. // get the accumulated bytes from the in-memory buffer.
@ -50,4 +56,6 @@ func main() {
// compare it with the png signature. // compare it with the png signature.
fmt.Printf("% x\n", pngSign) fmt.Printf("% x\n", pngSign)
return nil
} }