Contents

# encoding: utf-8
require "logstash/namespace"
require "logstash/outputs/websocket"
require "sinatra/base"
require "rack/handler/ftw" # from ftw
require "ftw/websocket/rack" # from ftw

class LogStash::Outputs::WebSocket::App < Sinatra::Base
  def initialize(channels, logger)
    @channels = channels
    @logger = logger
  end

  set :reload_templates, false

  get "/" do
    # TODO(sissel): Support filters/etc.
    ws = ::FTW::WebSocket::Rack.new(env)
    @logger.debug("New websocket client")
    stream(:keep_open) do |out|
      ws.each do |payload|
        json = payload.to_json
        if json.type == 'subscribe-topic'
          @channels[json.topic].subscribe do |event| 
            ws.publish(event)
          end 
        elsif json.type == 'subscribe-all'
          @channels.values.each do |channel|
            channel.subscribe do |event|
              ws.publish(event)
            end 
          end
        end 
      end # pubsub
    end # stream

    ws.rack_response
  end # get /
end # class LogStash::Outputs::WebSocket::App

Version data entries

3 entries across 3 versions & 2 rubygems

Version	Path
logstash-output-websocket_topics-2.1.4	lib/logstash/outputs/websocket/app.rb
logstash-output-websocket-topics-2.1.3	lib/logstash/outputs/websocket/app.rb
logstash-output-websocket-topics-2.1.2	lib/logstash/outputs/websocket/app.rb