golang--类似mochiweb的多进程监听tcp链接

用Erlang开发web，很可能就会使用到mochiweb。mochiweb使用16个Erlang式的进程来监听tcp链接，使得任何链接进来都会很快得到处理。Golang跟Erlang很像，在这里实现了一个类似mochiweb的tcp监控链接，一个简单的echo服务器。先看代码。

1 package main  2
3 import (  4     "log"
5     "runtime"
6     "net"
7 )  8
9 func main() { 10
11     tcpAddr, err := net.ResolveTCPAddr("tcp", ":9988") 12  check_error(err) 13     tcpListener, err := net.ListenTCP("tcp", tcpAddr) 14  check_error(err) 15     for i := 0; i < 100; i++ { 16  go handle_tcp_accept(tcpListener) 17  } 18
19     select {} 20 } 21
22
23
24 func handle_tcp_accept(tcpListener *net.TCPListener) { 25     for { 26         tcpConn, err := tcpListener.AcceptTCP() 27         if err != nil { 28             log.Println("tcp accept failed!") 29             continue
30         } else { 31             connChan := make(chan []byte) 32  go write_tcp_conn(tcpConn, connChan) 33  go read_tcp_conn(tcpConn, connChan) 34  } 35  } 36
37 } 38
39
40 func read_tcp_conn(tcpConn *net.TCPConn, connChan chan []byte) { 41     buffer := make([]byte, 2048) 42     tcpConn.SetReadBuffer(2048) 43     for { 44         n, err := tcpConn.Read(buffer[0:]) 45         if err != nil { 46             log.Println("one tcp connection read function failed!") 47             log.Println("one tcp connection close now!") 48  tcpConn.Close() 49  runtime.Goexit() 50         } else { 51             connChan <- buffer[0 : n-1] 52  } 53  } 54 } 55
56 func write_tcp_conn(tcpConn *net.TCPConn, connChan chan []byte) { 57     for { 58         msg := <-connChan 59         log.Println(string(msg)) 60         tcpConn.Write([]byte(msg)[0 : len(msg)+1]) 61  } 62 } 63
64
65
66
67 func check_error(err error) { 68     if err != nil { 69         log.Printf("Fatal error : %s", err.Error()) 70  } 71
72 }

上面代码保存为multi_thread_echo.go, 使用gotool编译

go build multi_thread_echo.go

上面代码使用了100个goroutine来监听TCP链接，每个链接进来，都会新建2个goroutine，一个read，一个write。下面看看效果图：



参考：

1.Golang Away: TCP Chat Server

http://www.badgerr.co.uk/2011/06/20/golang-away-tcp-chat-server/

2.Network programming with Go

http://jan.newmarch.name/go/

转贴请注明来自：格通