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

 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 }

go build multi_thread_echo.go

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传