前言
上一篇博客提到了Kqueue
的基本原理并用一个小的编程例子进行实现,对于Kqueue
的基本使用尚不熟练的小伙伴可以回头看一下这篇博客。下面言归正传,用Kqueue
实现一个能实用的echo
服务器,并讨论多客户端连接的应用场景。
单线程echo服务器
设计思想:服务端创建一个套接字监听来自客户端的连接,并将其放入内核事件队列中,每次接到客户端的连接请求,创建套接字放入监听列表待其活跃之后进行读写。
注意:在单线程的服务器中,多个客户端同时连接,服务器会阻塞在第一个进行读写的客户端,直至该客户端断开连接后才会对活跃列表中的下一个事件进行读写
代码比较占篇幅,下面只贴出核心的代码,完整代码在我的github上,之后会对一些尚未解决的小bug进行更新
func main() {
//server initialization
serverAddr.Port = ServerPort
copy(serverAddr.Addr[:], net.ParseIP(ServerAddr).To4())
......
//create kqueue
kq, err := syscall.Kqueue()
if err != nil {
panic(err)
}
//add sockets to changes
changes = append(changes, syscall.Kevent_t{
Ident: uint64(serverFd), Filter: syscall.EVFILT_READ, Flags: syscall.EV_ADD})
events := make([]syscall.Kevent_t, MaxEvents) //要创建一个有长度的切片,可以试试events = []syscall.Kevent_t,看看发生什么情况
fdToClient = make(map[int]syscall.SockaddrInet4) //创建fd与socket address的映射关系
for {
nev, err := syscall.Kevent(kq, changes, events, nil)
if err != nil && err != syscall.EINTR {
panic(err)
}
for i:=0; i<nev; i++ {
if int(events[i].Ident) == serverFd {
//如果有客户端连接,将这个连接的文件描述符放入changes数组等待其活跃
clientFd, sockAddr, err := syscall.Accept(serverFd)
if err != nil {
panic(err)
}
clientAddr, err := SockaddrToAddr(sockAddr)
fdToClient[clientFd] = clientAddr