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

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

package main

import (
    "log"
    "runtime"
    "net"
)

func main() {

    tcpAddr, err := net.ResolveTCPAddr("tcp", ":9988")
    check_error(err)
    tcpListener, err := net.ListenTCP("tcp", tcpAddr)
    check_error(err)
    for i := 0; i < 100; i++ {
        go handle_tcp_accept(tcpListener)
    }

    select {}
}



func handle_tcp_accept(tcpListener *net.TCPListener) {
    for {
        tcpConn, err := tcpListener.AcceptTCP()
        if err != nil {
            log.Println("tcp accept failed!")
            continue
        } else {
            connChan := make(chan []byte)
            go write_tcp_conn(tcpConn, connChan)
            go read_tcp_conn(tcpConn, connChan)
        }
    }

}


func read_tcp_conn(tcpConn *net.TCPConn, connChan chan []byte) {
    buffer := make([]byte, 2048)
    tcpConn.SetReadBuffer(2048)
    for {
        n, err := tcpConn.Read(buffer[0:])
        if err != nil {
            log.Println("one tcp connection read function failed!")
            log.Println("one tcp connection close now!")
            tcpConn.Close()
            runtime.Goexit()
        } else {
            connChan <- buffer[0 : n-1]
        }
    }
}

func write_tcp_conn(tcpConn *net.TCPConn, connChan chan []byte) {
    for {
        msg := <-connChan
        log.Println(string(msg))
        tcpConn.Write([]byte(msg)[0 : len(msg)+1])
    }
}




func check_error(err error) {
    if err != nil {
        log.Printf("Fatal error : %s", err.Error())
    }

}

上面代码保存为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/

转贴请注明来自：格通