七叶笔记 » golang编程 » go-websocket 分布式IM

go-websocket 分布式IM

基于golang实现的分布式聊天系统,支持一对一聊天,聊天室等功能。为了测试方便发送消息数据暂未存入数据库,后期会加入数据库,也可自行加入数据库,方便永久存储聊天内容,以及支持消息必达等功能。

依赖包

 github.com/go-redis/redis
github.com/gin-gonic/gin
github.com/gorilla/websocket
github.com/smallnest/rpcx  

包说明:

redis :用于缓存ws服务器信息,用心跳形式维护ws服务器信息。

gin:实现web服务

websocket: 实现websocket协议

rpcx:服务器建rpc通信

架构图

一对一发消息

  • 客户端 发送建立长连接请求,经过nginx负载均衡分配给其中一台ws服务器(这里假设分配的是A服务器)处理。
  • A服务器响应长连接请求,并缓存客户端地址和用户连接,用户id等信息。
  • 客户端收到服务端响应建立websocet连接。
  • 客户端发送信息,nginx负载均衡分配给其中一台ws服务器(这里假设是B服务器)。
  • B服务器从发送的信息中解析接收用户(假设为a)信息,先验证a用户是否和B服务器建立websocet连接,若建立则直接发送消息给a用户。否则通过redis缓存中获取ws服务器信息列表,通过rpc方式发送消息到ws服务器列表中除B服务器之外的每台ws服务器,这些接收到发送信息的ws服务器,先验证和a用户是否建立连接,建立则发送信息给a用户,否则丢弃。

群发消息

  • 客户端 发送建立长连接请求,经过nginx负载均衡分配给其中一台ws服务器(这里假设分配的是A服务器)处理。
  • A服务器响应长连接请求,并缓存客户端地址和用户连接,用户id等信息。
  • 客户端收到服务端响应建立websocet连接。
  • 客户端发送信息,nginx负载均衡分配给其中一台ws服务器(这里假设是B服务器)。
  • B服务器从发送的信息中解析出群信息,根据群信息获取用户列表,遍历用户发送信息(发送方式跟一对一类似)。
  • 先验证用户是否和B服务器建立websocet连接,若建立则直接发送消息给用户。否则通过redis缓存中获取ws服务器信息列表,通过rpc方式发送消息到ws服务器列表中除B服务器之外的每台ws服务器,这些接收到发送信息的ws服务器,先验证和用户是否建立连接,建立则发送信息给用户,否则丢弃。

快速搭建

1、拉取代码

 git clone   

注:这里代码版本控制使用go modules

2、运行系统

 go run main.go  

3、配置nginx

 upstream  go-http
{
    server 127.0.0.1:8282 weight=1 max_fails=2 fail_timeout=10s;
    keepalive 16;
}

upstream  go-ws
{
    server 127.0.0.1:8089 weight=1 max_fails=2 fail_timeout=10s;
    keepalive 16;
}

server {
        listen        80;
        server_name  ws.test;
        root   "D:/phpstudy/WWW/";
          location /ws {
        proxy_set_header Host $host;
        proxy_pass 
        proxy_http_version 1.1;
        proxy_set_header Upgrade $http_upgrade;
        proxy_set_header Connection $connection_upgrade;
        proxy_set_header Connection "";
        proxy_redirect off;
        proxy_intercept_errors on;
        client_max_body_size 10m;
    }

    location /
    {
        proxy_set_header X-Forwarded-For $remote_addr;
        proxy_set_header Host $host;
        proxy_pass 
    }
}
  

4、测试一对一聊天

浏览器打开两个窗口访问

5、测试群聊天

浏览器打开两个窗口访问

相关资料:

github.com/gorilla/websocket

相关文章