Merge pull request astaxie#189 from syhou/master

添加服务端读取TCP请求内容和处理的部分

Author: astaxie

ebook/08.1.md

@@ -226,6 +226,76 @@ Go语言中通过net包中的`DialTCP`函数来建立一个TCP连接，并返回
 
 通过把业务处理分离到函数`handleClient`，我们就可以进一步地实现多并发执行了。看上去是不是很帅，增加`go`关键词就实现了服务端的多并发，从这个小例子也可以看出goroutine的强大之处。
 
+有的朋友可能要问：这个服务端没有处理客户端实际请求的内容。如果我们需要通过从客户端发送不同的请求来获取不同的时间格式，而且需要一个长连接，该怎么做呢？请看：
+
+	package main
+
+	import (
+		"fmt"
+		"net"
+		"os"
+		"time"
+		"strconv"
+	)
+
+	func main() {
+		service := ":1200"
+		tcpAddr, err := net.ResolveTCPAddr("tcp4", service)
+		checkError(err)
+		listener, err := net.ListenTCP("tcp", tcpAddr)
+		checkError(err)
+		for {
+			conn, err := listener.Accept()
+			if err != nil {
+				continue
+			}
+			go handleClient(conn)
+		}
+	}
+
+	func handleClient(conn net.Conn) {
+		conn.SetReadDeadline(time.Now().Add(2 * time.Minute)) // set 2 minutes timeout
+		request := make([]byte, 128) // set maxium request length to 128KB to prevent flood attack
+		
+		for {
+			read_len, err := conn.Read(request)
+
+			if err != nil {
+				if err != io.EOF { // ignore EOF since client might send nothing for the moment
+					fmt.Println(err)
+					break
+				}
+
+				neterr, ok := err.(net.Error)
+				if ok && neterr.Timeout() {
+					fmt.Println(neterr)
+					break
+				}
+			}
+
+    		if read_len == 0 {
+      			continue
+    		} else if string(request) == "timestamp" {
+    			daytime := strconv.FormatInt(time.Now().Unix(), 10)
+    			conn.Write([]byte(daytime))
+    		} else {
+    			daytime := time.Now().String()
+    			conn.Write([]byte(daytime)) 
+    		}
+
+    		request = make([]byte, 128) // clear last read content
+		}
+	}
+
+	func checkError(err error) {
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
+			os.Exit(1)
+		}
+	}
+
+在上面这个例子中，我们使用`conn.Read()`不断读取客户端发来的请求。由于我们需要保持与客户端的长连接，所以不能在读取完一次请求后就关闭连接。由于`conn.SetReadDeadline()`设置了超时，当一定时间内客户端无请求发送，`conn`便会自动关闭，下面的for循环即会因为连接已关闭而跳出。需要注意的是，`request`在创建时需要指定一个最大长度以防止flood attack；每次读取到请求处理完毕后，需要清理request，因为`conn.Read()`会将新读取到的内容append到原内容之后。
+
 ### 控制TCP连接
 TCP有很多连接控制函数，我们平常用到比较多的有如下几个函数：