go-server 一个在学习中形成的socket服务

最新推荐文章于 2024-08-13 08:05:22 发布

zboyco

最新推荐文章于 2024-08-13 08:05:22 发布

阅读量905

点赞数

分类专栏： golang socket 文章标签： go golang socket tcp

本文链接：https://blog.csdn.net/zboyco/article/details/104314488

版权

go-server是一个基于Golang的学习项目，发展成了一个具备TCP功能、支持自定义拆包协议、命令路由调用方法的socket服务。它包括Server服务、AppSession会话管理，提供过滤器和处理模块的接口，支持TLS。项目逐步完善了数据处理、命令路由、超时管理等功能，并记录了开发过程。

摘要由CSDN通过智能技术生成

go-server

项目地址：https://github.com/zboyco/go-server

go-server 是我在学习golang的过程中，从最简单的socket一步一步改造形成的。

目前功能如下：

普通的tcp功能（废话）
使用标准库bufio.Scanner实现拆包，可以直接使用bufio.Scanner内置的拆包协议，当然也可以自定义拆包协议
提供普通OnMessage和命令路由两种使用模式
提供单个Action添加路由方法,同时也采用实现ActionModule接口的方式批量添加路由
过滤器支持自定义，只需实现ReceiveFilter接口
支持设置会话超时时间，超时的会话会自动关闭
提供会话查找方法，可以根据ID或自定义属性查找会话
支持tls
…

问题如下：

目前只支持tcp4协议
…原谅我不会写文档 (╥╯^╰╥)
有什么问题大家随便留言
…

使用方法

安装

go get github.com/zboyco/go-server

简单使用

默认使用换行符\n拆分数据包

// main
func main() {
   
	// 新建服务
	mainServer := goserver.New("", 9043)
	// 注册OnMessage事件
	mainServer.SetOnMessage(onMessage)
	// 开启服务
	mainServer.Start()
}

// 接收数据方法
func onMessage(client *goserver.AppSession, token []byte) {
   
	// 将bytes转为字符串
	result := string(token)
	// 输出结果
	log.Println("接收到客户[", client.ID, "]数据:", result)
	// 发送给客户端
	client.Send([]byte("Got!"))
}

使用tls

使用NewWithTLS方法新建一个tls服务

	crt, err := tls.LoadX509KeyPair("server.crt", "server.key")
	if err != nil {
   
		log.Fatalln(err.Error())
	}
	// 新建服务
	mainServer := goserver.NewWithTLS("", 9043, &tls.Config{
   
		Certificates: []tls.Certificate{
   crt},
	})

自定义拆包协议

go-server 采用标准库bufio.Scanner实现数据拆包，默认使用ScanLines实现换行符拆包，支持自定义拆包规则，可以根据自己的需求制定，只需要自定义一个bufio.SplitFunc方法即可。
假设我们采用 head+body的方式定义package，并指定第1个字节是'$'，第4个字节是'#',第2、3位两个字节使用int16存储body长度，例子如下：

func main() {
   
	// 新建服务
	mainServer := goserver.New("", 9043)
	// 根据协议定义拆包规则
	mainServer.SetSplitFunc(func(data []byte, atEOF bool) (int, []byte, error) {
   
		if atEOF {
   
			return 0, nil, errors.New("EOF")
		}
		if data[0