golang就是爽,代码简单易懂。稍微看了一下gnatsd的源码,应该这样改就可以:
- 首先要自定义鉴权系统,每个用户都需要使用不同的用户名进行鉴权登录。
- 修改server.users新增一个clientid字段,记录登录成功的用户id,当然要记得写入时上server的锁。
- client增加一个kicked变量做为标志。
- 鉴权成功在server.users中查看是否已经有clientid,如果有则在server.cliens中查找client,找到client设置此client的kicked标志,并更新server.users的clientid为当前clientid。
- 在client.readloop中的read下面检测client.kicked标志,如果标志被设置则给客户端发出一个被重入的消息,然后closeconnection。
- 使用server里的锁保护server.users.client字段,可以保证原子性。
- 客户端收到被重入的错误消息以后,需要在界面上做出响应。
自定义鉴权系统的问题
自定义鉴权系统打算使用外部数据库Redis保存user和pwd。
每次都去Redis中查询进行鉴权。
此时本地的server.users列表就不用来鉴权,只用来保存当前登录user对应的clientid。
本地配置文件还有可能被reload,这样users列表会被覆盖,考虑不使用原本的users列表再新增定义一个userclientids列表。
集群时的问题
当使用集群时,同一个帐号可能登录在不同实例上,以上修改只能踢出当前登陆实例上的老Client,而在其他实例上还是保持着连接。
所以还需要给所有其他集群广播一条kickuser的消息。