mirror of
https://github.com/gopl-zh/gopl-zh.github.com.git
synced 2024-11-14 10:23:42 +00:00
40 lines
742 B
Go
40 lines
742 B
Go
// Copyright © 2016 Alan A. A. Donovan & Brian W. Kernighan.
|
|
// License: https://creativecommons.org/licenses/by-nc-sa/4.0/
|
|
|
|
// See page 227.
|
|
|
|
// Netcat is a simple read/write client for TCP servers.
|
|
package main
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
)
|
|
|
|
//!+
|
|
func main() {
|
|
conn, err := net.Dial("tcp", "localhost:8000")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
done := make(chan struct{})
|
|
go func() {
|
|
io.Copy(os.Stdout, conn) // NOTE: ignoring errors
|
|
log.Println("done")
|
|
done <- struct{}{} // signal the main goroutine
|
|
}()
|
|
mustCopy(conn, os.Stdin)
|
|
conn.Close()
|
|
<-done // wait for background goroutine to finish
|
|
}
|
|
|
|
//!-
|
|
|
|
func mustCopy(dst io.Writer, src io.Reader) {
|
|
if _, err := io.Copy(dst, src); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|