Go语言开发分布式聊天室

穿精又带淫゛_ 提交于 2021-02-10 07:37:11

声明

我是一个刚学go语言的菜鸟,还没有资格谈论什么技术分享,只是为了展示fooking的实际应用,同时把我用go写的聊天室代码贴出来供大家消遣,如果有入不了各位法眼的代码,望轻喷。该聊天室基于fooking,而业务代码是采用Go + Fastcgi。

完整的源代码在  https://github.com/scgywx/fooking/blob/master/example/chat/gateway.go,全代码200多行,去掉router部分代码,实际逻辑代码只有170来行,逻辑简单,功能强大。


详解

聊天服务器的入口main函数里有3个IP和端口配置,分别是Chat服务器、Router服务器和Redis服务器。

func main() {
	listener, _ := net.Listen("tcp", "0.0.0.0:9001")//Chat服务器配置
	srv := newChatServer("test:9010", "test:6379");//Router配置与Redis配置
	fcgi.Serve(listener, srv)
}

Chat服务器就是实现主要的聊天逻辑,Router服务器是用于转发消息,Redis用来存储用户信息。上面我讲过这个聊天室是基于fooking,所以客户端不是直接与go通信,而是透过fooking来访问的。我们使用go内置的fastcgi模块来创建一个服务,然后处理请求即可,这跟http服务器非常像,只是协议规范不同而已,看下面的代码就你知道创建一个fastcgi服务器有多简单了。

func (s *ChatServer) ServeHTTP(rep http.ResponseWriter, req *http.Request) {
	//短连接要调用一下这个,否则go不会主动断开连接
	//req.ParseForm();
	req.Form = make(url.Values);
	
	body, e := ioutil.ReadAll(req.Body)
	if e != nil {
		fmt.Printf("read request error\n")
	}else{
		sessionid := req.Header.Get("SESSIONID")
		event := req.Header.Get("EVENT")
		fmt.Printf("sid=%s, event=%s\n", sessionid, event);
		//具体的业务逻辑处理
        }
}

代码中的sessionid就是当前发送请求的客户端ID,当我们要做一些uid与客户端id映射或者是要发消息给指定用户的时候就可以使用这个ID。event就是当前请求的事件类型(0-表示请求,1-新连接,2-关闭连接),而聊天室只需要关心客户端请求和断开事件。请求可能是登陆、发消息或者是加入频道,而断开事件我们就需要把用户信息删除,并且把他的退出信息广播给所有在聊天室的人。代码如下:

switch event {
	case "1"://新连接
		//TODO
	case "2"://连接关闭
		s.logout(sessionid)
	default://消息处理				
		if len(body) > 0 {
			js, err := simplejson.NewJson(body)
			if err != nil {
				fmt.Printf("parse JSON error, data=")
				fmt.Println(body)
			}else{
				r := s.handle(sessionid, js)
				if len(r) > 0 {
					rep.Header().Add("Content-Length", strconv.Itoa(len(r)))
					rep.Write(r)
				}else{
					fmt.Println("no message response")
				}
			}
		}
}

func (s *ChatServer) handle(sid string, req *simplejson.Json) []byte{
	t, _ := req.Get("type").String()
	switch t {
		case "login": //登陆
			.....
		case "join": //加入房间
			....
		case "msg": //发送消息
			.....
		default:
			fmt.Printf("invalid type")
	}
	
	return []byte("")
}

在上面的消息处理部分,有两句rep.Header().Add和rep.Write,这表示如果需要返回数据给当前发请求的客户端,可以添加Content-Length头(表示要返回给客户端的数据长度),然后调用rep.Write发送数据(类似Http的Request对应一个Response)。

代码里面的消息广播、用户分频道都是在Router部分实现,他是fooking消息转发与用户数据维持的中间件。


为什么使用FastCGI

其实开发一个聊天室可以很简单,可以直接使用websocket协议,让客户端跟服务器直接通信,简单方便,通信代价低。并且做socket服务也不受协议限制,完全可以自定义或者是使用其它轻量级的协议,比如mqtt什么的。然而为什么要使用fastcgi呢?主要是他的协议实现简单,并且扩展性也非常强,目前世界上最强大的语言php的fpm就是使用该协议与nginx进行通信,如果你的服务使用fastcgi协议开发,那么你可以完美的使用nginx与你的服务进行通信,这样做的好处是写socket服务能像写web服务一样调试,开发完一个功能你只需要简单的执行如下命令即可(当然你可能需要添加少量的代码来判断来源是从nginx还是你自己的客户端)。

curl -d '{"type":"login","name":"xxx"}' 'http://fooking/gateway.php?SESSIONID=aaa&EVENT=0'

网关与逻辑分离另一个好处是,当业务逻辑代码需要更新,客户端毫无察觉的,真正做到无痛更新。另外fastcgi协议本身已经支持多路复用(当然这个需要服务端的支持),这个功能可是大名鼎鼎的http到2.0才支持的哟。

协议详细说明请见:http://www.fastcgi.com/drupal/node/6?q=node/22


为什么使用Fooking

文章开篇已经说了,我是一个Go语言的初学者,到这里来不是为了秀go技,而是为了展示fooking与各语言的衔接。那么为什么要使用fooking?我相信很多人都知道网关这个东西(可能在游戏领域应用的更广泛一些),他其主要的目的是用于承载客户端的连接,把消息转发与业务逻辑分开,后端开发人员只需要专心写逻辑即可。这就好比我们写web的时候,从来不需要自己去实现http server。那么fooking也是这样一个开源软件,他将socket服务变的更简单。更多的特性如下:

1 动态网关添加.
2 每个客户端唯一SessionID.
3 组播(类似redis的pub/sub).
4 服务器状态监控.
5 客户端事件通知(如:新连接、关闭连接).
6 后端无语言限制(php, python, go, nodejs, etc...).
7 自定义消息协议.
8 后端长连接维持.

fooking的详细介绍请参见: https://github.com/scgywx/fooking 或者 http://git.oschina.net/scgywx/fooking


使用方法

  • 第一步(下载和编译)
    git clone https://github.com/scgywx/fooking.git
    cd {$FOOKING_PATH}
    make

  • 第二步(启动Router)
    cd src
    ./fooking ../router.lua

  • 第三步(启动Gateway)
    ./fooking ../config.lua

  • 第四步(启动Chat服务器)

    go run gateway.go

  • 第五步(测试) 修改example/chat/index.html文件的Websocket的服务器IP和端口(查找ws://)
    然后用浏览器打开index.html即可


鸣谢

感谢@CFC4N牛腩五花肉对本次项目的大力支持。

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!