"io"
"net"
"time"
)
func main() {
server := ":8330"
tcpAddr, err := net.ResolveTCPAddr("tcp", server)
if err != nil {
fmt.Println("resolve err:", err)
return
}
listen, err := net.ListenTCP("tcp", tcpAddr)
if err != nil {
fmt.Println("listen err:", err)
return
}
defer listen.Close()
for {
conn, err := listen.Accept()
if err != nil {
fmt.Println("accept err:", err)
continue
}
go handler(conn)
}
}
func handler(conn net.Conn) {
go func() {
for {
// 指定从buffer中读取数据的最大容量
var buf = make([]byte, 1024)
// 从buffer中读取数据并保存到buf中,n代表实际返回的数据大小
n, err := conn.Read(buf)
if err != nil {
// 客户端关闭会触发EOF
if err == io.EOF {
conn.Close()
return
}
fmt.Println("read err:", err)
return
}
fmt.Println("read data ", n, ":", string(buf))
}
}()
curTime := time.Now().String()
// 数据写到缓冲区
_, err := conn.Write([]byte(curTime))
if err != nil {
fmt.Println("write err:", err)
return
}
fmt.Println("send data:", curTime)
}
首先要明白,操作系统内核会为每个连接的客户端和服务端分配发送缓冲区和接收缓冲区。
- 当客户端需要发送数据到服务端,调用conn.Write从客户端缓冲区发送数据到操作系统内核的发送缓冲区。实际所做的事情是把数据从应用程序缓冲区中拷贝到操作系统内核的发送缓冲区中,并不一定是把数据通过套接字写出去。
- 数据通过tcp发送到服务端的接收缓冲区,然后服务端的程序从接收缓冲区读取数据。
非阻塞I/O,当应用程序调用非阻塞 I/O 完成某个操作时,内核立即返回,不会把 CPU 时间切换给其他进程,应用程序在返回后,可以得到足够的 CPU 时间继续完成其他事情。
读操作:如果套接字对应的接收缓冲区没有数据可读,在非阻塞情况下 read 调用会立即返回,一般返回 EWOULDBLOCK 或 EAGAIN 出错信息。
写操作:在非阻塞 I/O 的情况下,如果套接字的发送缓冲区已达到了极限,不能容纳更多的字节,那么操作系统内核会尽最大可能从应用程序拷贝数据到发送缓冲区中,并立即从 write 等函数调用中返回。可想而知,在拷贝动作发生的瞬间,有可能一个字符也没拷贝,有可能所有请求字符都被拷贝完成,那么这个时候就需要返回一个数值,告诉应用程序到底有多少数据被成功拷贝到了发送缓冲区中,应用程序需要再次调用 write 函数,以输出未完成拷贝的字节。
非阻塞 I/O 操作:拷贝→返回→再拷贝→再返回。
阻塞 I/O 操作:拷贝→直到所有数据拷贝至发送缓冲区完成→返回。
golang中底层使用的还是非阻塞的I/O,但是在代码层面做了一些处理,让用户感觉是以阻塞方式调用的。
...
for {
n, err := ignoringEINTRIO(syscall.Read, fd.Sysfd, p)
if err != nil {
n = 0
// 非阻塞方式调用,如果遇到syscall.EAGAIN报错,代表没拿到数据,继续循环
if err == syscall.EAGAIN && fd.pd.pollable() {
if err = fd.pd.waitRead(fd.isFile); err == nil {
continue
}
}
}
err = fd.eofError(n, err)
return n, err
}
...
在socket中,当客户端调用close()方法时,其实就是发送一个FIN标志位,意思就是我要主动关闭TCP连接了。Close方法会让对端的所有读写操作结束阻塞,并返回。
在golang中调用Close方法,会让对端的Read读取到EOF的错误,此时就代表我想关闭连接。对端接收到关闭的请求后也可以调用Close方法关闭连接。
客户端:
package main
import (
"fmt"
"io"
"net"
)
func main() {
serverAddr := ":8330"
tcpAddr, err := net.ResolveTCPAddr("tcp", serverAddr)
if err != nil {
fmt.Println("resolve err:", err)
return
}
conn, err := net.DialTCP("tcp", nil, tcpAddr)
if err != nil {
fmt.Println("dial err:", err)
return
}
closed := make(chan bool)
go func() {
for {
var buf = make([]byte, 1024)
n, err := conn.Read(buf)
if err != nil {
// 读取到EOF,服务端关闭连接
if err == io.EOF {
conn.Close()
closed <- true
return
}
fmt.Println("read err:", err)
return
}
fmt.Println("read data ", n, ":", string(buf))
}
}()
for {
select {
case <-closed:
fmt.Println("服务端关闭")
return
}
}
}
服务端