lib/angelo/responder/websocket.rb in angelo-0.1.11 vs lib/angelo/responder/websocket.rb in angelo-0.1.12

@@ -28,10 +28,12 @@
       begin
         if @response_handler
           Angelo.log @connection, @request, @websocket, :switching_protocols
           @bound_response_handler ||= @response_handler.bind @base
           @websocket.on_pong &WebsocketResponder.on_pong
+          @base.before if @base.respond_to? :before
           @bound_response_handler[@websocket]
+          @base.after if @base.respond_to? :after
         else
           raise NotImplementedError
         end
       rescue IOError => ioe
         warn "#{ioe.class} - #{ioe.message}"