WebSocket用Redis实时双工通信

最近在梳理一些知识点,已脱敏并去除公司实现,做一些自己理解上的实践。

结构

本次打算模拟下一个实时双工交互的业务实践,先来张图。

模式结构图

可以看出,实时双工通信的基础在于Redis部分,核心就在于Pub/Sub模型,其余部分在此基础上丰富了交互内容。

  • Server端 ,用于模拟平时业务机器,对来自客户端的Request给予Response
  • WebSocket Server端,比如直播业务中在直播间内聊天,肯定要用websocket来维系链接状态,这里可以做到语言无关,既可以用Java,也可以用golang。原理都是类似的。根据双工的特征,websocket服务器与客户端发生信息交互的场景无非主动和被动,场景如下:
    • 主动触发, 指的是来自另一个客户端的action,触发了websocket服务器的push行为。
    • 被动触发,比如定时器触发,状态检查等行为,都属于被动触发
  • APP端,一般团队都会以APP形式落地到终端用户,web网页也是类似。在直播场景中,主播、观众实际上可以统一虚化为客户端

实现

从图示上来看,每一个模块都不难,一点点去实现就好了。因为不想涉及公司业务上的东西,所以会有一些改动,如下:

  • APP端我这里会用JavaScript来作为客户端,简单模拟下就。
  • WebSocket服务器端,不打算用公司里Java版本,想试试golang版本。
  • Server端就用PHP简单模拟下。

websocket服务器端实现

// server.go
package main

import (
	"fmt"
	"log"
	"net/http"

	"time"

	"github.com/garyburd/redigo/redis"
	"golang.org/x/net/websocket"
)

var clients map[*websocket.Conn]string = make(map[*websocket.Conn]string)

// websocket 服务器端测试

func Echo(ws *websocket.Conn) {
	var err error
	if _, ok := clients[ws]; ok != true {
		clients[ws] = "匿名"
	}
	for {
		var reply string
		if err = websocket.Message.Receive(ws, &reply); err != nil {
			fmt.Println("Cannot receive")
			break
		}
		fmt.Println("Current client number: ", len(clients))
		fmt.Println("Received back from client: ", reply)
		msg := "RECEIVED: " + reply
		fmt.Println("Sending to client: " + msg)
		for client, _ := range clients {
			fmt.Println(client)
			if err = websocket.Message.Send(client, msg); err != nil {
				fmt.Println("Sending failed")
				break
			}

		}
	}
}

func tick() {
	for {
		time.Sleep(time.Second * 10)
		fmt.Println("checking ping")
		for key, _ := range clients {
			if key.IsClientConn() == false {
				// delete(clients, key)
			}
		}
		// 对所有客户端进行订阅消息的推送
		consume(clients)
	}
}

func consume(clients map[*websocket.Conn]string) {
	client, err := redis.Dial("tcp", "localhost:6379")
	defer client.Close()
	if err != nil {
		fmt.Println(err)
	}
	psc := redis.PubSubConn{Conn: client}
	psc.Subscribe("channel")
	for {
		switch v := psc.Receive().(type) {
		case redis.Message:
			fmt.Printf("%s: message: %s\n", v.Channel, v.Data)
			for client, _ := range clients {
				fmt.Println(client)
				if err = websocket.Message.Send(client, string(v.Data)); err != nil {
					fmt.Println("Sending failed")
					break
				}
			}
		case redis.Subscription:
			fmt.Printf("%s: %s %d\n", v.Channel, v.Kind, v.Count)
		}
	}
}

func main() {
	http.Handle("/", websocket.Handler(Echo))
	go tick()
	if err := http.ListenAndServe(":1234", nil); err != nil {
		log.Fatal("ListenAndServe failed: ", err)
	}
}

app端实现

// client.js
var wsServer = 'ws://localhost:1234';
var websocket = new WebSocket(wsServer);
websocket.onopen = function (evt) {
    console.log("Connected to WebSocket server.");
};

websocket.onclose = function (evt) {
    console.log("Disconnected");
};

websocket.onmessage = function (evt) {
    console.log('Retrieved data from server: ' + evt.data);
};

websocket.onerror = function (evt, e) {
    console.log('Error occured: ' + evt.data);
};
// 发送消息
websocket.send("hello world!")

server端实现

<?php

$redis = new Redis();
$redis->pconnect("localhost", 6379);
$ret = $redis->publish("channel", "data from PHP");
var_dump($ret);

测试

按照图例,打算对主动触发和被动触发进行下测试。

主动触发

主动触发其实就是对websocket基本功能的测试,一般都是开俩客户端,一段发消息,看看另一端是否能收到就可以了,流程是

1.启动websocket服务器

→ go run server.go

2.客户端连接websocket服务器

Chrome 打开调试器console输入上面的JavaScript代码即可。

3.交互测试
主动触发测试

被动触发

被动触发一般都是定时任务,如定时器timer来触发的。在上面golang代码中,有这么一段调用。

go tick()
// 内部调用了consume方法,来实现对subscribe消息的消费

1.启动websocket服务器

➜ go run server.go
checking ping

2.客户端js链接websocket服务器
javascripe链接websocket服务器
3.PHP去publish消息

➜ php publish.php
int(1)

4.查看客户端是否可以收到websocket消费到的数据
查看客户端的确收到了对应publish的消息

整理

至此,基本上双工实时通信的demo就结束了。相比于公司Java实现的版本,大体框架是类似的,无非是有没有业务的支撑。

展开阅读全文

没有更多推荐了,返回首页