白羊的空间

Go实现简单的Socket服务端笔记(三)

利用goroutine实现同时多个客户端连接

将数据读取放入单独的方法中,利用goroutine运行

本文代码查看github:
https://github.com/zboyco/go-server/tree/step-3

要实现多个客户端连接,需要将每个客户端放入单独的线程处理,通常我们使用多线程实现这个功能,但Go语言提供的 goroutine, 可以非常方便的实现并行(goroutine并不是多线程)

  1. 将接收数据的相关代码放入单独的方法中:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
func handleClient(conn net.Conn) {
	//获取连接地址
	remoteAddr := conn.RemoteAddr()

	fmt.Println("客户地址:", remoteAddr)

	//定义一个数据接收Buffer
	var buf [1024]byte

	for {
		fmt.Println("等待接收数据...")
		//读取数据
		n, err := conn.Read(buf[0:])

		if err != nil {
			fmt.Println("数据接收错误, ", err)
			return
		}

		//将bytes转为字符串
		result := string(buf[0:n])

		//输出结果
		fmt.Println("接收到数据:", result)
	}
}
  1. 在接收到新客户端连接后,开启一个新的goroutine处理
    使用方法:
    go func()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
for {
	fmt.Println("等待客户连接...")

	//开始接收连接
	conn, err := tcpListener.Accept()

	if err != nil {
		fmt.Println("客户连接失败, ", err)
		continue
	}

	//启用goroutine处理
	go handleClient(conn)
}

完整的代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package main

import (
	"fmt"
	"net"
)

func main() {

	//定义一个本机端口
	localAddress, _ := net.ResolveTCPAddr("tcp4", ":9043")

	//监听端口
	tcpListener, err := net.ListenTCP("tcp", localAddress)

	if err != nil {
		fmt.Println("监听出错, ", err)
		return
	}

	//程序返回后关闭socket
	defer tcpListener.Close()

	for {
		fmt.Println("等待客户连接...")

		//开始接收连接
		conn, err := tcpListener.Accept()

		if err != nil {
			fmt.Println("客户连接失败, ", err)
			continue
		}

		//启用goroutine处理
		go handleClient(conn)
	}
}

func handleClient(conn net.Conn) {
	//获取连接地址
	remoteAddr := conn.RemoteAddr()

	fmt.Println("客户地址:", remoteAddr)

	//定义一个数据接收Buffer
	var buf [1024]byte

	for {
		fmt.Println("等待接收数据...")
		//读取数据
		n, err := conn.Read(buf[0:])

		if err != nil {
			fmt.Println("数据接收错误, ", err)
			return
		}

		//将bytes转为字符串
		result := string(buf[0:n])

		//输出结果
		fmt.Println("接收到数据:", result)
	}
}

Golang Socket