jam-cloud/lib/jam_websockets/server.rb

require 'em-websocket'

module JamWebsockets
  
  class Server

    def initialize(options={})
      @log = Logging.logger[self]
      @count=0
      @router = Router.new
    end

    def run(options={})

      host = "0.0.0.0"
      port = options[:port]
      max_stale_connection_time = options[:max_stale_connection_time]
      max_reconnect_time = options[:max_reconnect_time]

      @log.info "starting server #{host}:#{port} with staleness_time=#{max_stale_connection_time}"

      EventMachine.run do
        @router.start(max_stale_connection_time)

        # if you don't do this, the app won't exit unless you kill -9
        at_exit do
            @log.info "cleaning up server"
            @router.cleanup
        end

        start_connection_flagger(max_stale_connection_time)
        start_connection_cleaner(max_reconnect_time)

        start_websocket_listener(host, port, options[:emwebsocket_debug])
      end
    end

    def start_websocket_listener(listen_ip, port, emwebsocket_debug)
      EventMachine::WebSocket.start(:host => listen_ip, :port => port, :debug => emwebsocket_debug) do |ws|
        @log.info "new client #{ws}"
        @router.new_client(ws)
      end
    end


    def start_connection_cleaner(stale_max_time)
      # one cleanup on startup
      cleanup_stale_connections(stale_max_time)

      EventMachine::PeriodicTimer.new(15) do
        cleanup_stale_connections(stale_max_time)
      end

    end

    def cleanup_stale_connections(stale_max_time)
      ConnectionManager.active_record_transaction do |connection_manager|
        connection_manager.remove_stale_connections(stale_max_time)
      end
    end

    def start_connection_flagger(flag_max_time)
      # one cleanup on startup
      flag_stale_connections(flag_max_time)

      EventMachine::PeriodicTimer.new(15) do
        flag_stale_connections(flag_max_time)
      end
    end

    def flag_stale_connections(flag_max_time)
      ConnectionManager.active_record_transaction do |connection_manager|
        connection_manager.flag_stale_connections(flag_max_time)
      end
    end

  end

end