go语言消息推送服务器,Golang 消息推送系统

## Golang 消息推送系统

- 使用 websocket 协议，实现全局推送，和分房间推送,实现用户加入房间，退出房间

- gateway 长连接网关服务器，管理全部的链接，以及房间等相关操作

- logic 逻辑服务器 管理发送全局推送，单个房间推送，加入房间，离开房间等相关操作

- 代码目录详情

```bash

├── gateway // 长连接网关服务器

│   ├── push_job.go // 分发任务

│   ├── room.go // 房间，可作为某一类型的推送管理中心

│   ├── room_manage.go // 房间管理

│   ├── ws_conn.go // 简单封装的websocket方法

│   ├── ws_handle.go // 处理websocket协议方法

│   └── ws_server.go // websocket服务

├── logic //逻辑服务器

│   ├── http_handle.go // 推送，房间相关

│   └── http_server.go // http服务

└── main.go

```

### websocket相关

- 定义WsConnection结构体

```go

type WsConnection struct {

mu sync.Mutex

connId string // 链接ID

ws *websocket.Conn // ws链接

readChan chan *WSMessage // 读取数据的管道

writeChan chan *WSMessage // 写数据的管道

closeChan chan bool // 关闭链接的管道

isOpen bool // 链接是否有效

addRoom *sync.Map // 该链接加入了那些房间

clientIp string // 客户端IP

}

```

- 维持心跳

```go

switch {

case msg.Type == 1://{"type":1,"data":"PING"},{"type":1,"data":"PONG"}

_ = w.ws.SetReadDeadline(time.Now().Add(time.Second * 10))

_ = w.SendMsg(&WSMessage{Type: 1, Data: "PONG"})

default:

fmt.Println("OTHER", msg.Type, msg.Data)

}

```

### 房间管理

```go

//一个房间代表一个订阅推送类型

type Room struct {

id int // 房间ID

title string // 房间标题

RConn sync.Map // 加入该房间的链接

}

type RoomManage struct {

AllRoom sync.Map // 所有的房间

AllConn sync.Map // 该网关的上的全部链接

}

```

## gateway 协议 [在线测试WS](http://www.easyswoole.com/wstool.html)

- ws://0.0.0.0:8888/connect

- PING Client -> Server

```json

{"type":1,"data":"PING"}

```

- PONG Server -> Client

```json

{"type":1,"data":"PONG"}

```

## logic 协议

- 全部推送

```bash

curl http://0.0.0.0:9999/push/all -X POST -d "val=test push&tag=3"

```

- 单个房间推送

```bash

curl http://0.0.0.0:9999/push/room -X POST -d "val=test push&tag=3&id=1"

```

- 加入房间 (wsId 在用户链接上gateway时可以得到)

```bash

curl http://0.0.0.0:9999/room/join -X POST -d "id=1&wsId=3cc97117-aa4d-55bb-86db-a1d77e51283b"

```

- 离开房间

```bash

curl http://0.0.0.0:9999/room/leave -X POST -d "id=1&wsId=3cc97117-aa4d-55bb-86db-a1d77e51283b"

```

### DEMO [代码地址](https://github.com/hwholiday/learning_tools/tree/master/go_push)

## 联系 QQ: 3355168235

有疑问加站长微信联系(非本文作者)