go-websocket 分布式IM

时间 2021-05-26

标签 php nginx git github golang web redis 数据库 segmentfault 浏览器栏目系统架构繁體版

原文原文链接

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

依赖包

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

包说明：nginx

redis ：用于缓存ws服务器信息，用心跳形式维护ws服务器信息。git

gin：实现web服务github

websocket：实现websocket协议golang

rpcx：服务器建rpc通讯web

架构图

一对一发消息

客户端发送创建长链接请求，通过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服务器，先验证和用户是否创建链接，创建则发送信息给用户，不然丢弃。

快速搭建

一、拉取代码

git clone https://github.com/guyan0319/go-websocket.git

注：这里代码版本控制使用go modulesredis

二、运行系统

go run main.go

三、配置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 http://go-ws;
        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 http://go-http;
    }
}