Contents

require 'em-websocket'
require 'json'

class DripDrop
  class WebSocketHandler
    attr_reader :ws, :address, :thread
   
    def initialize(address,opts={})
      @raw    = false #Deal in strings or ZMQ::Message objects
      @thread = Thread.new do
          host, port = address.host, address.port.to_i
          @debug = opts[:debug] || false

          ws_conn = EventMachine::WebSocket::Connection
          EventMachine::start_server(host,port,ws_conn,:debug => @debug) do |ws|
            @ws = ws
            @ws.onopen do
              @onopen_handler.call(ws) if @onopen_handler
            end
            @ws.onmessage do |message|
              unless @raw
                begin
                  parsed = JSON.parse(message)
                  message = DripDrop::Message.new(parsed['name'], :body => parsed['body'], :head => parsed['head'] || {})
                rescue StandardError => e
                  puts "Could not parse message: #{e.message}"
                end
              end
              @onmessage_handler.call(message,ws) if @onmessage_handler
            end
            @ws.onclose do
              @onclose_handler.call(@ws) if @onclose_handler
            end
            @ws.onerror do
              @onerror_handler.call(@ws) if @onerror_handler
            end
          end
      end
    end
    
    def on_recv(&block)
      @raw = false
      @onmessage_handler = block
      self
    end

    def on_recv_raw(&block)
      @raw = true
      @onmessage_handler = block
      self
    end

    def on_open(&block)
      @onopen_handler = block
      self
    end
  
    def on_close(&block)
      @onclose_handler = block
      self
    end
    
    def on_error(&block)
      @onerror_handler = block
      self
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
dripdrop-0.3.1	lib/dripdrop/handlers/websockets.rb
dripdrop-0.3.0	lib/dripdrop/handlers/websockets.rb
dripdrop-0.2.0	lib/dripdrop/handlers/websockets.rb
dripdrop-0.1.0	lib/dripdrop/handlers/websockets.rb