如何使用sinatra流式api关闭redis连接?

时间:2011-12-20 04:56:40

标签: ruby sinatra redis publish-subscribe

我有以下sinatra应用程序:

require 'sinatra'
require 'redis'
require 'json'


class FeedStream < Sinatra::Application


  helpers do
    include SessionsHelper

    def redis
      @redis ||= Redis.connect
    end
  end

  get '/feed', provides: 'text/event-stream' do


    stream do |out|

      redis.subscribe "feed" do |on|

        on.message do |channel, message|
          event_data = JSON.parse message
          logger.info "received event = #{event_data}"
          out << "event: #{event_data['event']}\n"
          out << "data: #{{:data => event_data['data'],
                           :by => current_user}}.to_json\n\n"
        end
      end
    end

  end

end

基本上,它使用redis pubsub接收其他用户发布到feed的事件,然后使用sinatra streaming api发送这些事件。 问题是,当浏览器重新连接到源时,redis客户端保持连接,并且它一直接收事件,因此redis服务器充满了无用的连接。 一旦broser关闭与Web服务器的连接,我该如何关闭所有这些连接?

2 个答案:

答案 0 :(得分:1)

我知道已经有一段时间了。

你在寻找quit吗?

答案 1 :(得分:1)

经过大量的研究和实验,这里是我与sinatra + sinatra sse gem一起使用的代码(应该很容易适应Rails 4):

class EventServer < Sinatra::Base
 include Sinatra::SSE
 set :connections, []
 .
 .
 .
 get '/channel/:channel' do
 .
 .
 .
  sse_stream do |out|
    settings.connections << out
    out.callback {
      puts 'Client disconnected from sse';
      settings.connections.delete(out);
    }
  redis.subscribe(channel) do |on|
      on.subscribe do |channel, subscriptions|
        puts "Subscribed to redis ##{channel}\n"
      end
      on.message do |channel, message|
        puts "Message from redis ##{channel}: #{message}\n"
        message = JSON.parse(message)
        .
        .
        .
        if settings.connections.include?(out)
          out.push(message)
        else
          puts 'closing orphaned redis connection'
          redis.unsubscribe
        end
      end
    end
  end
end

redis连接阻止on.message并且只接受(p)subscribe /(p)unsubscribe命令。取消订阅后,redis连接不再被阻止,并且可以由初始sse请求实例化的Web服务器对象释放。当您在redis上收到消息时,它会自动清除,并且集合数组中不再存在与浏览器的连接。