diff options
author | David Heinemeier Hansson <david@loudthinking.com> | 2015-07-04 21:30:48 +0200 |
---|---|---|
committer | David Heinemeier Hansson <david@loudthinking.com> | 2015-07-04 21:30:48 +0200 |
commit | 0de65cf2d8860377d45020a83866073c5fec188c (patch) | |
tree | d39b559162936fd79049595d69098fd7712a8fe3 /lib/action_cable/connection/message_buffer.rb | |
parent | 9886a995f5f0b32d0d400074c48221cb0f6b911e (diff) | |
parent | 5de01033150b70982f23a42670c55348a7371c4b (diff) | |
download | rails-0de65cf2d8860377d45020a83866073c5fec188c.tar.gz rails-0de65cf2d8860377d45020a83866073c5fec188c.tar.bz2 rails-0de65cf2d8860377d45020a83866073c5fec188c.zip |
Merge branch 'master' of github.com:basecamp/action_cable
Diffstat (limited to 'lib/action_cable/connection/message_buffer.rb')
-rw-r--r-- | lib/action_cable/connection/message_buffer.rb | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/lib/action_cable/connection/message_buffer.rb b/lib/action_cable/connection/message_buffer.rb new file mode 100644 index 0000000000..615266e0cb --- /dev/null +++ b/lib/action_cable/connection/message_buffer.rb @@ -0,0 +1,51 @@ +module ActionCable + module Connection + class MessageBuffer + def initialize(connection) + @connection = connection + @buffered_messages = [] + end + + def append(message) + if valid? message + if processing? + receive message + else + buffer message + end + else + connection.logger.error "Couldn't handle non-string message: #{message.class}" + end + end + + def processing? + @processing + end + + def process! + @processing = true + receive_buffered_messages + end + + private + attr_reader :connection + attr_accessor :buffered_messages + + def valid?(message) + message.is_a?(String) + end + + def receive(message) + connection.send_async :receive, message + end + + def buffer(message) + buffered_messages << message + end + + def receive_buffered_messages + receive buffered_messages.shift until buffered_messages.empty? + end + end + end +end
\ No newline at end of file |