ruby - 发送给 EventMachine pubsub 订阅者的重复数据

Question

堆栈:Ruby 2.3.1，Rack，瘦

简单的 websocket 服务器:

require 'redis'
require 'em-hiredis'
require 'faye/websocket'
require 'json'

ws_channel = {}

App = lambda do |env|
$redis ||= EM::Hiredis.connect('redis://127.0.0.1:6379')

if Faye::WebSocket.websocket?(env)
ws = Faye::WebSocket.new(env, nil,
                         headers: {'Access-Control-Allow-Origin' => '*'},
                         ping: 15
)

ws.on :open do |event|
  puts 'client connected'
  query_string = event.current_target.env['REQUEST_PATH'].gsub(/[^a-z0-9\-_\/]/, '')
  ws_channel[query_string] ||= EM::Channel.new

  pubsub = $redis.pubsub

  puts "subscribing to ws channel: ws:#{query_string}"
  sid = ws_channel[query_string].subscribe do |msg|
    puts "WS -> ws:#{query_string}/ #{sid} #{ws_channel[query_string]}"
    ws.send msg
  end

  puts "subscribing to redis: #{query_string}"
  pubsub.subscribe(query_string) do |msg|
    puts "REDIS -> ws:#{query_string}/"
    $redis.setex(query_string, 60, msg)
    ws_channel[query_string].push msg
  end

  EventMachine.add_periodic_timer(5) do
    ws.send ({ :ts => Time.now.to_i}.to_json) if ws
  end

  ws.on :close do |event|
    puts "client ##{query_string} disconnected"
    pubsub.unsubscribe(query_string) if pubsub
    ws_channel[query_string].unsubscribe(sid) if ws_channel[query_string]
    ws = nil
    pubsub = nil
  end
end

ws.rack_response
end
end

config.ru:

require 'rubygems'
require 'bundler/setup'
require 'logger'
require File.expand_path('../app', __FILE__)
Faye::WebSocket.load_adapter('thin')

run App

启动服务器:

bundle exec thin -p 9292 -R config.ru start

发行条件:

1. 从同一 IP 建立到同一 WS channel 的多个连接(多个浏览器选项卡在同一台计算机上打开了同一游戏)。
2. 来自 WS 服务器的单个数据推送导致数据到达每个订阅者的次数与订阅者的次数一样多。
3. 如果刷新其中一个选项卡(与 WS 服务器的连接关闭并重新打开)，后续数据推送不会导致数据重复。
4. 建立新连接后，#2 中的场景再次出现。

我的解决方法是在连接打开时取消订阅/重新订阅。所以:

pubsub = $redis.pubsub
pubsub.unsubscribe(query_string) if pubsub
pubsub = $redis.pubsub

但这引入了另一个问题:当一个选项卡关闭时，数据停止到达其他选项卡大约 30 秒。 WS 连接永远不会关闭，我可以在 JS 控制台中看到 5 秒的 ping。

redis-cli $> PUBSUB NUMSUB <channel> 

• 这只显示一个 channel 订阅者，无论有多少订阅者订阅了该 channel

目标功能:

1. 多个客户端(订阅者)从同一个 IP 连接到同一个 channel 。
2. 每个订阅者收到一份WS服务器推送的数据
3. 客户端断开/新客户端连接不会导致其他客户端的任何服务中断。

Answer 1

为每个 WS 连接创建一个唯一的 EM channel 并取消订阅 ws.close 上的特定过程，似乎可以完成这项工作:

require 'redis'
require 'em-hiredis'
require 'faye/websocket'
require 'json'

App = lambda do |env|
$redis ||= EM::Hiredis.connect('redis://127.0.0.1:6379')
$pubsub ||= $redis.pubsub

if Faye::WebSocket.websocket?(env)
ws = Faye::WebSocket.new(env, nil,
                         headers: {'Access-Control-Allow-Origin' => '*'},
                         ping: 15
)

ws.on :open do |event|
  puts 'client connected'

  query_string = event.current_target.env['REQUEST_PATH'].gsub(/[^a-z0-9\-_\/]/, '')
  channel = EM::Channel.new

  puts "subscribing to ws channel: ws:#{query_string}"
  sid = channel.subscribe do |msg|
    puts "WS -> ws:#{query_string}/ #{sid} #{channel}"
    ws.send msg
  end

  puts "subscribing to redis: #{query_string}"
  subs = {}; r_callback = rand(Time.now.to_i)
  subs[r_callback] = Proc.new { |msg|
    puts "REDIS -> ws:#{query_string}/"
    $redis.setex(query_string, 60, msg)
    channel.push msg
  }
  $pubsub.subscribe(query_string, subs[r_callback])
  #puts $pubsub.inspect

  ws.on :close do |event|
    puts "client ##{query_string} disconnected"
    $pubsub.unsubscribe_proc(query_string, subs[r_callback]) if $pubsub
    puts "Unsubscribed proc: #{subs[r_callback]}"
    channel.unsubscribe(sid) if channel
    ws = nil
  end
end

ws.rack_response
end
end