12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- // Solution to part 3 of the Whispering Gophers code lab.
- //
- // This program listens on the host and port specified by the -listen flag.
- // For each incoming connection, it launches a goroutine that reads and decodes
- // JSON-encoded messages from the connection and prints them to standard
- // output.
- //
- // You can test this program by running it in one terminal:
- // $ part3 -listen=localhost:8000
- // And running part2 in another terminal:
- // $ part2 -dial=localhost:8000
- // Lines typed in the second terminal should appear as JSON objects in the
- // first terminal.
- //
- package main
- import (
- "encoding/json"
- "flag"
- "fmt"
- "io"
- "log"
- "net"
- )
- var listenAddr = flag.String("listen", "localhost:8000", "host:port to listen on")
- type Message struct {
- Body string
- }
- func main() {
- flag.Parse()
- // TODO: Create a net.Listener listening from the address in the "listen" flag.
- l, err := net.Listen("tcp", *listenAddr)
- if err != nil {
- log.Fatal(err)
- }
- for {
- // TODO: Accept a new connection from the listener.
- c, err := l.Accept()
- if err != nil {
- log.Fatal(err)
- }
- go serve(c)
- }
- }
- func serve(c net.Conn) {
- // TODO: Use defer to Close the connection when this function returns.
- defer c.Close()
- // TODO: Create a new json.Decoder reading from the connection.
- e := json.NewDecoder(c)
- for {
- // TODO: Create an empty message.
- m := Message{}
- // TODO: Decode a new message into the variable you just created.
- err := e.Decode(&m)
- switch err {
- case nil:
- // TODO: Print the message to the standard output.
- fmt.Printf("%+v\n", m)
- case io.EOF:
- log.Println("End of input")
- return
- default:
- log.Fatal(err)
- }
- }
- }
|