NBIO快速入门：5分钟搭建高性能网络服务器-171主机测评

NBIO快速入门：5分钟搭建高性能网络服务器

【免费下载链接】nbio Pure Go 1000k+ connections solution, support tls/http1.x/websocket and basically compatible with net/http, with high-performance and low memory cost, non-blocking, event-driven, easy-to-use. 项目地址: https://gitcode.com/gh_mirrors/nb/nbio

NBIO是一款基于纯Go语言开发的高性能网络框架，专注于提供百万级连接解决方案，支持TLS、HTTP1.x和WebSocket协议，并且与net/http基本兼容。它采用非阻塞、事件驱动的设计模式，具有高性能、低内存消耗和易用性等特点，是构建高并发网络应用的理想选择。

为什么选择NBIO？

在当今互联网时代，高并发已经成为许多应用的基本需求。NBIO作为一款优秀的网络框架，具有以下显著优势：

卓越性能：NBIO采用事件驱动模型，能够高效处理大量并发连接，轻松支持百万级连接。
低内存消耗：精心优化的内存管理机制，确保在高并发场景下依然保持较低的内存占用。
丰富协议支持：全面支持TLS、HTTP1.x和WebSocket等主流网络协议。
易于使用：提供简洁的API接口，降低开发难度，让开发者能够快速上手。

环境准备

在开始使用NBIO之前，需要确保你的开发环境满足以下要求：

Go 1.13或更高版本
Git

首先，克隆NBIO仓库到本地：

git clone https://gitcode.com/gh_mirrors/nb/nbio

快速上手：构建简单TCP服务器

下面我们将通过一个简单的示例，展示如何使用NBIO构建一个高性能的TCP服务器。

代码实现

创建一个名为main.go的文件，添加以下代码：

package main

import (
"log"
"github.com/nbio/nbio"
)

func main() {
engine := nbio.NewEngine(nbio.Config{
Network: "tcp",
Addrs: []string{":8888"},
MaxWriteBufferSize: 6 * 1024 * 1024,
})

// 处理新连接
engine.OnOpen(func(c *nbio.Conn) {
log.Println("OnOpen:", c.RemoteAddr().String())
})

// 处理连接关闭
engine.OnClose(func(c *nbio.Conn, err error) {
log.Println("OnClose:", c.RemoteAddr().String(), err)
})

// 处理数据
engine.OnData(func(c *nbio.Conn, data []byte) {
c.Write(append([]byte{}, data…))
})

err := engine.Start()
if err != nil {
log.Fatalf("engine.Start failed: %v", err)
}
defer engine.Stop()

select {}
}

代码解析

在上面的代码中，我们首先创建了一个NBIO引擎实例，并配置了网络类型为TCP，监听地址为:8888，最大写缓冲区大小为6MB。

然后，我们注册了三个回调函数：

OnOpen：当有新连接建立时触发，我们在这里打印客户端的地址。
OnClose：当连接关闭时触发，我们在这里打印客户端地址和错误信息（如果有的话）。
OnData：当接收到数据时触发，我们在这里将接收到的数据原样返回给客户端，实现了一个简单的回显服务器功能。

最后，我们启动引擎，并在程序退出时停止引擎。

运行服务器

在终端中执行以下命令，编译并运行服务器：

go run main.go

如果一切正常，你将看到服务器启动成功的日志信息。此时，你可以使用telnet或其他TCP客户端工具连接到localhost:8888，发送消息并查看服务器的回显。

构建HTTP服务器

NBIO不仅支持TCP，还提供了对HTTP协议的良好支持。下面我们来构建一个简单的HTTP服务器。

代码实现

创建一个名为http_server.go的文件，添加以下代码：

package main

import (
"fmt"
"net/http"
"github.com/nbio/nbio/nbhttp"
)

func main() {
mux := &http.ServeMux{}
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "Hello, NBIO!")
})

server := nbhttp.NewServer(nbhttp.Config{
Network: "tcp",
Addrs: []string{":8080"},
Handler: mux,
})

err := server.Start()
if err != nil {
fmt.Printf("server.Start failed: %v", err)
return
}
defer server.Stop()

select {}
}

运行HTTP服务器

在终端中执行以下命令，编译并运行HTTP服务器：

go run http_server.go

然后，打开浏览器访问http://localhost:8080，你将看到"Hello, NBIO!"的响应。

构建WebSocket服务器

WebSocket是一种在单个TCP连接上进行全双工通信的协议，NBIO对WebSocket提供了良好的支持。下面我们来构建一个简单的WebSocket服务器。

代码实现

创建一个名为ws_server.go的文件，添加以下代码：

package main

import (
"log"
"net/http"
"github.com/nbio/nbio/nbhttp"
"github.com/nbio/nbio/nbhttp/websocket"
)

func onWebsocket(w http.ResponseWriter, r *http.Request) {
upgrader := websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool {
return true
},
}

conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Printf("upgrader.Upgrade failed: %v", err)
return
}

go func() {
defer conn.Close()
for {
_, message, err := conn.ReadMessage()
if err != nil {
log.Printf("conn.ReadMessage failed: %v", err)
break
}
err = conn.WriteMessage(websocket.TextMessage, message)
if err != nil {
log.Printf("conn.WriteMessage failed: %v", err)
break
}
}
}()
}

func main() {
mux := &http.ServeMux{}
mux.HandleFunc("/ws", onWebsocket)

server := nbhttp.NewServer(nbhttp.Config{
Network: "tcp",
Addrs: []string{":8080"},
Handler: mux,
})

err := server.Start()
if err != nil {
log.Fatalf("server.Start failed: %v", err)
}
defer server.Stop()

select {}
}

运行WebSocket服务器

在终端中执行以下命令，编译并运行WebSocket服务器：

go run ws_server.go

你可以使用WebSocket客户端工具连接到ws://localhost:8080/ws，发送消息并查看服务器的回显。

总结

通过本文的介绍，我们了解了NBIO的基本特点和优势，并通过简单的示例展示了如何使用NBIO构建TCP、HTTP和WebSocket服务器。NBIO作为一款高性能的网络框架，为开发者提供了简单易用的API，帮助我们快速构建高并发的网络应用。

如果你想深入了解NBIO的更多功能和用法，可以参考项目中的示例代码和文档。希望本文能够帮助你快速入门NBIO，并在实际项目中发挥它的强大作用。

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

NBIO快速入门：5分钟搭建高性能网络服务器